""" 机构网点CRUD操作 """ from typing import List, Optional, Tuple from sqlalchemy import select, and_, or_, func from sqlalchemy.orm import Session from app.models.organization import Organization from app.schemas.organization import OrganizationCreate, OrganizationUpdate class OrganizationCRUD: """机构网点CRUD操作类""" def get(self, db: Session, id: int) -> Optional[Organization]: """ 根据ID获取机构 Args: db: 数据库会话 id: 机构ID Returns: Organization对象或None """ return db.query(Organization).filter( and_( Organization.id == id, Organization.deleted_at.is_(None) ) ).first() def get_by_code(self, db: Session, code: str) -> Optional[Organization]: """ 根据代码获取机构 Args: db: 数据库会话 code: 机构代码 Returns: Organization对象或None """ return db.query(Organization).filter( and_( Organization.org_code == code, Organization.deleted_at.is_(None) ) ).first() def get_multi( self, db: Session, skip: int = 0, limit: int = 20, org_type: Optional[str] = None, status: Optional[str] = None, keyword: Optional[str] = None ) -> Tuple[List[Organization], int]: """ 获取机构列表 Args: db: 数据库会话 skip: 跳过条数 limit: 返回条数 org_type: 机构类型筛选 status: 状态筛选 keyword: 搜索关键词 Returns: (机构列表, 总数) """ query = db.query(Organization).filter(Organization.deleted_at.is_(None)) # 筛选条件 if org_type: query = query.filter(Organization.org_type == org_type) if status: query = query.filter(Organization.status == status) if keyword: query = query.filter( or_( Organization.org_code.ilike(f"%{keyword}%"), Organization.org_name.ilike(f"%{keyword}%") ) ) # 排序 query = query.order_by(Organization.tree_level.asc(), Organization.sort_order.asc(), Organization.id.asc()) # 总数 total = query.count() # 分页 items = query.offset(skip).limit(limit).all() return items, total def get_tree(self, db: Session, status: Optional[str] = None) -> List[Organization]: """ 获取机构树 Args: db: 数据库会话 status: 状态筛选 Returns: 机构树列表 """ query = db.query(Organization).filter(Organization.deleted_at.is_(None)) if status: query = query.filter(Organization.status == status) # 获取所有机构 all_orgs = query.order_by(Organization.tree_level.asc(), Organization.sort_order.asc()).all() # 构建树形结构 org_map = {org.id: org for org in all_orgs} tree = [] for org in all_orgs: # 清空children列表 org.children = [] if org.parent_id is None: # 根节点 tree.append(org) else: # 添加到父节点的children parent = org_map.get(org.parent_id) if parent: if not hasattr(parent, 'children'): parent.children = [] parent.children.append(org) return tree def get_children(self, db: Session, parent_id: int) -> List[Organization]: """ 获取子机构列表（直接子节点） Args: db: 数据库会话 parent_id: 父机构ID Returns: 子机构列表 """ return db.query(Organization).filter( and_( Organization.parent_id == parent_id, Organization.deleted_at.is_(None) ) ).order_by(Organization.sort_order.asc(), Organization.id.asc()).all() def get_all_children(self, db: Session, parent_id: int) -> List[Organization]: """ 递归获取所有子机构（包括子节点的子节点） Args: db: 数据库会话 parent_id: 父机构ID Returns: 所有子机构列表 """ # 获取父节点的tree_path parent = self.get(db, parent_id) if not parent: return [] # 构建查询路径 if parent.tree_path: search_path = f"{parent.tree_path}{parent.id}/" else: search_path = f"/{parent.id}/" # 查询所有以该路径开头的机构 return db.query(Organization).filter( and_( Organization.tree_path.like(f"{search_path}%"), Organization.deleted_at.is_(None) ) ).order_by(Organization.tree_level.asc(), Organization.sort_order.asc()).all() def get_parents(self, db: Session, child_id: int) -> List[Organization]: """ 递归获取所有父机构（从根到直接父节点） Args: db: 数据库会话 child_id: 子机构ID Returns: 所有父机构列表（从根到父） """ child = self.get(db, child_id) if not child or not child.tree_path: return [] # 解析tree_path，提取所有ID path_ids = [int(id) for id in child.tree_path.split("/") if id] if not path_ids: return [] # 查询所有父机构 return db.query(Organization).filter( and_( Organization.id.in_(path_ids), Organization.deleted_at.is_(None) ) ).order_by(Organization.tree_level.asc()).all() def create( self, db: Session, obj_in: OrganizationCreate, creator_id: Optional[int] = None ) -> Organization: """ 创建机构 Args: db: 数据库会话 obj_in: 创建数据 creator_id: 创建人ID Returns: 创建的Organization对象 """ # 检查代码是否已存在 if self.get_by_code(db, obj_in.org_code): raise ValueError(f"机构代码 '{obj_in.org_code}' 已存在") # 计算tree_path和tree_level tree_path = None tree_level = 0 if obj_in.parent_id: parent = self.get(db, obj_in.parent_id) if not parent: raise ValueError(f"父机构ID {obj_in.parent_id} 不存在") # 构建tree_path if parent.tree_path: tree_path = f"{parent.tree_path}{parent.id}/" else: tree_path = f"/{parent.id}/" tree_level = parent.tree_level + 1 db_obj = Organization( **obj_in.model_dump(), tree_path=tree_path, tree_level=tree_level, created_by=creator_id ) db.add(db_obj) db.commit() db.refresh(db_obj) return db_obj def update( self, db: Session, db_obj: Organization, obj_in: OrganizationUpdate, updater_id: Optional[int] = None ) -> Organization: """ 更新机构 Args: db: 数据库会话 db_obj: 数据库对象 obj_in: 更新数据 updater_id: 更新人ID Returns: 更新后的Organization对象 """ obj_data = obj_in.model_dump(exclude_unset=True) # 如果更新了parent_id，需要重新计算tree_path和tree_level if "parent_id" in obj_data: new_parent_id = obj_data["parent_id"] old_parent_id = db_obj.parent_id if new_parent_id != old_parent_id: # 重新计算当前节点的路径 if new_parent_id: new_parent = self.get(db, new_parent_id) if not new_parent: raise ValueError(f"父机构ID {new_parent_id} 不存在") if new_parent.tree_path: db_obj.tree_path = f"{new_parent.tree_path}{new_parent.id}/" else: db_obj.tree_path = f"/{new_parent.id}/" db_obj.tree_level = new_parent.tree_level + 1 else: # 变为根节点 db_obj.tree_path = None db_obj.tree_level = 0 # TODO: 需要递归更新所有子节点的tree_path和tree_level # 这里需要批量更新，暂时跳过 for field, value in obj_data.items(): if field != "parent_id": # parent_id已经处理 setattr(db_obj, field, value) db_obj.updated_by = updater_id db.add(db_obj) db.commit() db.refresh(db_obj) return db_obj def delete(self, db: Session, id: int, deleter_id: Optional[int] = None) -> bool: """ 删除机构（软删除） Args: db: 数据库会话 id: 机构ID deleter_id: 删除人ID Returns: 是否删除成功 """ obj = self.get(db, id) if not obj: return False # 检查是否有子机构 children = self.get_children(db, id) if children: raise ValueError("该机构下存在子机构，无法删除") obj.deleted_at = func.now() obj.deleted_by = deleter_id db.add(obj) db.commit() return True # 创建全局实例 organization = OrganizationCRUD()