zlwuandCursor
5e563a6669
协议样本、生产产出和盘点明细按行更新并保留原 id；补齐存储证明质检预填、离职人员显示名，以及样本采集通知模板字段。 Co-authored-by: Cursor <cursoragent@cursor.com>
272 lines
8.0 KiB
Python
272 lines
8.0 KiB
Python
"""职位 CRUD（sys_post）"""
|
||
from __future__ import annotations
|
||
|
||
import math
|
||
from datetime import datetime
|
||
from typing import Any, Dict, List, Optional, Tuple
|
||
|
||
from sqlalchemy import and_, func, or_
|
||
from sqlalchemy.orm import Session
|
||
|
||
from app.core.utils import generate_uuid
|
||
from app.crud.child_row_sync import sync_link_rows
|
||
from app.models.models import SysPost, SysPostRole, SysRole, SysUser, SysUserPost
|
||
|
||
|
||
def _user_display_name(nick_name: Optional[str], user_no: Optional[str]) -> str:
|
||
n = (nick_name or "").strip()
|
||
if n:
|
||
return n
|
||
return (user_no or "").strip()
|
||
|
||
|
||
def _bound_user_names_by_post_ids(
|
||
db: Session, post_ids: List[str], tenant_no: str
|
||
) -> Dict[str, List[str]]:
|
||
"""批量查询职位关联用户显示名（nick_name 优先，否则 user_no）。"""
|
||
out: Dict[str, List[str]] = {}
|
||
ids = [pid for pid in (post_ids or []) if pid]
|
||
tenant = (tenant_no or "").strip()
|
||
if not ids or not tenant:
|
||
return out
|
||
rows = (
|
||
db.query(SysUserPost.post_id, SysUser.nick_name, SysUser.user_no)
|
||
.join(
|
||
SysUser,
|
||
and_(
|
||
SysUser.id == SysUserPost.user_id,
|
||
SysUser.del_flag == 0,
|
||
),
|
||
)
|
||
.filter(
|
||
and_(
|
||
SysUserPost.post_id.in_(ids),
|
||
SysUserPost.tenant_no == tenant,
|
||
SysUserPost.del_flag == 0,
|
||
)
|
||
)
|
||
.all()
|
||
)
|
||
for post_id, nick_name, user_no in rows:
|
||
name = _user_display_name(nick_name, user_no)
|
||
if not name:
|
||
continue
|
||
out.setdefault(post_id, []).append(name)
|
||
return out
|
||
|
||
|
||
def _row_to_dict(p: SysPost, bound_user_names: Optional[List[str]] = None) -> Dict[str, Any]:
|
||
return {
|
||
"id": p.id,
|
||
"postName": p.post_name,
|
||
"postCode": p.post_code,
|
||
"status": p.status,
|
||
"tenantNo": p.tenant_no,
|
||
"orderNum": p.order_num if p.order_num is not None else 0,
|
||
"remark": p.remark or "",
|
||
"boundUserNames": list(bound_user_names or []),
|
||
"createTime": p.create_time.strftime("%Y-%m-%d %H:%M:%S") if p.create_time else "",
|
||
"updateTime": p.update_time.strftime("%Y-%m-%d %H:%M:%S") if p.update_time else "",
|
||
}
|
||
|
||
|
||
def list_posts(
|
||
db: Session,
|
||
*,
|
||
page: int = 1,
|
||
page_size: int = 10,
|
||
keyword: Optional[str] = None,
|
||
tenant_no: Optional[str] = None,
|
||
user_info=None,
|
||
) -> Tuple[List[Dict[str, Any]], int]:
|
||
def _filtered():
|
||
q0 = db.query(SysPost).filter(SysPost.del_flag == 0)
|
||
if tenant_no:
|
||
q0 = q0.filter(SysPost.tenant_no == tenant_no)
|
||
kw = (keyword or "").strip()
|
||
if kw:
|
||
like = f"%{kw}%"
|
||
q0 = q0.filter(or_(SysPost.post_name.like(like), SysPost.post_code.like(like)))
|
||
if user_info:
|
||
from app.core.data_scope import apply_data_scope
|
||
q0 = apply_data_scope(q0, SysPost, user_info, db, tenant_no)
|
||
return q0
|
||
|
||
total = _filtered().order_by(None).with_entities(func.count(SysPost.id)).scalar() or 0
|
||
page = max(1, page)
|
||
page_size = min(max(1, page_size), 100)
|
||
rows = (
|
||
_filtered()
|
||
.order_by(SysPost.create_time.desc())
|
||
.offset((page - 1) * page_size)
|
||
.limit(page_size)
|
||
.all()
|
||
)
|
||
name_map = _bound_user_names_by_post_ids(db, [x.id for x in rows], tenant_no or "")
|
||
return [
|
||
_row_to_dict(x, bound_user_names=name_map.get(x.id, [])) for x in rows
|
||
], int(total)
|
||
|
||
|
||
def get_post(db: Session, post_id: str) -> Optional[SysPost]:
|
||
return (
|
||
db.query(SysPost)
|
||
.filter(and_(SysPost.id == post_id, SysPost.del_flag == 0))
|
||
.first()
|
||
)
|
||
|
||
|
||
def post_code_exists(db: Session, code: str, tenant_no: str, exclude_id: Optional[str] = None) -> bool:
|
||
q = db.query(SysPost).filter(and_(SysPost.post_code == code.strip(), SysPost.tenant_no == tenant_no, SysPost.del_flag == 0))
|
||
if exclude_id:
|
||
q = q.filter(SysPost.id != exclude_id)
|
||
return q.first() is not None
|
||
|
||
|
||
def create_post(
|
||
db: Session,
|
||
*,
|
||
post_name: str,
|
||
post_code: str,
|
||
status: str = "1",
|
||
tenant_no: str = "zkxw",
|
||
order_num: Optional[int] = None,
|
||
remark: Optional[str] = None,
|
||
user_id: Optional[str] = None,
|
||
) -> SysPost:
|
||
pc = (post_code or "").strip()
|
||
if not pc:
|
||
raise ValueError("职位编码不能为空")
|
||
if post_code_exists(db, pc, tenant_no):
|
||
raise ValueError("职位编码已存在")
|
||
obj = SysPost(
|
||
id=generate_uuid(),
|
||
post_name=(post_name or "").strip() or pc,
|
||
post_code=pc,
|
||
status=status or "1",
|
||
tenant_no=tenant_no,
|
||
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_post(
|
||
db: Session,
|
||
post_id: str,
|
||
*,
|
||
post_name: Optional[str] = None,
|
||
post_code: Optional[str] = None,
|
||
status: Optional[str] = None,
|
||
order_num: Optional[int] = None,
|
||
remark: Optional[str] = None,
|
||
user_id: Optional[str] = None,
|
||
) -> SysPost:
|
||
obj = get_post(db, post_id)
|
||
if not obj:
|
||
raise ValueError("职位不存在")
|
||
if post_code is not None:
|
||
new_code = post_code.strip()
|
||
if not new_code:
|
||
raise ValueError("职位编码不能为空")
|
||
if new_code != obj.post_code and post_code_exists(db, new_code, obj.tenant_no, exclude_id=post_id):
|
||
raise ValueError("职位编码已存在")
|
||
obj.post_code = new_code
|
||
if post_name is not None:
|
||
obj.post_name = (post_name or "").strip() or obj.post_name
|
||
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 soft_delete_post(db: Session, post_id: str) -> None:
|
||
obj = get_post(db, post_id)
|
||
if not obj:
|
||
raise ValueError("职位不存在")
|
||
now = datetime.now()
|
||
obj.del_flag = 1
|
||
obj.update_time = now
|
||
db.add(obj)
|
||
db.query(SysPostRole).filter(
|
||
and_(SysPostRole.post_id == post_id, SysPostRole.del_flag == 0)
|
||
).update({"del_flag": 1, "update_time": now}, synchronize_session=False)
|
||
db.commit()
|
||
|
||
|
||
def list_role_ids_for_post(db: Session, post_id: str, tenant_no: str) -> List[str]:
|
||
rows = (
|
||
db.query(SysPostRole.role_id)
|
||
.filter(
|
||
and_(
|
||
SysPostRole.post_id == post_id,
|
||
SysPostRole.tenant_no == tenant_no,
|
||
SysPostRole.del_flag == 0,
|
||
)
|
||
)
|
||
.all()
|
||
)
|
||
return [r[0] for r in rows]
|
||
|
||
|
||
def sync_post_role_bindings(
|
||
db: Session,
|
||
post_id: str,
|
||
tenant_no: str,
|
||
role_ids: List[str],
|
||
updater_id: Optional[str],
|
||
) -> None:
|
||
"""按角色更新本职位的角色关联。"""
|
||
ordered = list(dict.fromkeys([rid for rid in (role_ids or []) if rid]))
|
||
for rid in ordered:
|
||
role = (
|
||
db.query(SysRole)
|
||
.filter(
|
||
and_(
|
||
SysRole.id == rid,
|
||
SysRole.del_flag == 0,
|
||
SysRole.tenant_no == tenant_no,
|
||
)
|
||
)
|
||
.first()
|
||
)
|
||
if not role:
|
||
raise ValueError("存在无效角色或不属于当前租户")
|
||
|
||
sync_link_rows(
|
||
db,
|
||
SysPostRole,
|
||
parent_attr="post_id",
|
||
parent_id=post_id,
|
||
link_attr="role_id",
|
||
link_ids=ordered,
|
||
tenant_no=tenant_no,
|
||
user_id=updater_id,
|
||
duplicate_message="角色重复",
|
||
foreign_message="角色关联不属于当前职位",
|
||
)
|
||
db.commit()
|
||
|
||
|
||
def total_pages(total: int, page_size: int) -> int:
|
||
if total <= 0:
|
||
return 0
|
||
return int(math.ceil(total / float(page_size)))
|