JmedChat/app/api/v1/auth/crud/user.py

# !/usr/bin/env python3
# -*- encoding : utf-8 -*-
# @Filename    : user.py
# @Software    : VSCode
# @Datetime    : 2021/11/04 21:25:44
# @Author      : leo liu
# @Version     : 1.0
# @Description :

from datetime import datetime
from typing import Optional
from loguru import logger
from pydantic.types import conint
from sqlalchemy import func, desc
from sqlalchemy.orm import Session
from sqlalchemy.sql.expression import and_, or_, join

from core.security import get_password_hash, verify_password
from extensions.curd_base import CRUDBase
from models import User, Role, LoginHistory

from .role import crud_role
from ..schemas import user_schema

class CRUDUser(CRUDBase[User, user_schema.UserCreate, user_schema.UserUpdate]):
    @staticmethod
    def query_all(db: Session, *, col_val: str = "", is_delete: int = 0, role: str = "", page: int = 1, page_size: conint(le=50) = 20, order_by: str = "id", is_desc: bool = False) -> dict:
        """
        查询用户列表
        :param db:
        :param page:
        :param page_size:
        :return:
        """

        temp_page = (page - 1) * page_size
        search = None
        order_type = None

        # 处理多个字段的模糊查询：username、nickname、email
        if col_val:
            search = and_(User.is_delete == is_delete, or_(User.username.like('%' + col_val + "%"), User.nickname.like('%' + col_val + "%"), User.email.like('%' + col_val + "%")))
        else:
            search = User.is_delete == is_delete

        # 处理角色查询：role
        if role:
            search = and_(search, Role.role_id == role)

        # 处理排序方式
        # 利用属性名反射类属性
        if is_desc:
            order_type = desc(User.getAttrFromName(User, order_by))
        else:
            order_type = User.getAttrFromName(User, order_by)

        # 查询数量
        total = db.query(func.count(User.user_id)).outerjoin(User.roles).filter(search).scalar()

        # 查询结果集
        if page > 0: # page > 0 分页查询
            query_obj = db.query(User).outerjoin(User.roles).filter(search).order_by(order_type).offset(temp_page).limit(page_size).all()
        else: # page <= 0 全量查询
            query_obj = db.query(User).outerjoin(User.roles).filter(search).order_by(order_type).all()

        items = [{
            "id": obj.id,
            "user_id": obj.user_id,
            "username": obj.username,
            "nickname": obj.nickname,
            "email": obj.email,
            "avatar": obj.avatar,
            "phone": obj.phone,
            "gender": obj.gender,
            "register_time": obj.register_time,
            "last_login_time": obj.last_login_time,
            "last_login_ip": obj.last_login_ip,
            "wechat_openid": obj.wechat_openid,
            "country": obj.country,
            "province": obj.province,
            "city": obj.city,
            "roles": obj.roles,
            "is_delete": obj.is_delete,
            "update_time": obj.update_time
        } for obj in query_obj]

        return {
            "items": items,
            "total": total
        }


    @staticmethod
    def get_by_email(db: Session, *, email: str) -> Optional[User]:
        """
        通过email获取用户
        参数里面的* 表示 后面调用的时候 要用指定参数的方法调用
        正确调用方式
            curd_user.get_by_email(db, email="xxx")
        错误调用方式
            curd_user.get_by_email(db, "xxx")
        :param db:
        :param email:
        :return:
        """
        return db.query(User).filter(User.email == email).first()

    @staticmethod
    def get_by_username(db: Session, *, username: str) -> Optional[User]:
        """
        通过用户名获取用户
        """
        return db.query(User).filter(User.username == username).first()

    def authenticate(self, db: Session, *, username: str, password: str, ip: str) -> Optional[User]:
        user = self.get_by_username(db, username=username)
        if not user:
            return None
        if not verify_password(password, user.hashed_password):
            return None

        user.last_login_time = datetime.now()
        user.last_login_ip = ip

        login_history = LoginHistory(
            user_id = user.user_id,
            username = user.username,
            login_time = user.last_login_time,
            login_ipv4 = user.last_login_ip
        )

        user.login_histories.append(login_history)

        db.commit()
        db.refresh(user)

        return user

    def authenticate_by_email(self, db: Session, *, email: str, password: str) -> Optional[User]:
        user = self.get_by_email(db, email=email)
        if not user:
            return None
        if not verify_password(password, user.hashed_password):
            return None
        return user

    @staticmethod
    def is_active(user: User) -> bool:
        return user.is_active == 1

    @staticmethod
    def is_delete(user: User) -> bool:
        return user.is_delete == 1

    @staticmethod
    def is_exist(db: Session, user_in: user_schema.UserCreate) -> bool:
        user = db.query(User).filter(User.username == user_in.username).first()

        if user:
            return True
        else:
            return False

    @staticmethod
    def create(db: Session, *, obj_in: user_schema.UserCreate) -> User:
        db_user = User(
            username=obj_in.username,
            nickname=obj_in.nickname,
            email=obj_in.email,
            hashed_password=get_password_hash(obj_in.password),
            q_limit_day=obj_in.q_limit_day,
            tokens_limit=obj_in.tokens_limit,
            # avatar=obj_in.avatar,
            is_active=obj_in.is_active
        )

        for role_id in obj_in.roles:
            role = crud_role.get_by_id(db, role_id=role_id)

            if (role):
                db_user.roles.append(role)

        db.add(db_user)
        db.commit()
        db.refresh(db_user)
        return db_user

    @staticmethod
    def update(db: Session, *, obj_in: user_schema.UserUpdate) -> User:
        db_obj = db.query(User).filter(User.username == obj_in.username).first()

        if db_obj:
            db_obj.nickname = obj_in.nickname
            db_obj.email = obj_in.email
            db_obj.hashed_password = get_password_hash(obj_in.password) if obj_in.password else db_obj.hashed_password

            if obj_in.q_day_limit >= 0:
                db_obj.q_day_limit = obj_in.q_day_limit

            if obj_in.tokens_limit >= 0:
                db_obj.tokens_limit = obj_in.tokens_limit

            db_obj.roles.clear()

            for role_id in obj_in.roles:
                role = crud_role.get_by_id(db, role_id=role_id)

                if (role):
                    db_obj.roles.append(role)

            db_obj.is_active = obj_in.is_active
            db_obj.is_delete = obj_in.is_delete

        db.commit()
        db.refresh(db_obj)
        return db_obj