admin.py 56 KB

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