__init__.py 31 KB

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