基本构建

applications/service/admin/user.py

@@ -0,0 +1,117 @@
+from flask import jsonify
+from flask_login import current_user
+from flask_marshmallow import Marshmallow
+from marshmallow import fields
+from sqlalchemy.sql import exists
+
+from applications.models import db
+from applications.models.admin import User, Role
+from sqlalchemy.orm.dynamic import AppenderQuery
+
+ma = Marshmallow()
+
+
+# 用户models的序列化类
+class UserSchema(ma.Schema):
+    id = fields.Integer()
+    username = fields.Str()
+    realname = fields.Str()
+    enable = fields.Bool()
+    create_at = fields.DateTime()
+    update_at = fields.DateTime()
+
+
+'''
+获取用户的sqlalchemy对象
+分页器
+'''
+
+
+def get_user_data(page, limit):
+    user = User.query.paginate(page=page, per_page=limit, error_out=False)
+    count = User.query.count()
+    return user, count
+
+
+'''
+获取用户的dict数据
+分页器
+'''
+
+
+def get_user_data_dict(page, limit):
+    user, count = get_user_data(page, limit)
+    user_schema = UserSchema(many=True)  # 用已继承ma.ModelSchema类的自定制类生成序列化类
+    output = user_schema.dump(user.items)  # 生成可序列化对象
+    return output, count
+
+
+'''
+通过名称获取用户
+'''
+
+
+def get_user_by_name(username):
+    return User.query.filter_by(username=username).first()
+
+
+# 判断用户是否存在
+def is_user_exists(username):
+    res = User.query.filter_by(username=username).count()
+    return bool(res)
+
+
+# 增加用户
+def add_user(username, realName, password):
+    user = User(username=username, realname=realName)
+    user.set_password(password)
+    db.session.add(user)
+    db.session.commit()
+
+
+# 更新用户信息
+def update_user(id, username, realname):
+    user = User.query.filter_by(id=id).update({'username': username, 'realname': realname})
+    db.session.commit()
+    return user
+
+
+def delete_by_id(id):
+    user = User.query.filter_by(id=id).delete()
+    return user
+
+
+def update_status(status):
+    user = User.query.filter_by(id=id).update({'status': status})
+    if user:
+        db.session.commit()
+        return jsonify(msg="更新成功", code=200)
+    return jsonify(msg="出错啦", code=999)
+
+
+def batch_remove(ids):
+    user = User.query.filter(User.id.in_(ids)).delete(synchronize_session=False)
+    db.session.commit()
+    if user:
+        res = jsonify(msg="批量删除成功", code=200)
+        return jsonify(res)
+    else:
+        res = jsonify(msg="批量删除失败", code=999)
+        return jsonify(res)
+
+
+def update_user_role(id, roles_list):
+    user = User.query.filter_by(id=id).first()
+    roles_id = []
+    for role in user.role:
+        print(role)
+        roles_id.append(role.id)
+    print('---------------------')
+    print(roles_id)
+    roles = Role.query.filter(Role.id.in_(roles_id)).all()
+    for r in roles:
+        user.role.remove(r)
+    roles = Role.query.filter(Role.id.in_(roles_list)).all()
+    for r in roles:
+        user.role.append(r)
+    db.session.commit()