from __future__ import annotations from datetime import datetime from typing import Any, Dict, List, Optional from sqlalchemy import and_ from sqlalchemy.orm import Session from app.core.utils import generate_uuid from app.models.models import SysMenu def _row_to_dict(m: SysMenu) -> Dict[str, Any]: return { "id": m.id, "menuName": m.menu_name, "parentId": m.parent_id, "menuType": m.menu_type or "M", "path": m.path or "", "component": m.component or "", "perms": m.perms or "", "isFrame": m.is_frame if m.is_frame is not None else 1, "visible": m.visible or "1", "status": m.status or "1", "tenantNo": m.tenant_no, "orderNum": m.order_num if m.order_num is not None else 0, "remark": m.remark or "", "createTime": m.create_time.strftime("%Y-%m-%d %H:%M:%S") if m.create_time else "", "updateTime": m.update_time.strftime("%Y-%m-%d %H:%M:%S") if m.update_time else "", } def _build_tree(items: List[Dict[str, Any]], parent_id: str = "0") -> List[Dict[str, Any]]: children = [item for item in items if item.get("parentId") == parent_id] for child in children: child["children"] = _build_tree(items, child["id"]) children.sort(key=lambda x: (x.get("orderNum", 0) or 0)) return children def list_menus( db: Session, *, keyword: Optional[str] = None, tenant_no: Optional[str] = None, user_info=None, ) -> List[Dict[str, Any]]: q = db.query(SysMenu).filter(SysMenu.del_flag == 0) if tenant_no: q = q.filter(SysMenu.tenant_no == tenant_no) kw = (keyword or "").strip() if kw: like = f"%{kw}%" q = q.filter(SysMenu.menu_name.like(like)) if user_info: from app.core.data_scope import apply_data_scope q = apply_data_scope(q, SysMenu, user_info, db, tenant_no) rows = q.order_by(SysMenu.create_time.desc()).all() return [_row_to_dict(x) for x in rows] def list_menus_tree( db: Session, *, keyword: Optional[str] = None, tenant_no: Optional[str] = None, user_info=None, ) -> List[Dict[str, Any]]: kw = (keyword or "").strip() # 关键词检索时：仅返回命中节点会导致其父节点缺失，从而无法组装为以 parentId=0 开始的树。 # 这里改为：命中节点 + 其所有祖先节点，再组树。 if not kw: flat = list_menus(db, keyword=None, tenant_no=tenant_no, user_info=user_info) return _build_tree(flat, "0") all_items = list_menus(db, keyword=None, tenant_no=tenant_no, user_info=user_info) id_map: Dict[str, Dict[str, Any]] = {x.get("id"): x for x in all_items if x.get("id")} like_items = [ x for x in all_items if kw and kw in (x.get("menuName") or "") ] picked: Dict[str, Dict[str, Any]] = {} def pick_with_ancestors(item_id: str) -> None: cur = id_map.get(item_id) while cur and cur.get("id") and cur.get("id") not in picked: picked[cur["id"]] = cur parent_id = (cur.get("parentId") or "0").strip() if not parent_id or parent_id == "0": break cur = id_map.get(parent_id) for it in like_items: if it.get("id"): pick_with_ancestors(it["id"]) return _build_tree(list(picked.values()), "0") def get_menu(db: Session, menu_id: str) -> Optional[SysMenu]: return ( db.query(SysMenu) .filter(and_(SysMenu.id == menu_id, SysMenu.del_flag == 0)) .first() ) def menu_name_exists( db: Session, name: str, tenant_no: str, exclude_id: Optional[str] = None, ) -> bool: q = db.query(SysMenu).filter( and_(SysMenu.menu_name == name, SysMenu.tenant_no == tenant_no, SysMenu.del_flag == 0) ) if exclude_id: q = q.filter(SysMenu.id != exclude_id) return q.first() is not None def create_menu( db: Session, *, menu_name: str, parent_id: str = "0", menu_type: str = "M", path: str = "", component: Optional[str] = None, perms: Optional[str] = None, is_frame: int = 1, visible: str = "1", status: str = "1", tenant_no: Optional[str] = None, order_num: Optional[int] = None, remark: Optional[str] = None, user_id: Optional[str] = None, ) -> SysMenu: name = (menu_name or "").strip() if not name: raise ValueError("菜单名称不能为空") obj = SysMenu( id=generate_uuid(), menu_name=name, parent_id=(parent_id or "0").strip(), menu_type=menu_type or "M", path=(path or "").strip(), component=(component or "").strip() or None, perms=(perms or "").strip() or None, is_frame=is_frame if is_frame is not None else 1, visible=visible or "1", status=status or "1", tenant_no=tenant_no or "zkxw", order_num=order_num if order_num is not None else 0, remark=(remark or "").strip() or None, create_by=user_id, create_dept_id="", create_time=datetime.now(), update_by=user_id, update_time=datetime.now(), del_flag=0, ) db.add(obj) db.commit() db.refresh(obj) return obj def update_menu( db: Session, menu_id: str, *, menu_name: Optional[str] = None, parent_id: Optional[str] = None, menu_type: Optional[str] = None, path: Optional[str] = None, component: Optional[str] = None, perms: Optional[str] = None, is_frame: Optional[int] = None, visible: Optional[str] = None, status: Optional[str] = None, order_num: Optional[int] = None, remark: Optional[str] = None, user_id: Optional[str] = None, ) -> SysMenu: obj = get_menu(db, menu_id) if not obj: raise ValueError("菜单不存在") if menu_name is not None: obj.menu_name = (menu_name or "").strip() or obj.menu_name if parent_id is not None: obj.parent_id = (parent_id or "0").strip() if menu_type is not None: obj.menu_type = menu_type if path is not None: obj.path = (path or "").strip() if component is not None: obj.component = (component or "").strip() or None if perms is not None: obj.perms = (perms or "").strip() or None if is_frame is not None: obj.is_frame = is_frame if visible is not None: obj.visible = visible if status is not None: obj.status = status if order_num is not None: obj.order_num = order_num if remark is not None: obj.remark = (remark or "").strip() or None obj.update_by = user_id obj.update_time = datetime.now() db.add(obj) db.commit() db.refresh(obj) return obj def _cascade_soft_delete(db: Session, menu_id: str) -> None: children = ( db.query(SysMenu) .filter(and_(SysMenu.parent_id == menu_id, SysMenu.del_flag == 0)) .all() ) for child in children: _cascade_soft_delete(db, child.id) child.del_flag = 1 child.update_time = datetime.now() db.add(child) def soft_delete_menu(db: Session, menu_id: str) -> None: obj = get_menu(db, menu_id) if not obj: raise ValueError("菜单不存在") _cascade_soft_delete(db, menu_id) obj.del_flag = 1 obj.update_time = datetime.now() db.add(obj) db.commit()