Build lightweight AI agent admin
This commit is contained in:
Codex
@@ -0,0 +1,461 @@
|
||||
#!/usr/bin/env python
|
||||
# -*- coding: utf-8 -*-
|
||||
"""
|
||||
Post Service - 岗位服务层
|
||||
"""
|
||||
from io import BytesIO
|
||||
from typing import Tuple, Dict, Any, Optional, List
|
||||
|
||||
from sqlalchemy import select, func, or_
|
||||
from sqlalchemy.ext.asyncio import AsyncSession
|
||||
|
||||
from app.base_service import BaseService
|
||||
from core.post.model import Post
|
||||
from core.post.schema import PostCreate, PostUpdate
|
||||
|
||||
|
||||
class PostService(BaseService[Post, PostCreate, PostUpdate]):
|
||||
"""
|
||||
岗位服务层
|
||||
继承BaseService，自动获得增删改查功能
|
||||
"""
|
||||
|
||||
model = Post
|
||||
|
||||
# Excel导入导出配置
|
||||
excel_columns = {
|
||||
"name": "岗位名称",
|
||||
"code": "岗位编码",
|
||||
"post_type": "岗位类型",
|
||||
"post_level": "岗位级别",
|
||||
"status": "状态",
|
||||
"description": "描述",
|
||||
}
|
||||
excel_sheet_name = "岗位列表"
|
||||
|
||||
@classmethod
|
||||
def _export_converter(cls, item: Any) -> Dict[str, Any]:
|
||||
"""导出数据转换器"""
|
||||
return {
|
||||
"name": item.name,
|
||||
"code": item.code,
|
||||
"post_type": item.get_post_type_display(),
|
||||
"post_level": item.get_post_level_display(),
|
||||
"status": "启用" if item.status else "禁用",
|
||||
"description": item.description or "",
|
||||
}
|
||||
|
||||
@classmethod
|
||||
def _import_processor(cls, row: Dict[str, Any]) -> Optional[Post]:
|
||||
"""导入数据处理器"""
|
||||
name = row.get("name")
|
||||
code = row.get("code")
|
||||
if not name or not code:
|
||||
return None
|
||||
|
||||
# 岗位类型映射
|
||||
type_map = {"管理岗": 0, "技术岗": 1, "业务岗": 2, "职能岗": 3, "其他": 4}
|
||||
post_type_str = row.get("post_type", "其他")
|
||||
post_type = type_map.get(post_type_str, 4)
|
||||
|
||||
# 岗位级别映射
|
||||
level_map = {"高层": 0, "中层": 1, "基层": 2, "一般员工": 3}
|
||||
post_level_str = row.get("post_level", "一般员工")
|
||||
post_level = level_map.get(post_level_str, 3)
|
||||
|
||||
status_str = row.get("status", "启用")
|
||||
status = status_str in ("启用", "true", "True", "1", True)
|
||||
|
||||
return Post(
|
||||
name=str(name),
|
||||
code=str(code),
|
||||
post_type=post_type,
|
||||
post_level=post_level,
|
||||
status=status,
|
||||
description=str(row.get("description") or "") or None,
|
||||
)
|
||||
|
||||
@classmethod
|
||||
async def export_to_excel(
|
||||
cls,
|
||||
db: AsyncSession,
|
||||
data_converter: Any = None
|
||||
) -> BytesIO:
|
||||
"""导出到Excel"""
|
||||
return await super().export_to_excel(db, cls._export_converter)
|
||||
|
||||
@classmethod
|
||||
async def import_from_excel(
|
||||
cls,
|
||||
db: AsyncSession,
|
||||
file_content: bytes,
|
||||
row_processor: Any = None
|
||||
) -> Tuple[int, int]:
|
||||
"""从Excel导入"""
|
||||
return await super().import_from_excel(db, file_content, cls._import_processor)
|
||||
|
||||
@classmethod
|
||||
async def get_user_count(cls, db: AsyncSession, post_id: str) -> int:
|
||||
"""获取岗位下的用户数量"""
|
||||
from core.user.model import User
|
||||
result = await db.execute(
|
||||
select(func.count(User.id)).where(
|
||||
User.post_id == post_id,
|
||||
User.is_deleted == False # noqa: E712
|
||||
)
|
||||
)
|
||||
return result.scalar() or 0
|
||||
|
||||
@classmethod
|
||||
async def can_delete(cls, db: AsyncSession, post_id: str) -> Tuple[bool, str]:
|
||||
"""
|
||||
检查岗位是否可以删除
|
||||
|
||||
:return: (是否可删除, 原因)
|
||||
"""
|
||||
user_count = await cls.get_user_count(db, post_id)
|
||||
if user_count > 0:
|
||||
return False, f"该岗位下还有 {user_count} 个用户，无法删除"
|
||||
return True, ""
|
||||
|
||||
@classmethod
|
||||
async def batch_update_status(
|
||||
cls,
|
||||
db: AsyncSession,
|
||||
ids: List[str],
|
||||
status: bool
|
||||
) -> int:
|
||||
"""
|
||||
批量更新岗位状态
|
||||
|
||||
:return: 更新的记录数
|
||||
"""
|
||||
count = 0
|
||||
for post_id in ids:
|
||||
post = await cls.get_by_id(db, post_id)
|
||||
if post:
|
||||
post.status = status
|
||||
count += 1
|
||||
|
||||
if count > 0:
|
||||
await db.commit()
|
||||
|
||||
return count
|
||||
|
||||
@classmethod
|
||||
async def batch_delete(
|
||||
cls,
|
||||
db: AsyncSession,
|
||||
ids: List[str],
|
||||
hard: bool = False
|
||||
) -> Tuple[int, List[str]]:
|
||||
"""
|
||||
批量删除岗位
|
||||
|
||||
:return: (删除成功数, 删除失败的ID列表)
|
||||
"""
|
||||
success_count = 0
|
||||
failed_ids = []
|
||||
|
||||
for post_id in ids:
|
||||
can_del, reason = await cls.can_delete(db, post_id)
|
||||
if can_del:
|
||||
if await cls.delete(db, post_id, hard=hard):
|
||||
success_count += 1
|
||||
else:
|
||||
failed_ids.append(post_id)
|
||||
else:
|
||||
failed_ids.append(post_id)
|
||||
|
||||
return success_count, failed_ids
|
||||
|
||||
@classmethod
|
||||
async def search(
|
||||
cls,
|
||||
db: AsyncSession,
|
||||
keyword: str,
|
||||
page: int = 1,
|
||||
page_size: int = 20
|
||||
) -> Tuple[List[Post], int]:
|
||||
"""
|
||||
搜索岗位（模糊匹配名称、编码、描述）
|
||||
"""
|
||||
if not keyword:
|
||||
return [], 0
|
||||
|
||||
# 构建搜索条件
|
||||
search_filter = or_(
|
||||
Post.name.ilike(f"%{keyword}%"),
|
||||
Post.code.ilike(f"%{keyword}%"),
|
||||
Post.description.ilike(f"%{keyword}%")
|
||||
)
|
||||
|
||||
# 查询总数
|
||||
count_result = await db.execute(
|
||||
select(func.count(Post.id)).where(
|
||||
search_filter,
|
||||
Post.is_deleted == False # noqa: E712
|
||||
)
|
||||
)
|
||||
total = count_result.scalar() or 0
|
||||
|
||||
# 查询数据
|
||||
result = await db.execute(
|
||||
select(Post).where(
|
||||
search_filter,
|
||||
Post.is_deleted == False # noqa: E712
|
||||
)
|
||||
.order_by(Post.sys_create_datetime.desc())
|
||||
.offset((page - 1) * page_size)
|
||||
.limit(page_size)
|
||||
)
|
||||
items = list(result.scalars().all())
|
||||
|
||||
return items, total
|
||||
|
||||
@classmethod
|
||||
async def search_with_data_scope(
|
||||
cls,
|
||||
db: AsyncSession,
|
||||
keyword: str,
|
||||
page: int,
|
||||
page_size: int
|
||||
) -> Tuple[List[Post], int]:
|
||||
"""
|
||||
搜索岗位（模糊匹配名称、编码、描述，自动应用数据权限过滤）
|
||||
|
||||
自动从上下文获取当前用户信息和请求信息，无需手动传递任何参数
|
||||
"""
|
||||
if not keyword:
|
||||
return [], 0
|
||||
|
||||
from utils.permission import apply_data_scope_filter
|
||||
from utils.context import get_current_user_info_from_context
|
||||
|
||||
# 从上下文获取用户信息和请求信息
|
||||
user_info = get_current_user_info_from_context()
|
||||
if not user_info:
|
||||
return [], 0
|
||||
|
||||
# 获取数据权限过滤条件
|
||||
data_scope_filter = await apply_data_scope_filter(
|
||||
db=db,
|
||||
role_id=user_info.get("role_id"),
|
||||
is_superuser=user_info.get("is_superuser", False),
|
||||
user_id=user_info.get("user_id"),
|
||||
user_dept_id=user_info.get("dept_id"),
|
||||
request_path=user_info.get("request_path"),
|
||||
http_method=user_info.get("http_method")
|
||||
)
|
||||
|
||||
# 构建搜索条件
|
||||
search_filter = or_(
|
||||
Post.name.ilike(f"%{keyword}%"),
|
||||
Post.code.ilike(f"%{keyword}%"),
|
||||
Post.description.ilike(f"%{keyword}%")
|
||||
)
|
||||
|
||||
# 构建基础查询
|
||||
base_query = select(Post).where(
|
||||
search_filter,
|
||||
Post.is_deleted == False # noqa: E712
|
||||
)
|
||||
|
||||
# 应用数据权限过滤
|
||||
base_query = cls._apply_data_scope_to_query(
|
||||
query=base_query,
|
||||
data_scope_filter=data_scope_filter
|
||||
)
|
||||
|
||||
# 查询总数
|
||||
count_result = await db.execute(
|
||||
select(func.count()).select_from(base_query.subquery())
|
||||
)
|
||||
total = count_result.scalar() or 0
|
||||
|
||||
# 查询数据
|
||||
result = await db.execute(
|
||||
base_query.order_by(Post.sys_create_datetime.desc())
|
||||
.offset((page - 1) * page_size)
|
||||
.limit(page_size)
|
||||
)
|
||||
items = list(result.scalars().all())
|
||||
|
||||
return items, total
|
||||
|
||||
@classmethod
|
||||
async def get_stats(cls, db: AsyncSession) -> Dict[str, Any]:
|
||||
"""获取岗位统计信息"""
|
||||
# 总数
|
||||
total_result = await db.execute(
|
||||
select(func.count(Post.id)).where(Post.is_deleted == False) # noqa: E712
|
||||
)
|
||||
total_count = total_result.scalar() or 0
|
||||
|
||||
# 启用数
|
||||
active_result = await db.execute(
|
||||
select(func.count(Post.id)).where(
|
||||
Post.status == True, # noqa: E712
|
||||
Post.is_deleted == False # noqa: E712
|
||||
)
|
||||
)
|
||||
active_count = active_result.scalar() or 0
|
||||
|
||||
# 按类型统计
|
||||
type_stats = {}
|
||||
for type_code, type_name in Post.POST_TYPE_CHOICES.items():
|
||||
count_result = await db.execute(
|
||||
select(func.count(Post.id)).where(
|
||||
Post.post_type == type_code,
|
||||
Post.is_deleted == False # noqa: E712
|
||||
)
|
||||
)
|
||||
type_stats[type_name] = count_result.scalar() or 0
|
||||
|
||||
# 按级别统计
|
||||
level_stats = {}
|
||||
for level_code, level_name in Post.POST_LEVEL_CHOICES.items():
|
||||
count_result = await db.execute(
|
||||
select(func.count(Post.id)).where(
|
||||
Post.post_level == level_code,
|
||||
Post.is_deleted == False # noqa: E712
|
||||
)
|
||||
)
|
||||
level_stats[level_name] = count_result.scalar() or 0
|
||||
|
||||
return {
|
||||
'total_count': total_count,
|
||||
'active_count': active_count,
|
||||
'inactive_count': total_count - active_count,
|
||||
'type_stats': type_stats,
|
||||
'level_stats': level_stats,
|
||||
}
|
||||
|
||||
@classmethod
|
||||
async def get_by_dept(cls, db: AsyncSession, dept_id: str) -> List[Post]:
|
||||
"""根据部门ID获取岗位列表"""
|
||||
result = await db.execute(
|
||||
select(Post).where(
|
||||
Post.dept_id == dept_id,
|
||||
Post.status == True, # noqa: E712
|
||||
Post.is_deleted == False # noqa: E712
|
||||
).order_by(Post.post_level, Post.name)
|
||||
)
|
||||
return list(result.scalars().all())
|
||||
|
||||
@classmethod
|
||||
async def get_by_type(cls, db: AsyncSession, post_type: int) -> List[Post]:
|
||||
"""根据岗位类型获取岗位列表"""
|
||||
result = await db.execute(
|
||||
select(Post).where(
|
||||
Post.post_type == post_type,
|
||||
Post.status == True, # noqa: E712
|
||||
Post.is_deleted == False # noqa: E712
|
||||
).order_by(Post.post_level, Post.name)
|
||||
)
|
||||
return list(result.scalars().all())
|
||||
|
||||
@classmethod
|
||||
async def get_by_level(cls, db: AsyncSession, post_level: int) -> List[Post]:
|
||||
"""根据岗位级别获取岗位列表"""
|
||||
result = await db.execute(
|
||||
select(Post).where(
|
||||
Post.post_level == post_level,
|
||||
Post.status == True, # noqa: E712
|
||||
Post.is_deleted == False # noqa: E712
|
||||
).order_by(Post.name)
|
||||
)
|
||||
return list(result.scalars().all())
|
||||
|
||||
@classmethod
|
||||
async def get_by_ids(cls, db: AsyncSession, ids: List[str]) -> List[Post]:
|
||||
"""根据ID列表批量获取岗位"""
|
||||
if not ids:
|
||||
return []
|
||||
|
||||
result = await db.execute(
|
||||
select(Post).where(
|
||||
Post.id.in_(ids),
|
||||
Post.is_deleted == False # noqa: E712
|
||||
)
|
||||
)
|
||||
return list(result.scalars().all())
|
||||
|
||||
@classmethod
|
||||
async def get_post_users(cls, db: AsyncSession, post_id: str) -> List[Any]:
|
||||
"""获取岗位下的用户列表"""
|
||||
from core.user.model import User
|
||||
|
||||
result = await db.execute(
|
||||
select(User).where(
|
||||
User.post_id == post_id,
|
||||
User.user_status == 1,
|
||||
User.is_deleted == False # noqa: E712
|
||||
)
|
||||
)
|
||||
return list(result.scalars().all())
|
||||
|
||||
@classmethod
|
||||
async def add_users_to_post(
|
||||
cls,
|
||||
db: AsyncSession,
|
||||
post_id: str,
|
||||
user_ids: List[str]
|
||||
) -> int:
|
||||
"""将用户添加到岗位"""
|
||||
from core.user.model import User
|
||||
|
||||
post = await cls.get_by_id(db, post_id)
|
||||
if not post:
|
||||
return 0
|
||||
|
||||
added_count = 0
|
||||
for user_id in user_ids:
|
||||
result = await db.execute(
|
||||
select(User).where(User.id == user_id)
|
||||
)
|
||||
user = result.scalar_one_or_none()
|
||||
if user and user.post_id != post_id:
|
||||
user.post_id = post_id
|
||||
added_count += 1
|
||||
|
||||
if added_count > 0:
|
||||
await db.commit()
|
||||
|
||||
return added_count
|
||||
|
||||
@classmethod
|
||||
async def remove_users_from_post(
|
||||
cls,
|
||||
db: AsyncSession,
|
||||
post_id: str,
|
||||
user_ids: List[str]
|
||||
) -> int:
|
||||
"""从岗位中移除用户"""
|
||||
from core.user.model import User
|
||||
|
||||
removed_count = 0
|
||||
for user_id in user_ids:
|
||||
result = await db.execute(
|
||||
select(User).where(User.id == user_id)
|
||||
)
|
||||
user = result.scalar_one_or_none()
|
||||
if user and user.post_id == post_id:
|
||||
user.post_id = None
|
||||
removed_count += 1
|
||||
|
||||
if removed_count > 0:
|
||||
await db.commit()
|
||||
|
||||
return removed_count
|
||||
|
||||
@classmethod
|
||||
async def get_all_simple(cls, db: AsyncSession) -> List[Post]:
|
||||
"""获取所有启用的岗位（简化版，用于选择器）"""
|
||||
result = await db.execute(
|
||||
select(Post).where(
|
||||
Post.status == True, # noqa: E712
|
||||
Post.is_deleted == False # noqa: E712
|
||||
).order_by(Post.post_level, Post.name)
|
||||
)
|
||||
return list(result.scalars().all())
|
||||
Reference in New Issue
Block a user