admin.py 49 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432
  1. """运营端管理接口(租户、角色下拉等)"""
  2. import os
  3. import uuid
  4. from typing import Any, List, Optional
  5. from fastapi import APIRouter, Depends, HTTPException, Query, UploadFile, File, Form
  6. from pydantic import BaseModel, ConfigDict, Field
  7. from sqlalchemy import and_, or_
  8. from sqlalchemy.orm import Session
  9. from app.core.config import settings
  10. from app.core.data_scope import apply_data_scope, assert_row_in_data_scope
  11. from app.core.database import get_db
  12. from app.core.menu_guard import require_admin_or_super_admin, require_menu_code
  13. from app.core.response import success_response, error_response
  14. from app.core.attachment_biz import resolve_upload_folder
  15. from app.core.storage import storage
  16. from app.crud import tenant as tenant_svc
  17. from app.crud import dict_type as dict_type_svc
  18. from app.crud import dict_data as dict_data_svc
  19. from app.crud import dept as dept_svc
  20. from app.crud import site as site_svc
  21. from app.crud import post as post_svc
  22. from app.crud import menu as menu_svc
  23. from app.crud import role as role_svc
  24. from app.crud import user as user_svc
  25. from app.crud import role as role_svc
  26. from app.crud import user as user_svc
  27. from app.crud import oss as oss_svc
  28. from app.models.models import SysDept, SysDictData, SysRole, SysUser
  29. from app.services.web_auth_roles import _map_role_name_to_web
  30. router = APIRouter(prefix="/web/admin", tags=["运营端管理"])
  31. # ---------------------------------------------------------------------------
  32. # 角色下拉(用户弹窗等)
  33. # ---------------------------------------------------------------------------
  34. @router.get("/roles")
  35. def list_roles_for_select(
  36. _user: dict = Depends(require_admin_or_super_admin),
  37. db: Session = Depends(get_db),
  38. ):
  39. """返回运营端角色下拉选项(label=库中角色名,value=Web 角色码)。"""
  40. rows: List[SysRole] = (
  41. db.query(SysRole)
  42. .filter(SysRole.del_flag == 0, SysRole.status == "1")
  43. .order_by(SysRole.create_time.desc())
  44. .all()
  45. )
  46. seen: set[str] = set()
  47. out: List[dict] = []
  48. for r in rows:
  49. name = (r.role_name or "").strip()
  50. if not name:
  51. continue
  52. code = _map_role_name_to_web(name)
  53. key = f"{code}:{name}"
  54. if key in seen:
  55. continue
  56. seen.add(key)
  57. out.append({"label": name, "value": code})
  58. if not out:
  59. out = [
  60. {"label": "超级管理员", "value": "super_admin"},
  61. {"label": "运营管理员", "value": "admin"},
  62. ]
  63. return success_response(out)
  64. @router.get("/users/options")
  65. def list_users_for_select(
  66. user: dict = Depends(require_admin_or_super_admin),
  67. db: Session = Depends(get_db),
  68. ):
  69. """返回用户下拉选项(label=昵称,value=用户ID,附带 nickName/userNo)。
  70. 供申请人/审批人等选择场景使用,返回本租户全部有效用户,
  71. 不做业务数据范围过滤(否则「仅本人」角色会按 create_by 滤空,连自己都选不到)。
  72. """
  73. tenant_no = _get_tenant_no(db, user)
  74. q = db.query(SysUser).filter(SysUser.del_flag == 0, SysUser.status == "1")
  75. q = q.filter(SysUser.tenant_no == tenant_no)
  76. rows = q.order_by(SysUser.create_time.desc()).all()
  77. out = [
  78. {
  79. "label": (r.nick_name or r.user_no or ""),
  80. "value": r.id,
  81. "nickName": r.nick_name or "",
  82. "userNo": r.user_no or "",
  83. }
  84. for r in rows
  85. ]
  86. return success_response(out)
  87. @router.get("/roles/options")
  88. def list_roles_for_designer(
  89. user: dict = Depends(require_admin_or_super_admin),
  90. db: Session = Depends(get_db),
  91. ):
  92. """返回角色下拉选项(label=角色名,value=角色ID UUID),供流程设计器审批人选择使用。
  93. 与 /roles 不同,此端点返回角色 UUID 作为 value(引擎解析审批人需要 UUID)。
  94. 角色为租户级主数据选项,不做 create_by 数据范围过滤。
  95. """
  96. tenant_no = _get_tenant_no(db, user)
  97. q = db.query(SysRole).filter(SysRole.del_flag == 0, SysRole.status == "1")
  98. q = q.filter(SysRole.tenant_no == tenant_no)
  99. rows = q.order_by(SysRole.create_time.desc()).all()
  100. out = [{"label": (r.role_name or "").strip(), "value": r.id} for r in rows if (r.role_name or "").strip()]
  101. return success_response(out)
  102. # ---------------------------------------------------------------------------
  103. # 租户管理
  104. # ---------------------------------------------------------------------------
  105. class TenantCreateBody(BaseModel):
  106. model_config = ConfigDict(populate_by_name=True)
  107. tenant_no: str = Field(..., alias="tenantNo")
  108. tenant_name: str = Field(..., alias="tenantName")
  109. contact_name: Optional[str] = Field(None, alias="contactName")
  110. contact_phone: Optional[str] = Field(None, alias="contactPhone")
  111. order_num: Optional[int] = Field(0, alias="orderNum")
  112. remark: Optional[str] = None
  113. class TenantUpdateBody(BaseModel):
  114. model_config = ConfigDict(populate_by_name=True)
  115. tenant_no: Optional[str] = Field(None, alias="tenantNo")
  116. tenant_name: Optional[str] = Field(None, alias="tenantName")
  117. contact_name: Optional[str] = Field(None, alias="contactName")
  118. contact_phone: Optional[str] = Field(None, alias="contactPhone")
  119. order_num: Optional[int] = Field(None, alias="orderNum")
  120. remark: Optional[str] = None
  121. @router.get("/tenants")
  122. def list_tenants(
  123. page: int = Query(1, ge=1),
  124. pageSize: int = Query(10, ge=1, le=100, alias="pageSize"),
  125. keyword: Optional[str] = None,
  126. _user: dict = Depends(require_admin_or_super_admin),
  127. db: Session = Depends(get_db),
  128. ):
  129. items, total = tenant_svc.list_tenants(db, page=page, page_size=pageSize, keyword=keyword, user_info=_user)
  130. total_pages = tenant_svc.total_pages(total, pageSize)
  131. return success_response(
  132. {
  133. "list": items,
  134. "total": total,
  135. "page": page,
  136. "pageSize": pageSize,
  137. "totalPages": total_pages,
  138. }
  139. )
  140. @router.post("/tenants")
  141. def create_tenant(
  142. body: TenantCreateBody,
  143. user: dict = Depends(require_admin_or_super_admin),
  144. db: Session = Depends(get_db),
  145. ):
  146. try:
  147. obj = tenant_svc.create_tenant(
  148. db,
  149. tenant_no=body.tenant_no,
  150. tenant_name=body.tenant_name,
  151. contact_name=body.contact_name,
  152. contact_phone=body.contact_phone,
  153. order_num=body.order_num,
  154. remark=body.remark,
  155. user_id=user.get("user_id"),
  156. )
  157. return success_response(tenant_svc._row_to_dict(obj), message="创建成功")
  158. except ValueError as e:
  159. return error_response(400, str(e))
  160. @router.put("/tenants/{tenant_id}")
  161. def update_tenant(
  162. tenant_id: str,
  163. body: TenantUpdateBody,
  164. user: dict = Depends(require_admin_or_super_admin),
  165. db: Session = Depends(get_db),
  166. ):
  167. try:
  168. payload: dict[str, Any] = body.model_dump(exclude_unset=True, by_alias=False)
  169. obj = tenant_svc.update_tenant(
  170. db,
  171. tenant_id,
  172. tenant_no=payload.get("tenant_no"),
  173. tenant_name=payload.get("tenant_name"),
  174. contact_name=payload.get("contact_name"),
  175. contact_phone=payload.get("contact_phone"),
  176. order_num=payload.get("order_num"),
  177. remark=payload.get("remark"),
  178. user_id=user.get("user_id"),
  179. )
  180. return success_response(tenant_svc._row_to_dict(obj), message="更新成功")
  181. except ValueError as e:
  182. return error_response(400, str(e))
  183. @router.delete("/tenants/{tenant_id}")
  184. def delete_tenant(
  185. tenant_id: str,
  186. _user: dict = Depends(require_admin_or_super_admin),
  187. db: Session = Depends(get_db),
  188. ):
  189. try:
  190. tenant_svc.soft_delete_tenant(db, tenant_id, user_info=_user)
  191. return success_response(None, message="删除成功")
  192. except ValueError as e:
  193. return error_response(400, str(e))
  194. # ---------------------------------------------------------------------------
  195. # Helper: resolve tenant_no from user context
  196. # ---------------------------------------------------------------------------
  197. def _get_tenant_no(db: Session, user: dict) -> str:
  198. user_id = user.get("user_id")
  199. if not user_id:
  200. return "shanxia2.0"
  201. sys_user = db.query(SysUser).filter(SysUser.id == user_id).first()
  202. return (sys_user.tenant_no if sys_user else None) or "shanxia2.0"
  203. # ---------------------------------------------------------------------------
  204. # 用户管理
  205. # ---------------------------------------------------------------------------
  206. class UserCreateBody(BaseModel):
  207. model_config = ConfigDict(populate_by_name=True)
  208. user_no: str = Field(..., alias="userNo")
  209. nick_name: str = Field(..., alias="nickName")
  210. password: str = Field(..., alias="password")
  211. dept_id: Optional[str] = Field(None, alias="deptId")
  212. email: Optional[str] = Field("", alias="email")
  213. phone_no: Optional[str] = Field("", alias="phoneNo")
  214. sex: Optional[str] = Field("0", alias="sex")
  215. avatar: Optional[str] = Field(None, alias="avatar")
  216. status: Optional[str] = Field("1", alias="status")
  217. order_num: Optional[int] = Field(None, alias="orderNum")
  218. remark: Optional[str] = None
  219. role_ids: Optional[List[str]] = Field(None, alias="roleIds")
  220. post_ids: Optional[List[str]] = Field(None, alias="postIds")
  221. class UserUpdateBody(BaseModel):
  222. model_config = ConfigDict(populate_by_name=True)
  223. nick_name: Optional[str] = Field(None, alias="nickName")
  224. dept_id: Optional[str] = Field(None, alias="deptId")
  225. email: Optional[str] = Field(None, alias="email")
  226. phone_no: Optional[str] = Field(None, alias="phoneNo")
  227. sex: Optional[str] = Field(None, alias="sex")
  228. avatar: Optional[str] = Field(None, alias="avatar")
  229. status: Optional[str] = Field(None, alias="status")
  230. order_num: Optional[int] = Field(None, alias="orderNum")
  231. remark: Optional[str] = None
  232. role_ids: Optional[List[str]] = Field(None, alias="roleIds")
  233. post_ids: Optional[List[str]] = Field(None, alias="postIds")
  234. @router.get("/users")
  235. def list_users(
  236. page: int = Query(1, ge=1),
  237. pageSize: int = Query(10, ge=1, le=100, alias="pageSize"),
  238. keyword: Optional[str] = None,
  239. deptId: Optional[str] = Query(None, alias="deptId"),
  240. status: Optional[str] = None,
  241. user: dict = Depends(require_menu_code("system:user:list")),
  242. db: Session = Depends(get_db),
  243. ):
  244. tenant_no = _get_tenant_no(db, user)
  245. query = db.query(SysUser).filter(
  246. SysUser.del_flag == 0,
  247. SysUser.tenant_no == tenant_no,
  248. )
  249. kw = (keyword or "").strip()
  250. if kw:
  251. like = f"%{kw}%"
  252. query = query.filter(
  253. or_(
  254. SysUser.user_no.like(like),
  255. SysUser.nick_name.like(like),
  256. SysUser.phone_no.like(like),
  257. )
  258. )
  259. if deptId:
  260. query = query.filter(SysUser.dept_id == deptId)
  261. if status:
  262. query = query.filter(SysUser.status == status)
  263. query = apply_data_scope(query, SysUser, user, db, tenant_no)
  264. total = query.count()
  265. items = query.order_by(SysUser.create_time.desc()).offset((page - 1) * pageSize).limit(pageSize).all()
  266. dept_ids = list({r.dept_id for r in items if r.dept_id})
  267. dept_map: dict[str, str] = {}
  268. if dept_ids:
  269. for d in db.query(SysDept).filter(SysDept.id.in_(dept_ids), SysDept.del_flag == 0).all():
  270. dept_map[d.id] = d.dept_name or ""
  271. result_list = [
  272. user_svc._row_to_dict(r, dept_name=dept_map.get(r.dept_id or "", "")) for r in items
  273. ]
  274. tp = user_svc.total_pages(total, pageSize)
  275. return success_response({"list": result_list, "total": total, "page": page, "pageSize": pageSize, "totalPages": tp})
  276. @router.get("/users/{user_id}")
  277. def get_user_by_id(
  278. user_id: str,
  279. user: dict = Depends(require_menu_code("system:user:list")),
  280. db: Session = Depends(get_db),
  281. ):
  282. tenant_no = _get_tenant_no(db, user)
  283. row = user_svc.get_user(db, user_id)
  284. if not row:
  285. return error_response(404, "用户不存在")
  286. if row.tenant_no != tenant_no and user.get("role") != "super_admin":
  287. return error_response(403, "无权查看该用户")
  288. try:
  289. assert_row_in_data_scope(db, row, SysUser, user, tenant_no)
  290. except HTTPException:
  291. return error_response(403, "无权查看该用户")
  292. data = user_svc.get_user_with_relations(db, user_id)
  293. return success_response(data)
  294. @router.post("/users")
  295. def create_user(
  296. body: UserCreateBody,
  297. user: dict = Depends(require_admin_or_super_admin),
  298. db: Session = Depends(get_db),
  299. ):
  300. try:
  301. tenant_no = _get_tenant_no(db, user)
  302. if user_svc.user_no_exists(db, body.user_no, tenant_no):
  303. return error_response(400, "用户账号已存在")
  304. obj = user_svc.create_user(
  305. db,
  306. user_no=body.user_no,
  307. nick_name=body.nick_name,
  308. password=body.password,
  309. dept_id=body.dept_id,
  310. email=body.email,
  311. phone_no=body.phone_no,
  312. sex=body.sex,
  313. avatar=body.avatar,
  314. status=body.status,
  315. tenant_no=tenant_no,
  316. order_num=body.order_num,
  317. remark=body.remark,
  318. user_id=user.get("user_id"),
  319. role_ids=body.role_ids,
  320. post_ids=body.post_ids,
  321. )
  322. return success_response(user_svc._row_to_dict(obj), message="创建成功")
  323. except ValueError as e:
  324. return error_response(400, str(e))
  325. @router.put("/users/{user_id}")
  326. def update_user(
  327. user_id: str,
  328. body: UserUpdateBody,
  329. user: dict = Depends(require_admin_or_super_admin),
  330. db: Session = Depends(get_db),
  331. ):
  332. try:
  333. payload: dict[str, Any] = body.model_dump(exclude_unset=True, by_alias=False)
  334. obj = user_svc.update_user(
  335. db,
  336. user_id,
  337. nick_name=payload.get("nick_name"),
  338. dept_id=payload.get("dept_id"),
  339. email=payload.get("email"),
  340. phone_no=payload.get("phone_no"),
  341. sex=payload.get("sex"),
  342. avatar=payload.get("avatar"),
  343. status=payload.get("status"),
  344. order_num=payload.get("order_num"),
  345. remark=payload.get("remark"),
  346. user_id_updater=user.get("user_id"),
  347. role_ids=payload.get("role_ids"),
  348. post_ids=payload.get("post_ids"),
  349. )
  350. return success_response(user_svc._row_to_dict(obj), message="更新成功")
  351. except ValueError as e:
  352. return error_response(400, str(e))
  353. @router.delete("/users/{user_id}")
  354. def delete_user(
  355. user_id: str,
  356. _user: dict = Depends(require_admin_or_super_admin),
  357. db: Session = Depends(get_db),
  358. ):
  359. try:
  360. user_svc.soft_delete_user(db, user_id, user_info=_user)
  361. return success_response(None, message="删除成功")
  362. except ValueError as e:
  363. return error_response(400, str(e))
  364. @router.post("/users/{user_id}/reset-password")
  365. def reset_user_password(
  366. user_id: str,
  367. _user: dict = Depends(require_admin_or_super_admin),
  368. db: Session = Depends(get_db),
  369. ):
  370. try:
  371. user_svc.reset_password(db, user_id, "111111")
  372. return success_response(None, message="密码已重置为 111111")
  373. except ValueError as e:
  374. return error_response(400, str(e))
  375. # ---------------------------------------------------------------------------
  376. # 字典类型管理
  377. # ---------------------------------------------------------------------------
  378. class DictTypeCreateBody(BaseModel):
  379. model_config = ConfigDict(populate_by_name=True)
  380. dict_name: str = Field(..., alias="dictName")
  381. dict_type: str = Field(..., alias="dictType")
  382. order_num: Optional[int] = Field(0, alias="orderNum")
  383. remark: Optional[str] = None
  384. class DictTypeUpdateBody(BaseModel):
  385. model_config = ConfigDict(populate_by_name=True)
  386. dict_name: Optional[str] = Field(None, alias="dictName")
  387. dict_type: Optional[str] = Field(None, alias="dictType")
  388. order_num: Optional[int] = Field(None, alias="orderNum")
  389. remark: Optional[str] = None
  390. @router.get("/dict-types")
  391. def list_dict_types(
  392. page: int = Query(1, ge=1),
  393. pageSize: int = Query(10, ge=1, le=100, alias="pageSize"),
  394. keyword: Optional[str] = None,
  395. user: dict = Depends(require_menu_code("system:dict:list")),
  396. db: Session = Depends(get_db),
  397. ):
  398. tenant_no = _get_tenant_no(db, user)
  399. items, total = dict_type_svc.list_dict_types(
  400. db, page=page, page_size=pageSize, keyword=keyword, tenant_no=tenant_no, user_info=user
  401. )
  402. tp = dict_type_svc.total_pages(total, pageSize)
  403. return success_response({"list": items, "total": total, "page": page, "pageSize": pageSize, "totalPages": tp})
  404. @router.post("/dict-types")
  405. def create_dict_type(
  406. body: DictTypeCreateBody,
  407. user: dict = Depends(require_admin_or_super_admin),
  408. db: Session = Depends(get_db),
  409. ):
  410. try:
  411. tenant_no = _get_tenant_no(db, user)
  412. obj = dict_type_svc.create_dict_type(
  413. db,
  414. dict_name=body.dict_name,
  415. dict_type=body.dict_type,
  416. tenant_no=tenant_no,
  417. order_num=body.order_num,
  418. remark=body.remark,
  419. user_id=user.get("user_id"),
  420. )
  421. return success_response(dict_type_svc._row_to_dict(obj), message="创建成功")
  422. except ValueError as e:
  423. return error_response(400, str(e))
  424. @router.put("/dict-types/{dict_type_id}")
  425. def update_dict_type(
  426. dict_type_id: str,
  427. body: DictTypeUpdateBody,
  428. user: dict = Depends(require_admin_or_super_admin),
  429. db: Session = Depends(get_db),
  430. ):
  431. try:
  432. tenant_no = _get_tenant_no(db, user)
  433. payload: dict[str, Any] = body.model_dump(exclude_unset=True, by_alias=False)
  434. obj = dict_type_svc.update_dict_type(
  435. db,
  436. dict_type_id,
  437. dict_name=payload.get("dict_name"),
  438. dict_type=payload.get("dict_type"),
  439. order_num=payload.get("order_num"),
  440. remark=payload.get("remark"),
  441. user_id=user.get("user_id"),
  442. tenant_no=tenant_no,
  443. )
  444. return success_response(dict_type_svc._row_to_dict(obj), message="更新成功")
  445. except ValueError as e:
  446. return error_response(400, str(e))
  447. @router.delete("/dict-types/{dict_type_id}")
  448. def delete_dict_type(
  449. dict_type_id: str,
  450. _user: dict = Depends(require_admin_or_super_admin),
  451. db: Session = Depends(get_db),
  452. ):
  453. try:
  454. dict_type_svc.soft_delete_dict_type(db, dict_type_id, user_info=_user)
  455. return success_response(None, message="删除成功")
  456. except ValueError as e:
  457. return error_response(400, str(e))
  458. # ---------------------------------------------------------------------------
  459. # 字典数据管理
  460. # ---------------------------------------------------------------------------
  461. class DictDataCreateBody(BaseModel):
  462. model_config = ConfigDict(populate_by_name=True)
  463. dict_label: str = Field(..., alias="dictLabel")
  464. dict_value: str = Field(..., alias="dictValue")
  465. dict_type: str = Field(..., alias="dictType")
  466. tag_type: Optional[str] = Field("", alias="tagType")
  467. order_num: Optional[int] = Field(0, alias="orderNum")
  468. remark: Optional[str] = None
  469. class DictDataUpdateBody(BaseModel):
  470. model_config = ConfigDict(populate_by_name=True)
  471. dict_label: Optional[str] = Field(None, alias="dictLabel")
  472. dict_value: Optional[str] = Field(None, alias="dictValue")
  473. tag_type: Optional[str] = Field(None, alias="tagType")
  474. order_num: Optional[int] = Field(None, alias="orderNum")
  475. remark: Optional[str] = None
  476. @router.get("/dict-data")
  477. def list_dict_data(
  478. page: int = Query(1, ge=1),
  479. pageSize: int = Query(10, ge=1, le=100, alias="pageSize"),
  480. dictType: Optional[str] = None,
  481. keyword: Optional[str] = None,
  482. user: dict = Depends(require_menu_code("system:dict:list")),
  483. db: Session = Depends(get_db),
  484. ):
  485. tenant_no = _get_tenant_no(db, user)
  486. items, total = dict_data_svc.list_dict_data(
  487. db, page=page, page_size=pageSize, dict_type_code=dictType, keyword=keyword, tenant_no=tenant_no, user_info=user
  488. )
  489. tp = dict_data_svc.total_pages(total, pageSize)
  490. return success_response({"list": items, "total": total, "page": page, "pageSize": pageSize, "totalPages": tp})
  491. @router.post("/dict-data")
  492. def create_dict_data(
  493. body: DictDataCreateBody,
  494. user: dict = Depends(require_admin_or_super_admin),
  495. db: Session = Depends(get_db),
  496. ):
  497. try:
  498. tenant_no = _get_tenant_no(db, user)
  499. obj = dict_data_svc.create_dict_data(
  500. db,
  501. dict_label=body.dict_label,
  502. dict_value=body.dict_value,
  503. dict_type=body.dict_type,
  504. tag_type=body.tag_type,
  505. tenant_no=tenant_no,
  506. order_num=body.order_num,
  507. remark=body.remark,
  508. user_id=user.get("user_id"),
  509. )
  510. return success_response(dict_data_svc._row_to_dict(obj), message="创建成功")
  511. except ValueError as e:
  512. return error_response(400, str(e))
  513. @router.put("/dict-data/{data_id}")
  514. def update_dict_data(
  515. data_id: str,
  516. body: DictDataUpdateBody,
  517. user: dict = Depends(require_admin_or_super_admin),
  518. db: Session = Depends(get_db),
  519. ):
  520. try:
  521. payload: dict[str, Any] = body.model_dump(exclude_unset=True, by_alias=False)
  522. obj = dict_data_svc.update_dict_data(
  523. db,
  524. data_id,
  525. dict_label=payload.get("dict_label"),
  526. dict_value=payload.get("dict_value"),
  527. tag_type=payload.get("tag_type"),
  528. order_num=payload.get("order_num"),
  529. remark=payload.get("remark"),
  530. user_id=user.get("user_id"),
  531. )
  532. return success_response(dict_data_svc._row_to_dict(obj), message="更新成功")
  533. except ValueError as e:
  534. return error_response(400, str(e))
  535. @router.delete("/dict-data/{data_id}")
  536. def delete_dict_data(
  537. data_id: str,
  538. _user: dict = Depends(require_admin_or_super_admin),
  539. db: Session = Depends(get_db),
  540. ):
  541. try:
  542. dict_data_svc.soft_delete_dict_data(db, data_id, user_info=_user)
  543. return success_response(None, message="删除成功")
  544. except ValueError as e:
  545. return error_response(400, str(e))
  546. @router.get("/dict-data/options")
  547. def dict_data_options(
  548. dictType: str = Query(..., alias="dictType"),
  549. user: dict = Depends(require_menu_code("system:dict:list")),
  550. db: Session = Depends(get_db),
  551. ):
  552. tenant_no = _get_tenant_no(db, user)
  553. # 字典选项为租户主数据,表单共用;不做数据范围过滤
  554. items = dict_data_svc.list_dict_data_options(
  555. db, dict_type_code=dictType, tenant_no=tenant_no, user_info=None
  556. )
  557. return success_response(items)
  558. @router.get("/dict-data/all")
  559. def dict_data_all(
  560. user: dict = Depends(require_admin_or_super_admin),
  561. db: Session = Depends(get_db),
  562. ):
  563. tenant_no = _get_tenant_no(db, user)
  564. q = db.query(SysDictData).filter(
  565. and_(SysDictData.tenant_no == tenant_no, SysDictData.del_flag == 0)
  566. )
  567. rows = q.order_by(SysDictData.order_num.asc()).all()
  568. grouped = {}
  569. for r in rows:
  570. key = r.dict_type
  571. if key not in grouped:
  572. grouped[key] = []
  573. grouped[key].append({
  574. "label": r.dict_label,
  575. "value": r.dict_value,
  576. "tagType": r.tag_type or ""
  577. })
  578. return success_response(grouped)
  579. # ---------------------------------------------------------------------------
  580. # 组织管理
  581. # ---------------------------------------------------------------------------
  582. class DeptCreateBody(BaseModel):
  583. model_config = ConfigDict(populate_by_name=True)
  584. parent_id: str = Field("0", alias="parentId")
  585. dept_name: str = Field(..., alias="deptName")
  586. dept_category_code: Optional[str] = Field(None, alias="deptCategoryCode")
  587. status: Optional[str] = Field("1")
  588. leader: Optional[str] = None
  589. leader_phone: Optional[str] = Field(None, alias="leaderPhone")
  590. order_num: Optional[int] = Field(0, alias="orderNum")
  591. remark: Optional[str] = None
  592. class DeptUpdateBody(BaseModel):
  593. model_config = ConfigDict(populate_by_name=True)
  594. dept_name: Optional[str] = Field(None, alias="deptName")
  595. dept_category_code: Optional[str] = Field(None, alias="deptCategoryCode")
  596. status: Optional[str] = None
  597. leader: Optional[str] = None
  598. leader_phone: Optional[str] = Field(None, alias="leaderPhone")
  599. order_num: Optional[int] = Field(None, alias="orderNum")
  600. remark: Optional[str] = None
  601. @router.get("/depts")
  602. def list_depts(
  603. keyword: Optional[str] = None,
  604. user: dict = Depends(require_menu_code("system:dept:list")),
  605. db: Session = Depends(get_db),
  606. ):
  607. tenant_no = _get_tenant_no(db, user)
  608. tree = dept_svc.list_depts_tree(db, keyword=keyword, tenant_no=tenant_no, user_info=user)
  609. return success_response(tree)
  610. @router.post("/depts")
  611. def create_dept(
  612. body: DeptCreateBody,
  613. user: dict = Depends(require_admin_or_super_admin),
  614. db: Session = Depends(get_db),
  615. ):
  616. try:
  617. tenant_no = _get_tenant_no(db, user)
  618. obj = dept_svc.create_dept(
  619. db,
  620. parent_id=body.parent_id,
  621. dept_name=body.dept_name,
  622. dept_category_code=body.dept_category_code,
  623. status=body.status,
  624. leader=body.leader,
  625. leader_phone=body.leader_phone,
  626. tenant_no=tenant_no,
  627. order_num=body.order_num,
  628. remark=body.remark,
  629. user_id=user.get("user_id"),
  630. )
  631. return success_response(dept_svc._row_to_dict(obj), message="创建成功")
  632. except ValueError as e:
  633. return error_response(400, str(e))
  634. @router.put("/depts/{dept_id}")
  635. def update_dept(
  636. dept_id: str,
  637. body: DeptUpdateBody,
  638. user: dict = Depends(require_admin_or_super_admin),
  639. db: Session = Depends(get_db),
  640. ):
  641. try:
  642. payload: dict[str, Any] = body.model_dump(exclude_unset=True, by_alias=False)
  643. obj = dept_svc.update_dept(
  644. db,
  645. dept_id,
  646. dept_name=payload.get("dept_name"),
  647. dept_category_code=payload.get("dept_category_code"),
  648. status=payload.get("status"),
  649. leader=payload.get("leader"),
  650. leader_phone=payload.get("leader_phone"),
  651. order_num=payload.get("order_num"),
  652. remark=payload.get("remark"),
  653. user_id=user.get("user_id"),
  654. )
  655. return success_response(dept_svc._row_to_dict(obj), message="更新成功")
  656. except ValueError as e:
  657. return error_response(400, str(e))
  658. @router.delete("/depts/{dept_id}")
  659. def delete_dept(
  660. dept_id: str,
  661. _user: dict = Depends(require_admin_or_super_admin),
  662. db: Session = Depends(get_db),
  663. ):
  664. try:
  665. dept_svc.soft_delete_dept(db, dept_id, user_info=_user)
  666. return success_response(None, message="删除成功")
  667. except ValueError as e:
  668. return error_response(400, str(e))
  669. # ---------------------------------------------------------------------------
  670. # 站点管理
  671. # ---------------------------------------------------------------------------
  672. class SiteCreateBody(BaseModel):
  673. model_config = ConfigDict(populate_by_name=True)
  674. site_name: str = Field(..., alias="siteName")
  675. site_code: str = Field(..., alias="siteCode")
  676. site_type: Optional[str] = Field(None, alias="siteType")
  677. status: Optional[str] = Field("1")
  678. address: Optional[str] = None
  679. contact_name: Optional[str] = Field(None, alias="contactName")
  680. contact_phone: Optional[str] = Field(None, alias="contactPhone")
  681. business_scope: Optional[str] = Field(None, alias="businessScope")
  682. order_num: Optional[int] = Field(0, alias="orderNum")
  683. remark: Optional[str] = None
  684. class SiteUpdateBody(BaseModel):
  685. model_config = ConfigDict(populate_by_name=True)
  686. site_name: Optional[str] = Field(None, alias="siteName")
  687. site_code: Optional[str] = Field(None, alias="siteCode")
  688. site_type: Optional[str] = Field(None, alias="siteType")
  689. status: Optional[str] = None
  690. address: Optional[str] = None
  691. contact_name: Optional[str] = Field(None, alias="contactName")
  692. contact_phone: Optional[str] = Field(None, alias="contactPhone")
  693. business_scope: Optional[str] = Field(None, alias="businessScope")
  694. order_num: Optional[int] = Field(None, alias="orderNum")
  695. remark: Optional[str] = None
  696. @router.get("/sites")
  697. def list_sites(
  698. page: int = Query(1, ge=1),
  699. pageSize: int = Query(10, ge=1, le=100, alias="pageSize"),
  700. keyword: Optional[str] = None,
  701. user: dict = Depends(require_menu_code("system:site:list")),
  702. db: Session = Depends(get_db),
  703. ):
  704. tenant_no = _get_tenant_no(db, user)
  705. items, total = site_svc.list_sites(db, page=page, page_size=pageSize, keyword=keyword, tenant_no=tenant_no, user_info=user)
  706. tp = site_svc.total_pages(total, pageSize)
  707. return success_response({"list": items, "total": total, "page": page, "pageSize": pageSize, "totalPages": tp})
  708. @router.post("/sites")
  709. def create_site(
  710. body: SiteCreateBody,
  711. user: dict = Depends(require_admin_or_super_admin),
  712. db: Session = Depends(get_db),
  713. ):
  714. try:
  715. tenant_no = _get_tenant_no(db, user)
  716. obj = site_svc.create_site(
  717. db,
  718. site_name=body.site_name,
  719. site_code=body.site_code,
  720. site_type=body.site_type,
  721. status=body.status,
  722. address=body.address,
  723. contact_name=body.contact_name,
  724. contact_phone=body.contact_phone,
  725. business_scope=body.business_scope,
  726. tenant_no=tenant_no,
  727. order_num=body.order_num,
  728. remark=body.remark,
  729. user_id=user.get("user_id"),
  730. )
  731. return success_response(site_svc._row_to_dict(obj), message="创建成功")
  732. except ValueError as e:
  733. return error_response(400, str(e))
  734. @router.put("/sites/{site_id}")
  735. def update_site(
  736. site_id: str,
  737. body: SiteUpdateBody,
  738. user: dict = Depends(require_admin_or_super_admin),
  739. db: Session = Depends(get_db),
  740. ):
  741. try:
  742. payload: dict[str, Any] = body.model_dump(exclude_unset=True, by_alias=False)
  743. obj = site_svc.update_site(
  744. db,
  745. site_id,
  746. site_name=payload.get("site_name"),
  747. site_code=payload.get("site_code"),
  748. site_type=payload.get("site_type"),
  749. status=payload.get("status"),
  750. address=payload.get("address"),
  751. contact_name=payload.get("contact_name"),
  752. contact_phone=payload.get("contact_phone"),
  753. business_scope=payload.get("business_scope"),
  754. order_num=payload.get("order_num"),
  755. remark=payload.get("remark"),
  756. user_id=user.get("user_id"),
  757. )
  758. return success_response(site_svc._row_to_dict(obj), message="更新成功")
  759. except ValueError as e:
  760. return error_response(400, str(e))
  761. @router.delete("/sites/{site_id}")
  762. def delete_site(
  763. site_id: str,
  764. _user: dict = Depends(require_admin_or_super_admin),
  765. db: Session = Depends(get_db),
  766. ):
  767. try:
  768. site_svc.soft_delete_site(db, site_id, user_info=_user)
  769. return success_response(None, message="删除成功")
  770. except ValueError as e:
  771. return error_response(400, str(e))
  772. # ---------------------------------------------------------------------------
  773. # 职位管理
  774. # ---------------------------------------------------------------------------
  775. class PostCreateBody(BaseModel):
  776. model_config = ConfigDict(populate_by_name=True)
  777. post_name: str = Field(..., alias="postName")
  778. post_code: str = Field(..., alias="postCode")
  779. status: Optional[str] = Field("1")
  780. order_num: Optional[int] = Field(0, alias="orderNum")
  781. remark: Optional[str] = None
  782. class PostUpdateBody(BaseModel):
  783. model_config = ConfigDict(populate_by_name=True)
  784. post_name: Optional[str] = Field(None, alias="postName")
  785. post_code: Optional[str] = Field(None, alias="postCode")
  786. status: Optional[str] = None
  787. order_num: Optional[int] = Field(None, alias="orderNum")
  788. remark: Optional[str] = None
  789. @router.get("/posts")
  790. def list_posts(
  791. page: int = Query(1, ge=1),
  792. pageSize: int = Query(10, ge=1, le=100, alias="pageSize"),
  793. keyword: Optional[str] = None,
  794. user: dict = Depends(require_menu_code("system:post:list")),
  795. db: Session = Depends(get_db),
  796. ):
  797. tenant_no = _get_tenant_no(db, user)
  798. items, total = post_svc.list_posts(db, page=page, page_size=pageSize, keyword=keyword, tenant_no=tenant_no, user_info=user)
  799. tp = post_svc.total_pages(total, pageSize)
  800. return success_response({"list": items, "total": total, "page": page, "pageSize": pageSize, "totalPages": tp})
  801. @router.post("/posts")
  802. def create_post(
  803. body: PostCreateBody,
  804. user: dict = Depends(require_admin_or_super_admin),
  805. db: Session = Depends(get_db),
  806. ):
  807. try:
  808. tenant_no = _get_tenant_no(db, user)
  809. obj = post_svc.create_post(
  810. db,
  811. post_name=body.post_name,
  812. post_code=body.post_code,
  813. status=body.status,
  814. tenant_no=tenant_no,
  815. order_num=body.order_num,
  816. remark=body.remark,
  817. user_id=user.get("user_id"),
  818. )
  819. return success_response(post_svc._row_to_dict(obj), message="创建成功")
  820. except ValueError as e:
  821. return error_response(400, str(e))
  822. @router.put("/posts/{post_id}")
  823. def update_post(
  824. post_id: str,
  825. body: PostUpdateBody,
  826. user: dict = Depends(require_admin_or_super_admin),
  827. db: Session = Depends(get_db),
  828. ):
  829. try:
  830. payload: dict[str, Any] = body.model_dump(exclude_unset=True, by_alias=False)
  831. obj = post_svc.update_post(
  832. db,
  833. post_id,
  834. post_name=payload.get("post_name"),
  835. post_code=payload.get("post_code"),
  836. status=payload.get("status"),
  837. order_num=payload.get("order_num"),
  838. remark=payload.get("remark"),
  839. user_id=user.get("user_id"),
  840. )
  841. return success_response(post_svc._row_to_dict(obj), message="更新成功")
  842. except ValueError as e:
  843. return error_response(400, str(e))
  844. @router.delete("/posts/{post_id}")
  845. def delete_post(
  846. post_id: str,
  847. _user: dict = Depends(require_admin_or_super_admin),
  848. db: Session = Depends(get_db),
  849. ):
  850. try:
  851. post_svc.soft_delete_post(db, post_id, user_info=_user)
  852. return success_response(None, message="删除成功")
  853. except ValueError as e:
  854. return error_response(400, str(e))
  855. class PostUsersUpdateBody(BaseModel):
  856. model_config = ConfigDict(populate_by_name=True)
  857. user_ids: List[str] = Field(default_factory=list, alias="userIds")
  858. @router.get("/posts/{post_id}/users")
  859. def list_post_bound_users(
  860. post_id: str,
  861. user: dict = Depends(require_menu_code("system:post:list")),
  862. db: Session = Depends(get_db),
  863. ):
  864. tenant_no = _get_tenant_no(db, user)
  865. post = post_svc.get_post(db, post_id)
  866. if not post or post.tenant_no != tenant_no:
  867. return error_response(404, "职位不存在")
  868. ids = user_svc.list_user_ids_bound_to_post(db, post_id, tenant_no)
  869. return success_response({"userIds": ids})
  870. @router.put("/posts/{post_id}/users")
  871. def update_post_bound_users(
  872. post_id: str,
  873. body: PostUsersUpdateBody,
  874. user: dict = Depends(require_admin_or_super_admin),
  875. db: Session = Depends(get_db),
  876. ):
  877. tenant_no = _get_tenant_no(db, user)
  878. post = post_svc.get_post(db, post_id)
  879. if not post or post.tenant_no != tenant_no:
  880. return error_response(404, "职位不存在")
  881. try:
  882. user_svc.sync_post_user_bindings(
  883. db, post_id, tenant_no, body.user_ids, user.get("user_id")
  884. )
  885. return success_response(None, message="保存成功")
  886. except ValueError as e:
  887. return error_response(400, str(e))
  888. # ---------------------------------------------------------------------------
  889. # 菜单管理
  890. # ---------------------------------------------------------------------------
  891. class MenuCreateBody(BaseModel):
  892. model_config = ConfigDict(populate_by_name=True)
  893. menu_name: str = Field(..., alias="menuName")
  894. parent_id: str = Field("0", alias="parentId")
  895. menu_type: str = Field("M", alias="menuType")
  896. path: Optional[str] = Field("", alias="path")
  897. component: Optional[str] = Field(None, alias="component")
  898. perms: Optional[str] = Field(None, alias="perms")
  899. is_frame: Optional[int] = Field(1, alias="isFrame")
  900. visible: Optional[str] = Field("1", alias="visible")
  901. status: Optional[str] = Field("1", alias="status")
  902. order_num: Optional[int] = Field(None, alias="orderNum")
  903. remark: Optional[str] = None
  904. class MenuUpdateBody(BaseModel):
  905. model_config = ConfigDict(populate_by_name=True)
  906. menu_name: Optional[str] = Field(None, alias="menuName")
  907. parent_id: Optional[str] = Field(None, alias="parentId")
  908. menu_type: Optional[str] = Field(None, alias="menuType")
  909. path: Optional[str] = Field(None, alias="path")
  910. component: Optional[str] = Field(None, alias="component")
  911. perms: Optional[str] = Field(None, alias="perms")
  912. is_frame: Optional[int] = Field(None, alias="isFrame")
  913. visible: Optional[str] = Field(None, alias="visible")
  914. status: Optional[str] = Field(None, alias="status")
  915. order_num: Optional[int] = Field(None, alias="orderNum")
  916. remark: Optional[str] = None
  917. @router.get("/menus")
  918. def list_menus(
  919. keyword: Optional[str] = None,
  920. user: dict = Depends(require_admin_or_super_admin),
  921. db: Session = Depends(get_db),
  922. ):
  923. tenant_no = _get_tenant_no(db, user)
  924. tree = menu_svc.list_menus_tree(db, keyword=keyword, tenant_no=tenant_no, user_info=user)
  925. return success_response(tree)
  926. @router.post("/menus")
  927. def create_menu(
  928. body: MenuCreateBody,
  929. user: dict = Depends(require_admin_or_super_admin),
  930. db: Session = Depends(get_db),
  931. ):
  932. try:
  933. tenant_no = _get_tenant_no(db, user)
  934. obj = menu_svc.create_menu(
  935. db,
  936. menu_name=body.menu_name,
  937. parent_id=body.parent_id,
  938. menu_type=body.menu_type,
  939. path=body.path,
  940. component=body.component,
  941. perms=body.perms,
  942. is_frame=body.is_frame,
  943. visible=body.visible,
  944. status=body.status,
  945. tenant_no=tenant_no,
  946. order_num=body.order_num,
  947. remark=body.remark,
  948. user_id=user.get("user_id"),
  949. )
  950. return success_response(menu_svc._row_to_dict(obj), message="创建成功")
  951. except ValueError as e:
  952. return error_response(400, str(e))
  953. @router.put("/menus/{menu_id}")
  954. def update_menu(
  955. menu_id: str,
  956. body: MenuUpdateBody,
  957. user: dict = Depends(require_admin_or_super_admin),
  958. db: Session = Depends(get_db),
  959. ):
  960. try:
  961. payload: dict[str, Any] = body.model_dump(exclude_unset=True, by_alias=False)
  962. obj = menu_svc.update_menu(
  963. db,
  964. menu_id,
  965. menu_name=payload.get("menu_name"),
  966. parent_id=payload.get("parent_id"),
  967. menu_type=payload.get("menu_type"),
  968. path=payload.get("path"),
  969. component=payload.get("component"),
  970. perms=payload.get("perms"),
  971. is_frame=payload.get("is_frame"),
  972. visible=payload.get("visible"),
  973. status=payload.get("status"),
  974. order_num=payload.get("order_num"),
  975. remark=payload.get("remark"),
  976. user_id=user.get("user_id"),
  977. )
  978. return success_response(menu_svc._row_to_dict(obj), message="更新成功")
  979. except ValueError as e:
  980. return error_response(400, str(e))
  981. @router.delete("/menus/{menu_id}")
  982. def delete_menu(
  983. menu_id: str,
  984. _user: dict = Depends(require_admin_or_super_admin),
  985. db: Session = Depends(get_db),
  986. ):
  987. try:
  988. menu_svc.soft_delete_menu(db, menu_id, user_info=_user)
  989. return success_response(None, message="删除成功")
  990. except ValueError as e:
  991. return error_response(400, str(e))
  992. # ---------------------------------------------------------------------------
  993. # 角色管理
  994. # ---------------------------------------------------------------------------
  995. class RoleCreateBody(BaseModel):
  996. model_config = ConfigDict(populate_by_name=True)
  997. role_name: str = Field(..., alias="roleName")
  998. data_scope: Optional[str] = Field("4", alias="dataScope")
  999. status: Optional[str] = Field("1", alias="status")
  1000. order_num: Optional[int] = Field(None, alias="orderNum")
  1001. remark: Optional[str] = None
  1002. class RoleUpdateBody(BaseModel):
  1003. model_config = ConfigDict(populate_by_name=True)
  1004. role_name: Optional[str] = Field(None, alias="roleName")
  1005. data_scope: Optional[str] = Field(None, alias="dataScope")
  1006. status: Optional[str] = Field(None, alias="status")
  1007. order_num: Optional[int] = Field(None, alias="orderNum")
  1008. remark: Optional[str] = None
  1009. class RoleMenuBody(BaseModel):
  1010. menu_ids: List[str] = Field(..., alias="menuIds")
  1011. class RoleDataScopeBody(BaseModel):
  1012. model_config = ConfigDict(populate_by_name=True)
  1013. data_scope: str = Field(..., alias="dataScope")
  1014. scope_details: Optional[List[dict]] = Field(None, alias="scopeDetails")
  1015. @router.get("/role-management")
  1016. def list_role_management(
  1017. page: int = Query(1, ge=1),
  1018. pageSize: int = Query(10, ge=1, le=100, alias="pageSize"),
  1019. keyword: Optional[str] = None,
  1020. user: dict = Depends(require_menu_code("system:role:list")),
  1021. db: Session = Depends(get_db),
  1022. ):
  1023. tenant_no = _get_tenant_no(db, user)
  1024. items, total = role_svc.list_roles(db, page=page, page_size=pageSize, keyword=keyword, tenant_no=tenant_no, user_info=user)
  1025. tp = role_svc.total_pages(total, pageSize)
  1026. return success_response({"list": items, "total": total, "page": page, "pageSize": pageSize, "totalPages": tp})
  1027. @router.post("/role-management")
  1028. def create_role_management(
  1029. body: RoleCreateBody,
  1030. user: dict = Depends(require_admin_or_super_admin),
  1031. db: Session = Depends(get_db),
  1032. ):
  1033. try:
  1034. tenant_no = _get_tenant_no(db, user)
  1035. obj = role_svc.create_role(
  1036. db,
  1037. role_name=body.role_name,
  1038. data_scope=body.data_scope,
  1039. status=body.status,
  1040. tenant_no=tenant_no,
  1041. order_num=body.order_num,
  1042. remark=body.remark,
  1043. user_id=user.get("user_id"),
  1044. )
  1045. return success_response(role_svc._row_to_dict(obj), message="创建成功")
  1046. except ValueError as e:
  1047. return error_response(400, str(e))
  1048. @router.put("/role-management/{role_id}")
  1049. def update_role_management(
  1050. role_id: str,
  1051. body: RoleUpdateBody,
  1052. user: dict = Depends(require_admin_or_super_admin),
  1053. db: Session = Depends(get_db),
  1054. ):
  1055. try:
  1056. payload: dict[str, Any] = body.model_dump(exclude_unset=True, by_alias=False)
  1057. obj = role_svc.update_role(
  1058. db,
  1059. role_id,
  1060. role_name=payload.get("role_name"),
  1061. data_scope=payload.get("data_scope"),
  1062. status=payload.get("status"),
  1063. order_num=payload.get("order_num"),
  1064. remark=payload.get("remark"),
  1065. user_id=user.get("user_id"),
  1066. )
  1067. return success_response(role_svc._row_to_dict(obj), message="更新成功")
  1068. except ValueError as e:
  1069. return error_response(400, str(e))
  1070. @router.delete("/role-management/{role_id}")
  1071. def delete_role_management(
  1072. role_id: str,
  1073. _user: dict = Depends(require_admin_or_super_admin),
  1074. db: Session = Depends(get_db),
  1075. ):
  1076. try:
  1077. role_svc.soft_delete_role(db, role_id, user_info=_user)
  1078. return success_response(None, message="删除成功")
  1079. except ValueError as e:
  1080. return error_response(400, str(e))
  1081. @router.get("/role-management/{role_id}/menus")
  1082. def get_role_menus(
  1083. role_id: str,
  1084. _user: dict = Depends(require_admin_or_super_admin),
  1085. db: Session = Depends(get_db),
  1086. ):
  1087. menu_ids = role_svc.get_role_menu_ids(db, role_id)
  1088. return success_response({"menuIds": menu_ids})
  1089. @router.put("/role-management/{role_id}/menus")
  1090. def update_role_menus(
  1091. role_id: str,
  1092. body: RoleMenuBody,
  1093. user: dict = Depends(require_admin_or_super_admin),
  1094. db: Session = Depends(get_db),
  1095. ):
  1096. try:
  1097. tenant_no = _get_tenant_no(db, user)
  1098. role_svc.update_role_menus(db, role_id, body.menu_ids, tenant_no, user_id=user.get("user_id"))
  1099. return success_response(None, message="更新成功")
  1100. except ValueError as e:
  1101. return error_response(400, str(e))
  1102. @router.get("/role-management/{role_id}/data-scope")
  1103. def get_role_data_scope(
  1104. role_id: str,
  1105. _user: dict = Depends(require_admin_or_super_admin),
  1106. db: Session = Depends(get_db),
  1107. ):
  1108. role = role_svc.get_role(db, role_id)
  1109. if not role:
  1110. return error_response(404, "角色不存在")
  1111. details = role_svc.get_role_data_scope(db, role_id)
  1112. return success_response({"dataScope": role.data_scope or "1", "scopeDetails": details})
  1113. @router.put("/role-management/{role_id}/data-scope")
  1114. def update_role_data_scope(
  1115. role_id: str,
  1116. body: RoleDataScopeBody,
  1117. user: dict = Depends(require_admin_or_super_admin),
  1118. db: Session = Depends(get_db),
  1119. ):
  1120. try:
  1121. tenant_no = _get_tenant_no(db, user)
  1122. role_svc.update_role_data_scope(
  1123. db,
  1124. role_id,
  1125. body.data_scope,
  1126. body.scope_details,
  1127. tenant_no=tenant_no,
  1128. user_id=user.get("user_id"),
  1129. )
  1130. return success_response(None, message="更新成功")
  1131. except ValueError as e:
  1132. return error_response(400, str(e))
  1133. # ---------------------------------------------------------------------------
  1134. # 附件上传
  1135. # ---------------------------------------------------------------------------
  1136. def _format_file_size(size_bytes: int) -> str:
  1137. if size_bytes < 1024:
  1138. return f"{size_bytes} B"
  1139. elif size_bytes < 1024 * 1024:
  1140. return f"{size_bytes / 1024:.1f} KB"
  1141. else:
  1142. return f"{size_bytes / (1024 * 1024):.1f} MB"
  1143. @router.post("/attachment/upload")
  1144. def upload_attachment(
  1145. file: UploadFile = File(...),
  1146. folder: Optional[str] = Form(""),
  1147. biz_type: Optional[str] = Form(None, alias="bizType"),
  1148. user: dict = Depends(require_admin_or_super_admin),
  1149. db: Session = Depends(get_db),
  1150. ):
  1151. if not file or not file.filename:
  1152. return error_response(400, "请选择文件")
  1153. try:
  1154. effective_folder = resolve_upload_folder(
  1155. folder=folder or "", biz_type=biz_type or ""
  1156. )
  1157. file_content = file.file.read()
  1158. full_url = storage.upload_file(file_content, file.filename, effective_folder)
  1159. base_url = settings.MINIO_BASE_URL.rstrip("/") if settings.MINIO_BASE_URL else ""
  1160. if base_url and full_url.startswith(base_url + "/"):
  1161. file_url = full_url[len(base_url) + 1:]
  1162. else:
  1163. file_url = full_url
  1164. file_size = len(file_content)
  1165. file_ext = os.path.splitext(file.filename)[1].lstrip(".") if file.filename else ""
  1166. file_type = file.content_type or ""
  1167. tenant_no = _get_tenant_no(db, user)
  1168. record = oss_svc.create_oss(
  1169. db,
  1170. tenant_no=tenant_no,
  1171. url=file_url,
  1172. original_name=file.filename,
  1173. file_suffix=file_ext,
  1174. file_size=file_size,
  1175. file_type=file_type,
  1176. biz_type=(biz_type or "").strip(),
  1177. user_id=user.get("user_id"),
  1178. ext1=effective_folder or None,
  1179. )
  1180. return success_response({
  1181. "id": record.id,
  1182. "fileName": record.original_name,
  1183. "fileUrl": record.url,
  1184. "fileSize": record.file_size,
  1185. "fileSizeDisplay": _format_file_size(record.file_size or 0),
  1186. "fileType": record.file_type,
  1187. "bizType": record.biz_type or "",
  1188. })
  1189. except Exception as e:
  1190. db.rollback()
  1191. return error_response(500, f"上传失败: {str(e)}")
  1192. class AttachmentDownloadUrlBody(BaseModel):
  1193. model_config = ConfigDict(populate_by_name=True)
  1194. file_url: str = Field(..., alias="fileUrl")
  1195. file_name: Optional[str] = Field(None, alias="fileName")
  1196. @router.post("/attachment/download-url")
  1197. def attachment_download_url(
  1198. body: AttachmentDownloadUrlBody,
  1199. user: dict = Depends(require_admin_or_super_admin),
  1200. db: Session = Depends(get_db),
  1201. ):
  1202. """返回带 Content-Disposition 的下载 URL,保证浏览器使用原始文件名。"""
  1203. file_url = (body.file_url or "").strip()
  1204. if not file_url:
  1205. return error_response(400, "文件地址不能为空")
  1206. tenant_no = _get_tenant_no(db, user)
  1207. att = oss_svc.get_oss_by_url(db, tenant_no, file_url)
  1208. download_name = (body.file_name or "").strip()
  1209. if not download_name and att:
  1210. download_name = (att.original_name or "").strip()
  1211. if not download_name:
  1212. download_name = file_url.rsplit("/", 1)[-1] or "download"
  1213. try:
  1214. url = storage.presigned_download_url(file_url, download_name)
  1215. return success_response({"downloadUrl": url})
  1216. except Exception as e:
  1217. return error_response(500, f"获取下载链接失败: {str(e)}")
  1218. @router.post("/attachment/preview-url")
  1219. def attachment_preview_url(
  1220. body: AttachmentDownloadUrlBody,
  1221. user: dict = Depends(require_admin_or_super_admin),
  1222. db: Session = Depends(get_db),
  1223. ):
  1224. """返回用于浏览器内联预览的 URL(PDF iframe 等)。"""
  1225. file_url = (body.file_url or "").strip()
  1226. if not file_url:
  1227. return error_response(400, "文件地址不能为空")
  1228. tenant_no = _get_tenant_no(db, user)
  1229. oss_svc.get_oss_by_url(db, tenant_no, file_url)
  1230. try:
  1231. url = storage.presigned_preview_url(file_url)
  1232. return success_response({"previewUrl": url})
  1233. except Exception as e:
  1234. return error_response(500, f"获取预览链接失败: {str(e)}")