108 lines
3.5 KiB
Python
108 lines
3.5 KiB
Python
from flask import render_template, request, make_response
|
|
from flask_restful import Resource
|
|
from applications.extensions import db
|
|
|
|
from applications.common.utils.http import fail_api, success_api
|
|
from applications.common.utils.rights import authorize
|
|
from applications.models import User, Role
|
|
from applications.view.company.users import user_api, users_bp, user_curd
|
|
|
|
|
|
@user_api.resource('/')
|
|
class Users(Resource):
|
|
"""用户列表数据操作"""
|
|
|
|
@authorize("admin:user:main", log=True)
|
|
def get(self):
|
|
return make_response(render_template('admin/user/main.html'))
|
|
|
|
@authorize("admin:user:add", log=True)
|
|
def post(self):
|
|
"""新建单个用户"""
|
|
req_json = request.json
|
|
a = req_json.get("roleIds")
|
|
username = req_json.get('username')
|
|
real_name = req_json.get('realName')
|
|
password = req_json.get('password')
|
|
role_ids = a.split(',')
|
|
|
|
if not username or not real_name or not password:
|
|
return fail_api(msg="账号姓名密码不得为空")
|
|
|
|
if user_curd.is_user_exists(username):
|
|
return fail_api(msg="用户已经存在")
|
|
|
|
_id = user_curd.add_user(username, real_name, password)
|
|
user_curd.add_user_role(_id, role_ids)
|
|
|
|
return success_api(msg="增加成功")
|
|
|
|
|
|
@user_api.resource('/<user_id>')
|
|
class CURDUser(Resource):
|
|
"""修改用户数据"""
|
|
|
|
@authorize("admin:user:edit", log=True)
|
|
def get(self, user_id):
|
|
# 获取编辑用户信息
|
|
user = User.query.filter_by(id=user_id).first()
|
|
roles = Role.query.all()
|
|
checked_roles = []
|
|
for r in user.role:
|
|
checked_roles.append(r.id)
|
|
return make_response(
|
|
render_template('admin/user/edit_users.html', user=user, roles=roles, checked_roles=checked_roles))
|
|
|
|
@authorize("admin:user:remove", log=True)
|
|
def delete(self, user_id):
|
|
# 删除用户
|
|
res = user_curd.delete_by_id(user_id)
|
|
if not res:
|
|
return fail_api(msg="删除失败")
|
|
return success_api(msg="删除成功")
|
|
|
|
@authorize("admin:user:edit", log=True)
|
|
def put(self, user_id):
|
|
# 更新用户数据
|
|
req_json = request.json
|
|
a = req_json.get("roleIds")
|
|
_id = req_json.get("userId")
|
|
username = req_json.get('username')
|
|
real_name = req_json.get('realName')
|
|
dept_id = req_json.get('deptId')
|
|
role_ids = a.split(',')
|
|
user_curd.update_user(user_id, username, real_name, dept_id)
|
|
user_curd.update_user_role(_id, role_ids)
|
|
return success_api(msg="更新成功")
|
|
|
|
|
|
# 批量删除
|
|
@users_bp.delete('/batchRemove')
|
|
@authorize("admin:user:remove", log=True)
|
|
def batch_remove():
|
|
ids = request.form.getlist('ids[]')
|
|
user_curd.batch_remove(ids)
|
|
return success_api(msg="批量删除成功")
|
|
|
|
|
|
@users_bp.put('/enable')
|
|
def user_enable():
|
|
# 启用或者禁用用户 enable disable
|
|
user_id = request.json.get('userId')
|
|
operate = request.json.get('operate')
|
|
user_id = int(user_id)
|
|
operate = int(operate)
|
|
if user_id:
|
|
if operate == 1:
|
|
user = User.query.filter_by(id=user_id).update({"enable": operate})
|
|
message = success_api(msg="启动成功")
|
|
else:
|
|
user = User.query.filter_by(id=user_id).update({"enable": operate})
|
|
message = success_api(msg="禁用成功")
|
|
if user:
|
|
db.session.commit()
|
|
else:
|
|
return fail_api(msg="出错啦")
|
|
return message
|
|
return fail_api(msg="数据错误")
|