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

# !/usr/bin/env python3
# -*- encoding : utf-8 -*-
# @Filename    : menu.py
# @Software    : VSCode
# @Datetime    : 2021/11/08 22:13:05
# @Author      : leo liu
# @Version     : 1.0
# @Description :

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

from extensions.curd_base import CRUDBase
from models import Menu, role_menu
from ..schemas import menu_schema

class CRUDMenu(CRUDBase[Menu, menu_schema.MenuCreate, menu_schema.MenuUpdate]):
    @staticmethod
    def query_all(db: Session, *, col_val: str = "", is_delete: int = 0, page: int = 0, page_size: conint(le=50) = 20, order_by: str = "id", is_desc: bool = False) -> dict:
        temp_page = (page - 1) * page_size
        search = None
        order_type = None

        # 处理多个字段的模糊查询：name
        if col_val:
            search = and_(Menu.is_delete == is_delete, or_(Menu.name.like('%' + col_val + "%"), Menu.path.like('%' + col_val + "%")))
        else:
            search = Menu.is_delete == is_delete

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

        # 查询数量
        total = db.query(func.count(Menu.menu_id)).filter(search).scalar()

        # 查询结果集
        if page > 0:
            query_obj = db.query(Menu).filter(search).order_by(order_type).offset(temp_page).limit(page_size).all()
        else:
            query_obj = db.query(Menu).filter(search).order_by(order_type).all()

        items = [{
            "menu_id": obj.menu_id,
            "path": obj.path,
            "name": obj.name,
            "super_menu": obj.super_menu and obj.super_menu or ""
        } for obj in query_obj]

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

    def create(self, db: Session, *, obj_in: menu_schema.MenuCreate):
        db_obj = Menu(
            path = obj_in.path,
            name = obj_in.name,
            super_menu = obj_in.super_menu
        )

        db.add(db_obj)
        db.commit()
        # db.refresh(db_obj)

    def sync_menus(self, db: Session, *, obj_arr: List):
        for obj_in in obj_arr:
            menu = db.query(Menu).filter(Menu.path == obj_in["path"]).first()

            if (menu):
                menu.name = obj_in["name"]
                menu.path = obj_in["path"]
                menu.super_menu = obj_in["super_menu"]
                menu.is_delete = obj_in["is_delete"]

                if (menu.is_delete):
                    db.query(role_menu).filter(role_menu.c.menu_id == menu.menu_id).delete()
            else:
                db_obj = Menu (
                    path = obj_in["path"],
                    name = obj_in["name"],
                    super_menu = obj_in["super_menu"]
                )

                db.add(db_obj)

        db.commit()

    def get_by_path(self, db: Session, *, path: str) -> Menu:
        return db.query(Menu).filter(Menu.path == path).first()

crud_menu = CRUDMenu(Menu)