__init__.py 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841
  1. #!/usr/bin/env python3
  2. # -*- coding: utf-8 -*-
  3. from fastapi import APIRouter, Request, Depends,Query,HTTPException
  4. from database import get_db
  5. from sqlalchemy.orm import Session
  6. from fastapi.responses import JSONResponse
  7. from models import *
  8. from utils import *
  9. from utils.ry_system_util import *
  10. from common.security import valid_access_token
  11. import traceback
  12. router = APIRouter()
  13. @router.get('/')
  14. async def usercreate01(
  15. db: Session = Depends(get_db),
  16. user_id: int = Depends(valid_access_token)):
  17. try:
  18. return {
  19. "code": 200,
  20. "msg": "成功",
  21. "data": {
  22. "postIds":None,
  23. "posts":None,
  24. "roleIds":None,
  25. "roles":get_role(db),
  26. "user":None
  27. }
  28. }
  29. except Exception as e:
  30. traceback.print_exc()
  31. raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
  32. @router.post('')
  33. async def usercreate(
  34. db: Session = Depends(get_db),
  35. user_id: int = Depends(valid_access_token),
  36. body = Depends(remove_xss_json)
  37. ):
  38. try:
  39. if 'deptId' in body:
  40. deptId = body['deptId']
  41. else:
  42. deptId = None
  43. if 'phonenumber' in body:
  44. phonenumber = body['phonenumber']
  45. else:
  46. phonenumber = None
  47. if 'email' in body:
  48. email = body['email']
  49. else:
  50. email = None
  51. if 'sex' in body:
  52. sex = body['sex']
  53. else:
  54. sex = None
  55. userName = body['userName']
  56. user = user_name_get_user_info(db,userName)
  57. if user:
  58. return JSONResponse(status_code=404, content={"code": 404, "msg": "用户名称已存在"})
  59. pattern = r'^[a-zA-Z0-9_]+$'
  60. if re.match(pattern, userName) == False:
  61. return JSONResponse(status_code=404, content={"code": 404, "msg": "用户名称又字母大小写、阿拉伯数字和下划线组成"})
  62. nickName = body['nickName']
  63. roleIds = body['roleIds']
  64. postIds = body['postIds']
  65. remark = body['remark']
  66. status = body['status']
  67. password = '$2a$10$b8yUzN0C71sbz.PhNOCgJe.Tu1yWC3RNrTyjSQ8p1W0.aaUXUJ.Ne'
  68. new_user = SysUser(
  69. user_name = userName,
  70. nick_name=nickName,
  71. dept_id=deptId,
  72. phonenumber=phonenumber,
  73. email=email,
  74. sex=sex,
  75. remark=remark,
  76. status=status,
  77. password=password
  78. )
  79. db.add(new_user)
  80. db.commit()
  81. db.refresh(new_user)
  82. # 创建新的用户角色关联
  83. new_roles = [SysUserRole(user_id=new_user.user_id, role_id=role_id) for role_id in roleIds]
  84. db.add_all(new_roles)
  85. # 创建新的用户角色关联
  86. new_posts = [SysUserPost(user_id=new_user.user_id, post_id=post_id) for post_id in postIds]
  87. db.add_all(new_posts)
  88. db.commit()
  89. return {"code": 200, "msg": "创建成功", "data": None}
  90. except Exception as e:
  91. traceback.print_exc()
  92. raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
  93. @router.put('')
  94. async def userupdate(
  95. db: Session = Depends(get_db),
  96. user_id: int = Depends(valid_access_token),
  97. body = Depends(remove_xss_json)
  98. ):
  99. try:
  100. user_id_1 = body['userId']
  101. user = user_id_get_user_info(db,user_id_1)
  102. nickName = body['nickName']
  103. deptId = body['deptId']
  104. phonenumber = body['phonenumber']
  105. email = body['email']
  106. sex = body['sex']
  107. roleIds = body['roleIds']
  108. postIds = body['postIds']
  109. remark = body['remark']
  110. user.nick_name=nickName
  111. user.dept_id=deptId
  112. user.phonenumber=phonenumber
  113. user.email=email
  114. user.sex=sex
  115. user.remark=remark
  116. # 解析角色ID列表
  117. # role_ids_list = [int(id) for id in roleIds.split(',')]
  118. # 清除当前用户的所有角色关联
  119. db.query(SysUserRole).filter(SysUserRole.user_id == user_id_1).delete()
  120. # 创建新的用户角色关联
  121. new_roles = [SysUserRole(user_id=user_id_1, role_id=role_id) for role_id in roleIds]
  122. db.add_all(new_roles)
  123. # 解析角色ID列表
  124. # post_ids_list = [int(id) for id in postIds.split(',')]
  125. # 清除当前用户的所有角色关联
  126. db.query(SysUserPost).filter(SysUserPost.user_id == user_id_1).delete()
  127. # 创建新的用户角色关联
  128. new_posts = [SysUserPost(user_id=user_id_1, post_id=post_id) for post_id in postIds]
  129. db.add_all(new_posts)
  130. db.commit()
  131. return {"code": 200, "msg": "更新成功", "data": None}
  132. except Exception as e:
  133. traceback.print_exc()
  134. raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
  135. @router.put('/resetPwd')
  136. async def userupdate(
  137. db: Session = Depends(get_db),
  138. user_id: int = Depends(valid_access_token),
  139. body = Depends(remove_xss_json)
  140. ):
  141. try:
  142. user_id_1 = body['userId']
  143. user = user_id_get_user_info(db,user_id_1)
  144. password = body['password']
  145. user.password=password
  146. user.update_by=user_id
  147. db.commit()
  148. return {"code": 200, "msg": "更新成功", "data": None}
  149. except Exception as e:
  150. traceback.print_exc()
  151. raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
  152. @router.get('/getInfo')
  153. async def getInfo(request: Request,db: Session = Depends(get_db), user_id: int = Depends(valid_access_token)):
  154. try:
  155. # 用户信息
  156. info = db.query(SysUser).filter(SysUser.user_id == user_id).first()
  157. if info is None:
  158. return {
  159. "code": 200,
  160. "msg": "操作成功",
  161. "data": None}
  162. # 部门信息
  163. dept_name = ""
  164. dept_info = db.query(SysDept).filter(SysDept.dept_id == info.dept_id).first()
  165. if dept_name is not None:
  166. dept_name = dept_info.dept_name
  167. # 角色信息
  168. roles = []
  169. role_ids = db.query(SysUserRole).filter(SysUserRole.user_id == user_id).all()
  170. for role in role_ids:
  171. role_info = db.query(SysRole).filter(SysRole.role_id == role.role_id).first()
  172. roles.append(
  173. {
  174. "roleId": role_info.role_id,
  175. "roleName": role_info.role_name,
  176. "roleKey": role_info.role_key,
  177. "roleSort": role_info.role_sort,
  178. "dataScope": role_info.data_scope,
  179. "menuCheckStrictly": role_info.menu_check_strictly,
  180. "deptCheckStrictly": role_info.dept_check_strictly,
  181. "status": role_info.status,
  182. "remark": role_info.remark,
  183. "createTime": get_datetime_str(role_info.create_time),
  184. "flag": False,
  185. "superAdmin": True
  186. }
  187. )
  188. role_keys = [
  189. n['roleKey']
  190. for n in roles
  191. ]
  192. return {
  193. "code": 200,
  194. "msg": "操作成功",
  195. "data": {
  196. "user": {
  197. "userId": info.user_id,
  198. "tenantId": info.tenant_id,
  199. "deptId": info.dept_id,
  200. "userName": info.user_name,
  201. "nickName": info.nick_name,
  202. "userType": info.user_type,
  203. "email": info.email,
  204. "phonenumber": info.phonenumber,
  205. "sex": info.sex,
  206. "avatar": info.avatar,
  207. "status": info.status,
  208. "loginIp": info.login_ip,
  209. "loginDate": get_datetime_str(info.login_date),
  210. "remark": info.remark,
  211. "createTime": get_datetime_str(info.create_time),
  212. "deptName": dept_name,
  213. "roles": roles,
  214. "roleIds": None,
  215. "postIds": None,
  216. "roleId": None
  217. },
  218. "permissions": [
  219. "*:*:*"
  220. ],
  221. "roles": role_keys
  222. }
  223. }
  224. except Exception as e:
  225. traceback.print_exc()
  226. raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
  227. @router.get('/deptTree')
  228. async def deptTree(request: Request,db: Session = Depends(get_db), user_id: int = Depends(valid_access_token)):
  229. def build_dept_tree(depts, parent_dept):
  230. dept_tree = []
  231. for dept_info in depts:
  232. dept = {
  233. "id": dept_info.dept_id,
  234. "label": dept_info.dept_name,
  235. "parentId": dept_info.parent_id,
  236. "weight": dept_info.order_num
  237. }
  238. # print(dept_info.dept_id)
  239. children = parent_id_get_dept_info(db, dept_info.dept_id)
  240. if len(children) > 0:
  241. children_depts = build_dept_tree(children, dept)
  242. dept["children"] = children_depts
  243. dept_tree.append(dept)
  244. return dept_tree
  245. result = build_dept_tree(parent_id_get_dept_info(db, 0),None)
  246. return {
  247. "code": 200,
  248. "msg": "操作成功",
  249. "data": result
  250. }
  251. @router.get('/avcon/deptTree')
  252. async def deptTree(request: Request,label: str = Query(None, description='部门名称'),db: Session = Depends(get_db), user_id: int = Depends(valid_access_token)):
  253. def build_dept_tree(depts, parent_dept):
  254. dept_tree = []
  255. for dept_info in depts:
  256. dept = {
  257. "id": dept_info.dept_id,
  258. "label": dept_info.dept_name,
  259. # "parentId": dept_info.parent_id,
  260. # "weight": dept_info.order_num
  261. }
  262. # print(dept_info.dept_id)
  263. children = parent_id_get_dept_info(db, dept_info.dept_id)
  264. if len(children) > 0:
  265. children_depts = build_dept_tree(children, dept)
  266. dept["children"] = children_depts
  267. else:
  268. dept['isLeaf']=True
  269. dept_tree.append(dept)
  270. return dept_tree
  271. # result = []
  272. if label:
  273. query = db.query(SysDept)
  274. query = query.filter(SysDept.del_flag != '2')
  275. query = query.filter(SysDept.dept_name.like(f'%{label}%'))
  276. # for dept_info in :
  277. result=build_dept_tree(query.all(), None)
  278. else:
  279. result=build_dept_tree(parent_id_get_dept_info(db, 0),None)
  280. return {
  281. "code": 200,
  282. "msg": "操作成功",
  283. "data": result
  284. }
  285. @router.get('/avcon/list/dept/{dept_id}')
  286. async def get_dept_user_list(
  287. # request: Request,
  288. dept_id: int,
  289. name: str = Query(None, description='昵称'),
  290. db: Session = Depends(get_db),
  291. body = Depends(remove_xss_json),
  292. user_id = Depends(valid_access_token)
  293. ):
  294. query = db.query(SysUser)
  295. query = query.filter(SysUser.del_flag != '2')
  296. def get_dept_chli(dept_list: list, parent_id: int):
  297. depts = parent_id_get_dept_info(db, parent_id)
  298. if depts:
  299. for dept in depts:
  300. dept_list.append(dept.dept_id)
  301. get_dept_chli(dept_list, dept.dept_id)
  302. return dept_list
  303. if dept_id:
  304. query = query.filter(SysUser.dept_id.in_(get_dept_chli([], dept_id)))
  305. if name:
  306. query = query.filter(SysUser.nick_name.like(f'%{name}%'))
  307. user_list = query.all()
  308. user_list_dict = []
  309. # 将模型实例转换为字典
  310. for user in user_list:
  311. dept = dept_id_get_ancestors_names(db,dept_id_get_dept_info(db,user.dept_id))
  312. roleIds = user_id_get_user_roleIds(db, user_id)
  313. user_roles = role_id_list_get_user_role(db,roleIds)
  314. dev_id = user_id_get_avcon_dev_id(db, user.user_id)
  315. user_list_dict.append({
  316. "id": user.user_id,
  317. "dept": dept,
  318. "name": user.nick_name,
  319. "mobile": dev_id,
  320. "duty": '/'.join([i["roleName"] for i in user_roles])
  321. } )
  322. return {
  323. "code": 200,
  324. "data": user_list_dict,
  325. "msg": "操作成功"
  326. }
  327. # def get_query_params(params: dict):
  328. # return params
  329. # def get_time_params(params: dict = Depends(get_query_params)):
  330. # try:
  331. # begin_time = params.get("beginTime")
  332. # end_time = params.get("endTime")
  333. # return begin_time, end_time
  334. # except:
  335. # return None,None
  336. @router.get('/list')
  337. async def userlist( deptId: int = Query(None ,description='部门id'),
  338. userName: str = Query(None, description='用户名'),
  339. status: int = Query(None, description='用户状态'),
  340. phonenumber : str = Query(None, description='手机号'),
  341. # params:dict = Query(None, description='创建日期'),
  342. # beginTime: str = Depends(get_time_params),
  343. # endTime: str = Depends(get_time_params),
  344. page: int = Query(1, gt=0, description='页码'),
  345. pageSize: int = Query(10, gt=0, description='每页条目数量'),
  346. db: Session = Depends(get_db),
  347. user_id: int = Depends(valid_access_token)):
  348. try:
  349. # 构建查询
  350. query = db.query(SysUser)
  351. query = query.filter(SysUser.del_flag != '2')
  352. # 应用查询条件
  353. # if beginTime:
  354. # query = query.filter(SysUser.create_time >= beginTime)
  355. # if endTime:
  356. # query = query.filter(SysUser.create_time <= endTime)
  357. # if params:
  358. # query = query.filter(SysUser.create_time >= params.get("beginTime"))
  359. # query = query.filter(SysUser.create_time <= params.get("endTime"))
  360. if userName:
  361. query =query.filter(SysUser.user_name.like(f'%{userName}%'))
  362. if status:
  363. query =query.filter(SysUser.status==status)
  364. if phonenumber:
  365. query =query.filter(SysUser.phonenumber.like(f'%{phonenumber}%'))
  366. def get_dept_chli(dept_list : list,parent_id : int):
  367. depts = parent_id_get_dept_info(db,parent_id)
  368. if depts:
  369. for dept in depts:
  370. dept_list.append(dept.dept_id)
  371. get_dept_chli(dept_list, dept.dept_id)
  372. return dept_list
  373. if deptId:
  374. query = query.filter(SysUser.dept_id.in_(get_dept_chli([],deptId)))
  375. # 计算总条目数
  376. total_items = query.count()
  377. # 排序
  378. query = query.order_by(SysUser.create_time.desc())
  379. # 执行分页查询
  380. users = query.offset((page - 1) * pageSize).limit(pageSize).all()
  381. # 将查询结果转换为列表形式的字典
  382. user_list = []
  383. for user in users:
  384. roleIds = user_id_get_user_roleIds(db, user.user_id)
  385. user_roles = role_id_list_get_user_role(db,roleIds)
  386. user_info = {
  387. "userId": user.user_id,
  388. "tenantId": user.tenant_id,
  389. "deptId": user.dept_id,
  390. "userName": user.user_name,
  391. "nickName": user.nick_name,
  392. "userType": user.user_type,
  393. "email": user.email,
  394. "phonenumber": user.phonenumber,
  395. "sex": user.sex,
  396. "avatar": user.avatar,
  397. "status": user.status,
  398. "loginIp": user.login_ip,
  399. "loginDate": user.login_date.strftime('%Y-%m-%d %H:%M:%S') if user.login_date else '',
  400. "remark": user.remark,
  401. "createTime": user.create_time.strftime('%Y-%m-%d %H:%M:%S') if user.create_time else '',
  402. "deptName": user.dept_name,
  403. "roles": user_roles,
  404. "roleIds": None,
  405. "postIds": None,
  406. "roleId": None
  407. }
  408. user_list.append(user_info)
  409. # 返回结果
  410. return {
  411. "code": 200,
  412. "msg": "成功用户列表",
  413. "rows": user_list,
  414. "total": total_items,
  415. "page": page,
  416. "pageSize": pageSize,
  417. "totalPages": (total_items + pageSize - 1) // pageSize
  418. }
  419. except Exception as e:
  420. traceback.print_exc()
  421. raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
  422. @router.get('/list/dept/{dept_id}')
  423. async def get_dept_user_list(
  424. # request: Request,
  425. dept_id: int,
  426. db: Session = Depends(get_db),
  427. body = Depends(remove_xss_json),
  428. user_id = Depends(valid_access_token)
  429. ):
  430. query = db.query(SysUser)
  431. query = query.filter(SysUser.del_flag != '2')
  432. if dept_id:
  433. query = query.filter(SysUser.dept_id == dept_id)
  434. user_list = query.all()
  435. # 将模型实例转换为字典
  436. user_list_dict = [{
  437. "userId": user.user_id,
  438. "tenantId": user.tenant_id,
  439. "deptId": user.dept_id,
  440. "userName": user.user_name,
  441. "nickName": user.nick_name,
  442. "userType": user.user_type,
  443. "email": user.email,
  444. "phonenumber": user.phonenumber,
  445. "sex": user.sex,
  446. "avatar": user.avatar,
  447. "status": user.status,
  448. "loginIp": user.login_ip,
  449. "loginDate": user.login_date.strftime('%Y-%m-%d %H:%M:%S') if user.login_date else '',
  450. "remark": user.remark,
  451. "createTime": user.create_time.strftime('%Y-%m-%d %H:%M:%S') if user.create_time else '',
  452. "deptName": user.dept_name,
  453. "roles": None,
  454. "roleIds": None,
  455. "postIds": None,
  456. "roleId": None
  457. } for user in user_list]
  458. return {
  459. "code": 200,
  460. "data": user_list_dict,
  461. "msg": "操作成功"
  462. }
  463. @router.put('/changeStatus')
  464. async def change_user_status(
  465. db: Session = Depends(get_db),
  466. body=Depends(remove_xss_json),
  467. user_id=Depends(valid_access_token)
  468. ):
  469. try:
  470. userId = body['userId']
  471. status = body['status']
  472. query = db.query(SysUser)
  473. query = query.filter(SysUser.del_flag != '2')
  474. query = query.filter(SysUser.user_id == userId)
  475. user = query.first()
  476. user.status= status
  477. user.update_by=user_id
  478. db.commit()
  479. return {
  480. "code": 200,
  481. "msg": "操作成功"
  482. }
  483. except Exception as e:
  484. # 处理异常
  485. db.rollback()
  486. traceback.print_exc()
  487. raise HTTPException(status_code=500, detail=str(e))
  488. @router.delete('/{user_id1}')
  489. async def delete_dept(
  490. user_id1:int,
  491. db: Session = Depends(get_db),
  492. body = Depends(remove_xss_json),
  493. user_id = Depends(valid_access_token)
  494. ):
  495. user = user_id_get_user_info(db,user_id1)
  496. # 将模型实例转换为字典
  497. print(user)
  498. user.del_flag = '2'
  499. user.update_by = user_id
  500. db.commit()
  501. return {
  502. "code": 200,
  503. "data": None,
  504. "msg": "操作成功"
  505. }
  506. @router.get('/videoPoints')
  507. async def get_user_video_points(
  508. db: Session = Depends(get_db),
  509. body = Depends(remove_xss_json),
  510. user_id = Depends(valid_access_token)):
  511. try:
  512. videoIds = user_id_get_user_videoIds(db,user_id)
  513. video_list = [i.video_code_int for i in videoIds]
  514. query = db.query(TpVideoLog)
  515. query = query.filter(TpVideoLog.video_code_int.in_(video_list))
  516. video_info = query.all()
  517. return {
  518. "code": 200,
  519. "msg": "成功",
  520. "data": {"videoIds":[i.video_code_int for i in videoIds],
  521. "videoInfos":[{"name":info.name,"video_code_int":info.video_code_int} for info in video_info]}
  522. }
  523. except Exception as e:
  524. traceback.print_exc()
  525. raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
  526. @router.put('/videoPoints')
  527. async def get_user_video_points(
  528. db: Session = Depends(get_db),
  529. body = Depends(remove_xss_json),
  530. user_id = Depends(valid_access_token)):
  531. try:
  532. query = db.query(SysUserVideo)
  533. query = query.filter(SysUserVideo.user_id == user_id)
  534. query.delete(synchronize_session=False)
  535. db.commit()
  536. for video in body:
  537. new_user_video = SysUserVideo(
  538. user_id=user_id,
  539. video_code_int = video
  540. )
  541. db.add(new_user_video)
  542. db.commit()
  543. return {
  544. "code": 200,
  545. "msg": "成功",
  546. "data":None
  547. }
  548. except Exception as e:
  549. traceback.print_exc()
  550. raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
  551. @router.put('/authRole')
  552. async def authRoleUpdate( userId: int = Query(None ,description='用户id'),
  553. roleIds: str = Query(None ,description='权限id'),
  554. db: Session = Depends(get_db),
  555. user_id: int = Depends(valid_access_token)):
  556. try:
  557. # 解析角色ID列表
  558. role_ids_list = [int(id) for id in roleIds.split(',')]
  559. # 清除当前用户的所有角色关联
  560. db.query(SysUserRole).filter(SysUserRole.user_id == userId).delete()
  561. # 创建新的用户角色关联
  562. new_roles = [SysUserRole(user_id=userId, role_id=role_id) for role_id in role_ids_list]
  563. db.add_all(new_roles)
  564. db.commit()
  565. return {"code": 200, "msg": "更新成功", "data": None}
  566. except Exception as e:
  567. traceback.print_exc()
  568. raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
  569. @router.get('/profile')
  570. async def userlist( #user_id_1: int = Query(None ,description='用户id'),
  571. db: Session = Depends(get_db),
  572. user_id: int = Depends(valid_access_token)):
  573. try:
  574. user = user_id_get_user_info(db,user_id)
  575. # 将查询结果转换为列表形式的字典
  576. roleIds = user_id_get_user_roleIds(db, user_id)
  577. user_roles = role_id_list_get_user_role(db,roleIds)
  578. dept = dept_id_get_dept_info(db,user.dept_id)
  579. user_list = {
  580. "admin":1 in roleIds,
  581. "avatar": user.avatar,
  582. "createBy":user_id_get_user_info(db,user.create_by).user_name,
  583. "createTime": user.create_time.strftime('%Y-%m-%d %H:%M:%S') if user.create_time else '',
  584. "delFlag":user.del_flag,
  585. "dept":{
  586. "createBy": user_id_get_user_info(db,dept.create_by).user_name if user.create_by else None,
  587. "createTime": dept.create_time.strftime('%Y-%m-%d %H:%M:%S') if user.create_time else None,
  588. "updateBy": user_id_get_user_info(db,dept.update_by).user_name if user.update_by else None,
  589. "updateTime": dept.create_time.strftime('%Y-%m-%d %H:%M:%S') if user.create_time else None,
  590. "remark": '',#dept.remark,
  591. "deptId": dept.dept_id,
  592. "parentId": dept.parent_id,
  593. "ancestors": dept.ancestors,
  594. "deptName": dept.dept_name,
  595. "orderNum": dept.order_num,
  596. "leader": dept.leader_name,
  597. "phone": dept.phone,
  598. "email": dept.email,
  599. "status": dept.status,
  600. "delFlag": dept.del_flag,
  601. "parentName": dept.parent_name,
  602. "children": []
  603. },
  604. "deptId": user.dept_id,
  605. "email": user.email,
  606. "loginDate": user.login_date.strftime('%Y-%m-%d %H:%M:%S') if user.login_date else '',
  607. "loginIp": user.login_ip,
  608. "nickName": user.nick_name,
  609. "password":"",
  610. "phonenumber": user.phonenumber,
  611. "postIds": None,
  612. "remark": user.remark,
  613. "roleId": None,
  614. "roleIds": roleIds,
  615. "roles": user_roles,
  616. "sex": user.sex,
  617. "status": user.status,
  618. "updateBy":user.update_by,
  619. "updateTime": user.create_time.strftime('%Y-%m-%d %H:%M:%S') if user.create_time else None,
  620. "userId": user.user_id,
  621. "userName": user.user_name,
  622. "tenantId": user.tenant_id,
  623. "userType": user.user_type,
  624. "deptName": dept.dept_name,
  625. }
  626. roles = get_role(db,roleIds)
  627. # 返回结果
  628. return {
  629. "code": 200,
  630. "msg": "成功用户列表",
  631. "data": user_list,
  632. # {
  633. # "user":user_list,
  634. # "roleIds":roleIds,
  635. # "roles":roles,
  636. # "postIds":[],
  637. # "posts":[]
  638. # }
  639. "postGroup":"工作人员",
  640. "roleGroup":'/'.join([i["roleName"] for i in user_roles])
  641. }
  642. except Exception as e:
  643. traceback.print_exc()
  644. raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
  645. @router.put('/profile')
  646. async def userlist( #user_id_1: int = Query(None ,description='用户id'),
  647. db: Session = Depends(get_db),
  648. body = Depends(remove_xss_json),
  649. user_id: int = Depends(valid_access_token)):
  650. try:
  651. user = user_id_get_user_info(db,user_id)
  652. if "email" in body:
  653. user.email=body['email']
  654. user.update_by=user_id
  655. if "nickName" in body:
  656. user.nick_name = body['nickName']
  657. user.update_by=user_id
  658. if "phonenumber" in body:
  659. user.phonenumber = body['phonenumber']
  660. user.update_by=user_id
  661. if "sex" in body:
  662. user.nick_name = body['sex']
  663. user.update_by=user_id
  664. db.commit()
  665. return {"code":200,"msg":"成功","data":None}
  666. except Exception as e:
  667. traceback.print_exc()
  668. raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
  669. @router.put('/profile/updatePwd')
  670. async def userlist( #user_id_1: int = Query(None ,description='用户id'),
  671. db: Session = Depends(get_db),
  672. body = Depends(remove_xss_json),
  673. user_id: int = Depends(valid_access_token)):
  674. try:
  675. user = user_id_get_user_info(db,user_id)
  676. if user.password!=body['oldPassword']:
  677. return JSONResponse(status_code=404,content={"code":404,"msg":"密码错误"})
  678. if "newPassword" in body:
  679. user.nick_name = body['newPassword']
  680. user.update_by=user_id
  681. db.commit()
  682. return {"code":200,"msg":"成功","data":None}
  683. except Exception as e:
  684. traceback.print_exc()
  685. raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
  686. @router.get('/authRole/{user_id_1}')
  687. @router.get('/{user_id_1}')
  688. async def userlist( user_id_1: int = Query(None ,description='用户id'),
  689. db: Session = Depends(get_db),
  690. user_id: int = Depends(valid_access_token)):
  691. try:
  692. user = user_id_get_user_info(db,user_id_1)
  693. # 将查询结果转换为列表形式的字典
  694. roleIds = user_id_get_user_roleIds(db, user_id_1)
  695. user_roles = role_id_list_get_user_role(db,roleIds)
  696. user_list = {
  697. "userId": user.user_id,
  698. "tenantId": user.tenant_id,
  699. "deptId": user.dept_id,
  700. "userName": user.user_name,
  701. "nickName": user.nick_name,
  702. "userType": user.user_type,
  703. "email": user.email,
  704. "phonenumber": user.phonenumber,
  705. "sex": user.sex,
  706. "avatar": user.avatar,
  707. "status": user.status,
  708. "loginIp": user.login_ip,
  709. "loginDate": user.login_date.strftime('%Y-%m-%d %H:%M:%S') if user.login_date else '',
  710. "remark": user.remark,
  711. "createTime": user.create_time.strftime('%Y-%m-%d %H:%M:%S') if user.create_time else '',
  712. "deptName": user.dept_name,
  713. "roles": user_roles,
  714. "roleIds": roleIds,
  715. "postIds": None,
  716. "roleId": None
  717. }
  718. roles = get_role(db,roleIds)
  719. # 返回结果
  720. return {
  721. "code": 200,
  722. "msg": "成功用户列表",
  723. "data": {
  724. "user":user_list,
  725. "roleIds":roleIds,
  726. "roles":roles,
  727. "postIds":[],
  728. "posts":[]
  729. }
  730. }
  731. except Exception as e:
  732. traceback.print_exc()
  733. raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")