__init__.py 32 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938
  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. """
  254. def build_dept_tree(depts, parent_dept):
  255. dept_tree = []
  256. for dept_info in depts:
  257. dept = {
  258. "id": dept_info.dept_id,
  259. "label": dept_info.dept_name,
  260. # "parentId": dept_info.parent_id,
  261. # "weight": dept_info.order_num
  262. }
  263. # print(dept_info.dept_id)
  264. children = parent_id_get_dept_info(db, dept_info.dept_id)
  265. if len(children) > 0:
  266. children_depts = build_dept_tree(children, dept)
  267. dept["children"] = children_depts
  268. # userlist=dept_id_get_user_info(db,dept_info.dept_id)
  269. # if userlist:
  270. # for user_info in userlist:
  271. # dept["children"].append({"id":user_info.user_id,"label":user_info.nick_name,"isLeaf":True})
  272. else:
  273. dept['isLeaf']=True
  274. # userlist = dept_id_get_user_info(db, dept_info.dept_id)
  275. # if userlist:
  276. # for user_info in userlist:
  277. # dept["children"].append({"id": user_info.user_id, "label": user_info.nick_name, "isLeaf": True})
  278. dept_tree.append(dept)
  279. return dept_tree
  280. # result = []
  281. if label:
  282. query = db.query(SysDept)
  283. query = query.filter(SysDept.del_flag != '2')
  284. query = query.filter(SysDept.dept_name.like(f'%{label}%'))
  285. # for dept_info in :
  286. result=build_dept_tree(query.all(), None)
  287. else:
  288. result=build_dept_tree(parent_id_get_dept_info(db, 0),None)
  289. """
  290. result = []
  291. # 暂时写死
  292. result.append({
  293. "id": "G2@mm.zw.yj",
  294. "label": "指挥终端",
  295. "isLeaf": True
  296. })
  297. result.append({
  298. "id": "G5@mm.zw.yj",
  299. "label": "移动视频",
  300. "isLeaf": True
  301. })
  302. result.append({
  303. "id": "G6@mm.zw.yj",
  304. "label": "H323会议",
  305. "isLeaf": True
  306. })
  307. return {
  308. "code": 200,
  309. "msg": "操作成功",
  310. "data": result
  311. }
  312. @router.get('/avcon/deptUserTree')
  313. async def deptTree(request: Request,label: str = Query(None, description='部门名称'),db: Session = Depends(get_db), user_id: int = Depends(valid_access_token)):
  314. def build_dept_tree(depts, parent_dept):
  315. dept_tree = []
  316. for dept_info in depts:
  317. dept = {
  318. "id": dept_info.dept_id,
  319. "label": dept_info.dept_name,
  320. # "parentId": dept_info.parent_id,
  321. # "weight": dept_info.order_num
  322. }
  323. # print(dept_info.dept_id)
  324. children = parent_id_get_dept_info(db, dept_info.dept_id)
  325. if len(children) > 0:
  326. children_depts = build_dept_tree(children, dept)
  327. dept["children"] = children_depts
  328. userlist=dept_id_get_user_info(db,dept_info.dept_id)
  329. if userlist:
  330. for user_info in userlist:
  331. dept["children"].append({"id":user_info.user_id,"label":user_info.nick_name,"isUser":True,"isLeaf":True})
  332. else:
  333. dept['isLeaf']=True
  334. userlist = dept_id_get_user_info(db, dept_info.dept_id)
  335. if userlist:
  336. for user_info in userlist:
  337. dept["children"].append({"id": user_info.user_id, "label": user_info.nick_name,"isUser":True, "isLeaf": True})
  338. dept_tree.append(dept)
  339. return dept_tree
  340. # result = []
  341. if label:
  342. query = db.query(SysDept)
  343. query = query.filter(SysDept.del_flag != '2')
  344. query = query.filter(SysDept.dept_name.like(f'%{label}%'))
  345. # for dept_info in :
  346. result=build_dept_tree(query.all(), None)
  347. else:
  348. result=build_dept_tree(parent_id_get_dept_info(db, 0),None)
  349. return {
  350. "code": 200,
  351. "msg": "操作成功",
  352. "data": result
  353. }
  354. @router.get('/avcon/list/dept/{dept_id}')
  355. async def get_dept_user_list(
  356. # request: Request,
  357. dept_id: str,
  358. name: str = Query(None, description='昵称'),
  359. db: Session = Depends(get_db),
  360. body = Depends(remove_xss_json),
  361. user_id = Depends(valid_access_token)
  362. ):
  363. def get_avcon_type_text(type: str) -> str:
  364. if type == '001':
  365. return '指挥视频终端'
  366. elif type == '060':
  367. return '营区监控'
  368. elif type == '085':
  369. return '监控网关服务器'
  370. elif type == '100':
  371. return 'H.323终端'
  372. return "未知终端"
  373. """
  374. query = db.query(SysUser)
  375. query = query.filter(SysUser.del_flag != '2')
  376. def get_dept_chli(dept_list: list, parent_id: int):
  377. depts = parent_id_get_dept_info(db, parent_id)
  378. if depts:
  379. for dept in depts:
  380. dept_list.append(dept.dept_id)
  381. get_dept_chli(dept_list, dept.dept_id)
  382. return dept_list
  383. if dept_id:
  384. query = query.filter(SysUser.dept_id.in_(get_dept_chli([], dept_id)))
  385. if name:
  386. query = query.filter(SysUser.nick_name.like(f'%{name}%'))
  387. user_list = query.all()
  388. user_list_dict = []
  389. # 将模型实例转换为字典
  390. for user in user_list:
  391. dept = dept_id_get_ancestors_names(db,dept_id_get_dept_info(db,user.dept_id))
  392. roleIds = user_id_get_user_roleIds(db, user_id)
  393. user_roles = role_id_list_get_user_role(db,roleIds)
  394. dev_id = user_id_get_avcon_dev_id(db, user.user_id)
  395. user_list_dict.append({
  396. "id": user.user_id,
  397. "dept": dept,
  398. "name": user.nick_name,
  399. "mobile": dev_id,
  400. "duty": '/'.join([i["roleName"] for i in user_roles])
  401. } )
  402. """
  403. user_list_dict = []
  404. query = db.query(AvconDevice).filter(AvconDevice.group_id == dept_id)
  405. query = query.filter(AvconDevice.dev_type == '001')
  406. # query = query.filter(AvconDevice.status == 1) # 上线状态
  407. query = query.order_by(AvconDevice.group_id.asc(), AvconDevice.dev_id.asc())
  408. rows = query.all()
  409. for row in rows:
  410. user_list_dict.append({
  411. "id": row.dev_id,
  412. "dept": get_avcon_type_text(row.dev_type),
  413. "name": row.dev_name,
  414. "mobile": row.dev_id
  415. })
  416. return {
  417. "code": 200,
  418. "data": user_list_dict,
  419. "msg": "操作成功"
  420. }
  421. # def get_query_params(params: dict):
  422. # return params
  423. # def get_time_params(params: dict = Depends(get_query_params)):
  424. # try:
  425. # begin_time = params.get("beginTime")
  426. # end_time = params.get("endTime")
  427. # return begin_time, end_time
  428. # except:
  429. # return None,None
  430. @router.get('/list')
  431. async def userlist( deptId: int = Query(None ,description='部门id'),
  432. userName: str = Query(None, description='用户名'),
  433. status: int = Query(None, description='用户状态'),
  434. phonenumber : str = Query(None, description='手机号'),
  435. # params:dict = Query(None, description='创建日期'),
  436. # beginTime: str = Depends(get_time_params),
  437. # endTime: str = Depends(get_time_params),
  438. page: int = Query(1, gt=0, description='页码'),
  439. pageSize: int = Query(10, gt=0, description='每页条目数量'),
  440. db: Session = Depends(get_db),
  441. user_id: int = Depends(valid_access_token)):
  442. try:
  443. # 构建查询
  444. query = db.query(SysUser)
  445. query = query.filter(SysUser.del_flag != '2')
  446. # 应用查询条件
  447. # if beginTime:
  448. # query = query.filter(SysUser.create_time >= beginTime)
  449. # if endTime:
  450. # query = query.filter(SysUser.create_time <= endTime)
  451. # if params:
  452. # query = query.filter(SysUser.create_time >= params.get("beginTime"))
  453. # query = query.filter(SysUser.create_time <= params.get("endTime"))
  454. if userName:
  455. query =query.filter(SysUser.user_name.like(f'%{userName}%'))
  456. if status:
  457. query =query.filter(SysUser.status==status)
  458. if phonenumber:
  459. query =query.filter(SysUser.phonenumber.like(f'%{phonenumber}%'))
  460. def get_dept_chli(dept_list : list,parent_id : int):
  461. depts = parent_id_get_dept_info(db,parent_id)
  462. if depts:
  463. for dept in depts:
  464. dept_list.append(dept.dept_id)
  465. get_dept_chli(dept_list, dept.dept_id)
  466. return dept_list
  467. if deptId:
  468. query = query.filter(SysUser.dept_id.in_(get_dept_chli([deptId],deptId)))
  469. # 计算总条目数
  470. total_items = query.count()
  471. # 排序
  472. query = query.order_by(SysUser.create_time.desc())
  473. # 执行分页查询
  474. users = query.offset((page - 1) * pageSize).limit(pageSize).all()
  475. # 将查询结果转换为列表形式的字典
  476. user_list = []
  477. for user in users:
  478. roleIds = user_id_get_user_roleIds(db, user.user_id)
  479. user_roles = role_id_list_get_user_role(db,roleIds)
  480. user_info = {
  481. "userId": user.user_id,
  482. "tenantId": user.tenant_id,
  483. "deptId": user.dept_id,
  484. "userName": user.user_name,
  485. "nickName": user.nick_name,
  486. "userType": user.user_type,
  487. "email": user.email,
  488. "phonenumber": user.phonenumber,
  489. "sex": user.sex,
  490. "avatar": user.avatar,
  491. "status": user.status,
  492. "loginIp": user.login_ip,
  493. "loginDate": user.login_date.strftime('%Y-%m-%d %H:%M:%S') if user.login_date else '',
  494. "remark": user.remark,
  495. "createTime": user.create_time.strftime('%Y-%m-%d %H:%M:%S') if user.create_time else '',
  496. "deptName": user.dept_name,
  497. "roles": user_roles,
  498. "roleIds": None,
  499. "postIds": None,
  500. "roleId": None
  501. }
  502. user_list.append(user_info)
  503. # 返回结果
  504. return {
  505. "code": 200,
  506. "msg": "成功用户列表",
  507. "rows": user_list,
  508. "total": total_items,
  509. "page": page,
  510. "pageSize": pageSize,
  511. "totalPages": (total_items + pageSize - 1) // pageSize
  512. }
  513. except Exception as e:
  514. traceback.print_exc()
  515. raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
  516. @router.get('/list/dept/{dept_id}')
  517. async def get_dept_user_list(
  518. # request: Request,
  519. dept_id: int,
  520. db: Session = Depends(get_db),
  521. body = Depends(remove_xss_json),
  522. user_id = Depends(valid_access_token)
  523. ):
  524. query = db.query(SysUser)
  525. query = query.filter(SysUser.del_flag != '2')
  526. if dept_id:
  527. query = query.filter(SysUser.dept_id == dept_id)
  528. user_list = query.all()
  529. # 将模型实例转换为字典
  530. user_list_dict = [{
  531. "userId": user.user_id,
  532. "tenantId": user.tenant_id,
  533. "deptId": user.dept_id,
  534. "userName": user.user_name,
  535. "nickName": user.nick_name,
  536. "userType": user.user_type,
  537. "email": user.email,
  538. "phonenumber": user.phonenumber,
  539. "sex": user.sex,
  540. "avatar": user.avatar,
  541. "status": user.status,
  542. "loginIp": user.login_ip,
  543. "loginDate": user.login_date.strftime('%Y-%m-%d %H:%M:%S') if user.login_date else '',
  544. "remark": user.remark,
  545. "createTime": user.create_time.strftime('%Y-%m-%d %H:%M:%S') if user.create_time else '',
  546. "deptName": user.dept_name,
  547. "roles": None,
  548. "roleIds": None,
  549. "postIds": None,
  550. "roleId": None
  551. } for user in user_list]
  552. return {
  553. "code": 200,
  554. "data": user_list_dict,
  555. "msg": "操作成功"
  556. }
  557. @router.put('/changeStatus')
  558. async def change_user_status(
  559. db: Session = Depends(get_db),
  560. body=Depends(remove_xss_json),
  561. user_id=Depends(valid_access_token)
  562. ):
  563. try:
  564. userId = body['userId']
  565. status = body['status']
  566. query = db.query(SysUser)
  567. query = query.filter(SysUser.del_flag != '2')
  568. query = query.filter(SysUser.user_id == userId)
  569. user = query.first()
  570. user.status= status
  571. user.update_by=user_id
  572. db.commit()
  573. return {
  574. "code": 200,
  575. "msg": "操作成功"
  576. }
  577. except Exception as e:
  578. # 处理异常
  579. db.rollback()
  580. traceback.print_exc()
  581. raise HTTPException(status_code=500, detail=str(e))
  582. @router.delete('/{user_id1}')
  583. async def delete_dept(
  584. user_id1:int,
  585. db: Session = Depends(get_db),
  586. body = Depends(remove_xss_json),
  587. user_id = Depends(valid_access_token)
  588. ):
  589. user = user_id_get_user_info(db,user_id1)
  590. # 将模型实例转换为字典
  591. print(user)
  592. user.del_flag = '2'
  593. user.update_by = user_id
  594. db.commit()
  595. return {
  596. "code": 200,
  597. "data": None,
  598. "msg": "操作成功"
  599. }
  600. @router.get('/videoPoints')
  601. async def get_user_video_points(
  602. db: Session = Depends(get_db),
  603. body = Depends(remove_xss_json),
  604. user_id = Depends(valid_access_token)):
  605. try:
  606. videoIds = user_id_get_user_videoIds(db,user_id)
  607. video_list = [i.video_code_int for i in videoIds]
  608. query = db.query(TpVideoLog)
  609. query = query.filter(TpVideoLog.video_code_int.in_(video_list))
  610. video_info = query.all()
  611. return {
  612. "code": 200,
  613. "msg": "成功",
  614. "data": {"videoIds":[i.video_code_int for i in videoIds],
  615. "videoInfos":[{"name":info.name,"video_code_int":info.video_code_int} for info in video_info]}
  616. }
  617. except Exception as e:
  618. traceback.print_exc()
  619. raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
  620. @router.put('/videoPoints')
  621. async def get_user_video_points(
  622. db: Session = Depends(get_db),
  623. body = Depends(remove_xss_json),
  624. user_id = Depends(valid_access_token)):
  625. try:
  626. query = db.query(SysUserVideo)
  627. query = query.filter(SysUserVideo.user_id == user_id)
  628. query.delete(synchronize_session=False)
  629. db.commit()
  630. for video in body:
  631. new_user_video = SysUserVideo(
  632. user_id=user_id,
  633. video_code_int = video
  634. )
  635. db.add(new_user_video)
  636. db.commit()
  637. return {
  638. "code": 200,
  639. "msg": "成功",
  640. "data":None
  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('/authRole')
  646. async def authRoleUpdate( userId: int = Query(None ,description='用户id'),
  647. roleIds: str = Query(None ,description='权限id'),
  648. db: Session = Depends(get_db),
  649. user_id: int = Depends(valid_access_token)):
  650. try:
  651. # 解析角色ID列表
  652. role_ids_list = [int(id) for id in roleIds.split(',')]
  653. # 清除当前用户的所有角色关联
  654. db.query(SysUserRole).filter(SysUserRole.user_id == userId).delete()
  655. # 创建新的用户角色关联
  656. new_roles = [SysUserRole(user_id=userId, role_id=role_id) for role_id in role_ids_list]
  657. db.add_all(new_roles)
  658. db.commit()
  659. return {"code": 200, "msg": "更新成功", "data": None}
  660. except Exception as e:
  661. traceback.print_exc()
  662. raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
  663. @router.get('/profile')
  664. async def userlist( #user_id_1: int = Query(None ,description='用户id'),
  665. db: Session = Depends(get_db),
  666. user_id: int = Depends(valid_access_token)):
  667. try:
  668. user = user_id_get_user_info(db,user_id)
  669. # 将查询结果转换为列表形式的字典
  670. roleIds = user_id_get_user_roleIds(db, user_id)
  671. user_roles = role_id_list_get_user_role(db,roleIds)
  672. dept = dept_id_get_dept_info(db,user.dept_id)
  673. user_list = {
  674. "admin":1 in roleIds,
  675. "avatar": user.avatar,
  676. "createBy":user_id_get_user_info(db,user.create_by).user_name,
  677. "createTime": user.create_time.strftime('%Y-%m-%d %H:%M:%S') if user.create_time else '',
  678. "delFlag":user.del_flag,
  679. "dept":{
  680. "createBy": user_id_get_user_info(db,dept.create_by).user_name if user.create_by else None,
  681. "createTime": dept.create_time.strftime('%Y-%m-%d %H:%M:%S') if user.create_time else None,
  682. "updateBy": user_id_get_user_info(db,dept.update_by).user_name if user.update_by else None,
  683. "updateTime": dept.create_time.strftime('%Y-%m-%d %H:%M:%S') if user.create_time else None,
  684. "remark": '',#dept.remark,
  685. "deptId": dept.dept_id,
  686. "parentId": dept.parent_id,
  687. "ancestors": dept.ancestors,
  688. "deptName": dept.dept_name,
  689. "orderNum": dept.order_num,
  690. "leader": dept.leader_name,
  691. "phone": dept.phone,
  692. "email": dept.email,
  693. "status": dept.status,
  694. "delFlag": dept.del_flag,
  695. "parentName": dept.parent_name,
  696. "children": []
  697. },
  698. "deptId": user.dept_id,
  699. "email": user.email,
  700. "loginDate": user.login_date.strftime('%Y-%m-%d %H:%M:%S') if user.login_date else '',
  701. "loginIp": user.login_ip,
  702. "nickName": user.nick_name,
  703. "password":"",
  704. "phonenumber": user.phonenumber,
  705. "postIds": None,
  706. "remark": user.remark,
  707. "roleId": None,
  708. "roleIds": roleIds,
  709. "roles": user_roles,
  710. "sex": user.sex,
  711. "status": user.status,
  712. "updateBy":user.update_by,
  713. "updateTime": user.create_time.strftime('%Y-%m-%d %H:%M:%S') if user.create_time else None,
  714. "userId": user.user_id,
  715. "userName": user.user_name,
  716. "tenantId": user.tenant_id,
  717. "userType": user.user_type,
  718. "deptName": dept.dept_name,
  719. }
  720. roles = get_role(db,roleIds)
  721. # 返回结果
  722. return {
  723. "code": 200,
  724. "msg": "成功用户列表",
  725. "data": user_list,
  726. # {
  727. # "user":user_list,
  728. # "roleIds":roleIds,
  729. # "roles":roles,
  730. # "postIds":[],
  731. # "posts":[]
  732. # }
  733. "postGroup":"工作人员",
  734. "roleGroup":'/'.join([i["roleName"] for i in user_roles])
  735. }
  736. except Exception as e:
  737. traceback.print_exc()
  738. raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
  739. @router.put('/profile')
  740. async def userlist( #user_id_1: int = Query(None ,description='用户id'),
  741. db: Session = Depends(get_db),
  742. body = Depends(remove_xss_json),
  743. user_id: int = Depends(valid_access_token)):
  744. try:
  745. user = user_id_get_user_info(db,user_id)
  746. if "email" in body:
  747. user.email=body['email']
  748. user.update_by=user_id
  749. if "nickName" in body:
  750. user.nick_name = body['nickName']
  751. user.update_by=user_id
  752. if "phonenumber" in body:
  753. user.phonenumber = body['phonenumber']
  754. user.update_by=user_id
  755. if "sex" in body:
  756. user.nick_name = body['sex']
  757. user.update_by=user_id
  758. db.commit()
  759. return {"code":200,"msg":"成功","data":None}
  760. except Exception as e:
  761. traceback.print_exc()
  762. raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
  763. @router.put('/profile/updatePwd')
  764. async def userlist( #user_id_1: int = Query(None ,description='用户id'),
  765. db: Session = Depends(get_db),
  766. body = Depends(remove_xss_json),
  767. user_id: int = Depends(valid_access_token)):
  768. try:
  769. user = user_id_get_user_info(db,user_id)
  770. if user.password!=body['oldPassword']:
  771. return JSONResponse(status_code=404,content={"code":404,"msg":"密码错误"})
  772. if "newPassword" in body:
  773. user.nick_name = body['newPassword']
  774. user.update_by=user_id
  775. db.commit()
  776. return {"code":200,"msg":"成功","data":None}
  777. except Exception as e:
  778. traceback.print_exc()
  779. raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
  780. @router.get('/authRole/{user_id_1}')
  781. @router.get('/{user_id_1}')
  782. async def userlist( user_id_1: int = Query(None ,description='用户id'),
  783. db: Session = Depends(get_db),
  784. user_id: int = Depends(valid_access_token)):
  785. try:
  786. user = user_id_get_user_info(db,user_id_1)
  787. # 将查询结果转换为列表形式的字典
  788. roleIds = user_id_get_user_roleIds(db, user_id_1)
  789. user_roles = role_id_list_get_user_role(db,roleIds)
  790. user_list = {
  791. "userId": user.user_id,
  792. "tenantId": user.tenant_id,
  793. "deptId": user.dept_id,
  794. "userName": user.user_name,
  795. "nickName": user.nick_name,
  796. "userType": user.user_type,
  797. "email": user.email,
  798. "phonenumber": user.phonenumber,
  799. "sex": user.sex,
  800. "avatar": user.avatar,
  801. "status": user.status,
  802. "loginIp": user.login_ip,
  803. "loginDate": user.login_date.strftime('%Y-%m-%d %H:%M:%S') if user.login_date else '',
  804. "remark": user.remark,
  805. "createTime": user.create_time.strftime('%Y-%m-%d %H:%M:%S') if user.create_time else '',
  806. "deptName": user.dept_name,
  807. "roles": user_roles,
  808. "roleIds": roleIds,
  809. "postIds": None,
  810. "roleId": None
  811. }
  812. roles = get_role(db,roleIds)
  813. # 返回结果
  814. return {
  815. "code": 200,
  816. "msg": "成功用户列表",
  817. "data": {
  818. "user":user_list,
  819. "roleIds":roleIds,
  820. "roles":roles,
  821. "postIds":[],
  822. "posts":[]
  823. }
  824. }
  825. except Exception as e:
  826. traceback.print_exc()
  827. raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")