yy_rms_39zhiyao_duizhao/apps/user/views.py

#!/usr/bin/env python
# -*- encoding: utf-8 -*-
'''
@Date:2022/07/19 16:34:34
'''
from flask import jsonify, request, Blueprint, g
from sqlalchemy import and_
from Common.auth import token_auth
from Common.Utils import PageParam, Utils

from db_logic.user import BllUser
from models.user_models import EntityUser


user_router = Blueprint('user', __name__)


@user_router.route("/login", methods=["POST"])
# @token_auth.login_required
def user_login():
    user_name = request.values.get("user_name")
    password = request.values.get('password')
    # user_name = body.user_name
    # password = body.password
    user_obj, user_info = BllUser().login(user_name=user_name, password=password)
    if user_obj:
        return jsonify(Utils.true_return(msg="登陆成功", data={"token":user_obj, "user_info": Utils.to_dict(user_info)}))
    else:
        return jsonify(Utils.false_return(status=201, msg="账号或密码输入有误！"))

# 修改密码,根据token获取用户信息，接受老密码进行校验，新密码写入
@user_router.route("/update_password", methods=["POST"])
@token_auth.login_required
def user_update_password():
    old_password = request.values.get("old_password")
    new_password = request.values.get("new_password")
    new_password1 = request.values.get("new_password1")
    # user_id = g.current_user.use
    if new_password != new_password1:
        return jsonify(Utils.false_return(msg="两次密码不一致"))
    msg_bool, msg_token = BllUser().update_password(g.current_user, old_password=old_password, new_password=new_password)
    if msg_bool:
        return jsonify(Utils.true_return(data={"token": msg_token}))
    else:
        return jsonify(Utils.false_return(msg=msg_token))


# 获取用户列表
@user_router.route("/get_user_list", methods=["GET", "POST"])
@token_auth.login_required
def get_user_info_list():
    user_code = request.values.get("user_code")
    real_name = request.values.get("real_name")
    role_name = request.values.get("role_name")
    is_enabled = request.values.get("is_enabled")

    page = int(request.values.get("page", 1))
    page_size = int(request.values.get("page_size", 10))
    page_param = PageParam(page, page_size)

    data_list = BllUser().get_seach_user_list(
        user_code=user_code,
        real_name=real_name,
        role_name=role_name,
        is_enabled=is_enabled,
        page_param=page_param
    )

    return jsonify(Utils.true_return(data={"total_count":page_param.totalRecords, "data_list": Utils.msyql_table_model(data_list)}))

# 新增用户
@user_router.route("/add", methods=["GET", "POST"])
@token_auth.login_required
def add_user_info():
    create_user_info = g.current_user
    new_obj = BllUser().execute(f"select * from rms_user where user_code={request.values.get('user_code')} and user_code is not null").fetchone()
    if new_obj:
        return jsonify(Utils.false_return(msg="工号不能重复"))
    try:
        obj = EntityUser(
            real_name=request.values.get("real_name"),
            mobile=request.values.get("mobile"),
            sex=request.values.get("sex"),
            email=request.values.get("email"),
            user_code=request.values.get("user_code"),
            qq=request.values.get("qq"),
            role_id=request.values.get("role_id"),
            role_name=request.values.get("role_name"),
            create_date=Utils.get_str_datetime(),
            create_user_id=create_user_info.user_id,
            create_user_name=create_user_info.real_name,
            is_enabled=request.values.get("is_enabled")
        )
        setattr(obj, "password", Utils.MD5('123456'))
        BllUser().insert(obj)
        return jsonify(Utils.true_return(msg=f"添加成功!"))
    except Exception as error:
        return jsonify(Utils.except_return(msg=error))

# 个人资料
@user_router.route("/getinfo", methods=["GET", "POST"])
@token_auth.login_required
def get_user_info():
    data = Utils.to_dict(g.current_user)
    try:
        data.pop("password")
    except Exception:
        pass
    return jsonify(Utils.true_return(data=data))

# 修改用户
@user_router.route("/update", methods=["GET", "POST"])
@token_auth.login_required
def update_user_info():
    user_id = request.values.get("user_id")
    try:
        obj = BllUser().findEntity(user_id)
        if not obj:
            return jsonify(Utils.false_return(msg="用户id有误"))
        else:
            # new_obj = BllUser().findEntity(
            #     and_(
            #     EntityUser.user_code == obj.user_code,
            #     EntityUser.user_code.isnot(None)
            #     )
            # )
            new_obj = BllUser().execute(f"select * from rms_user where user_code={request.values.get('user_code')} and user_code is not null").fetchone()

            if new_obj and  obj.user_id == new_obj.user_id:
                return jsonify(Utils.false_return(msg="工号不能重复"))
            finds_list = dir(EntityUser)
            for i in finds_list:
                va = request.values.get(i)
                if va:
                    setattr(obj, i, va)
            obj.update_last_visit_date()
            BllUser().update(obj)
            return jsonify(Utils.true_return(msg="修改成功"))
    except Exception as error:
        return jsonify(Utils.except_return(msg=error))
    

# 删除用户
@user_router.route("/del", methods=["GET", "POST"])
@token_auth.login_required
def del_user():
    user_id = request.values.get("user_id")
    user_id_list = user_id.split(',')
    for i in user_id_list:
    # if user_id:
        BllUser().delete(EntityUser.user_id==i)
    return jsonify(Utils.true_return(msg="删除成功"))
    # return jsonify(Utils.false_return("未选择用户"))