person.py 35 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961
  1. from fastapi import APIRouter, Request, Depends, HTTPException, Query, BackgroundTasks
  2. from sqlalchemy.exc import IntegrityError
  3. from fastapi.responses import HTMLResponse, FileResponse
  4. from fastapi.responses import JSONResponse
  5. from database import get_db
  6. from sqlalchemy import text, exists, and_, or_, not_
  7. from sqlalchemy.orm import Session
  8. from models import *
  9. import json
  10. import random
  11. import os
  12. from sqlalchemy import create_engine, select
  13. from typing import Optional
  14. from utils.StripTagsHTMLParser import *
  15. from utils.three_proofing_responsible_util import *
  16. from utils.ry_system_util import *
  17. from common.db import db_event_management, db_user, db_area, db_emergency_plan
  18. from common.security import valid_access_token
  19. import traceback
  20. from utils import *
  21. from datetime import datetime, timedelta
  22. import pandas as pd
  23. import xlrd
  24. from common.db import db_dept
  25. from exceptions import AppException, HmacException
  26. from common.enc import mpfun, three_proofing_responsible_person_data
  27. from common.db import db_czrz
  28. from common.auth_user import *
  29. router = APIRouter()
  30. @router.post('/create')
  31. async def create_contact(
  32. request: Request,
  33. auth_user: AuthUser = Depends(find_auth_user),
  34. db: Session = Depends(get_db),
  35. body=Depends(remove_xss_json),
  36. user_id=Depends(valid_access_token)
  37. ):
  38. try:
  39. # 提取请求数据
  40. unit_name = body['unit_name']
  41. name = body['name']
  42. area_code = body['area_code']
  43. area_info = id_get_area_info(db, body['area_code'])
  44. if area_info:
  45. area_code2 = area_info.area_code
  46. position = body['position']
  47. phone = body['phone']
  48. telephone= ''
  49. if 'telephone' in body:
  50. telephone=body['telephone']
  51. order_num = -1
  52. if 'order_num'in body:
  53. order_num=body['order_num']
  54. unit_id = db_dept.get_dept_id_by_name(db, unit_name)
  55. user_id_1 = db_user.get_user_id_by_phonenumber(db,phone)
  56. # 创建新的记录
  57. new_person = ThreeProofingResponsiblePerson(
  58. unit_id=unit_id,
  59. unit_name=unit_name,
  60. name=name,
  61. area_code2=area_code2,
  62. area_code=area_code,
  63. position=position,
  64. phone=phone,
  65. telephone=telephone,
  66. user_id = user_id_1,
  67. order_num=order_num,
  68. create_by=user_id
  69. )
  70. # 添加到数据库会话并提交
  71. type_list = body['type_list']
  72. if isinstance(type_list,list) and len(type_list)>0:
  73. db.add(new_person)
  74. db.commit()
  75. three_proofing_responsible_person_data.sign_table()
  76. else:
  77. return JSONResponse(status_code=404,content={
  78. 'code': 404,
  79. 'msg': '责任类型不能为空'
  80. })
  81. try:
  82. for type_info in type_list:
  83. type_parent_id = type_info['type_parent_id']#get_type_parent_id_by_type_id(db,type_id)
  84. for type_id in type_info['children']:
  85. new_person_type = ThreeProofingResponsiblePersonType(
  86. type_parent_id = type_parent_id,
  87. type_id = type_id,
  88. person_id = new_person.id,
  89. create_by=user_id
  90. )
  91. db.add(new_person_type)
  92. if type_parent_id in ('5','7','9'):
  93. if 'children2' in type_info:
  94. for other_type_id in type_info['children2']:
  95. new_person_other_type = ThreeProofingResponsiblePersonOtherType(
  96. type_parent_id=type_parent_id,
  97. other_type_id=other_type_id,
  98. person_id=new_person.id,
  99. create_by=user_id
  100. )
  101. db.add(new_person_other_type)
  102. if type_parent_id in ('4','5','7','10','11'):
  103. dept_name = None
  104. if 'dept_name' in type_info:
  105. dept_name = type_info['dept_name']
  106. other_type_2_name = None
  107. if 'other_type_2_name' in type_info:
  108. other_type_2_name = type_info['other_type_2_name']
  109. denger_point_name = None
  110. if 'denger_point_name' in type_info:
  111. denger_point_name = type_info['denger_point_name']
  112. new_person_other_info = ThreeProofingResponsiblePersonOtherInfo(
  113. type_parent_id = type_parent_id,
  114. dept_name = dept_name,
  115. other_type_2_name = other_type_2_name,
  116. denger_point_name = denger_point_name,
  117. person_id = new_person.id,
  118. create_by=user_id
  119. )
  120. db.add(new_person_other_info)
  121. except:
  122. db.rollback()
  123. traceback.print_exc()
  124. new_person.del_flag='2'
  125. db.commit()
  126. db_czrz.log(db, auth_user, "系统管理", f"后台管理新建三防责任人管理人员信息成功", request.client.host)
  127. # 返回创建成功的响应
  128. return {
  129. "code": 200,
  130. "msg": "创建成功",
  131. "data": None
  132. }
  133. except Exception as e:
  134. # 处理异常
  135. db.rollback()
  136. traceback.print_exc()
  137. raise HTTPException(status_code=500, detail=str(e))
  138. @router.put('/update')
  139. async def update_contact(
  140. request: Request,
  141. auth_user: AuthUser = Depends(find_auth_user),
  142. db: Session = Depends(get_db),
  143. body=Depends(remove_xss_json),
  144. user_id=Depends(valid_access_token)
  145. ):
  146. try:
  147. # 提取请求数据
  148. id = body['id']
  149. person = get_person_info_by_id(db,id)
  150. if not person:
  151. return JSONResponse(status_code=404, content={
  152. 'errcode': 404,
  153. 'errmsg': '责任人不存在'
  154. })
  155. person.unit_name = body['unit_name']
  156. person.name = body['name']
  157. person.area_code = body['area_code']
  158. area_info = id_get_area_info(db, body['area_code'])
  159. if area_info:
  160. person.area_code2 = area_info.area_code
  161. person.position = body['position']
  162. person.phone = mpfun.enc_data(body['phone'])
  163. if 'telephone' in body:
  164. person.telephone = mpfun.enc_data(body['telephone'])
  165. person.order_num = body['order_num']
  166. if body['order_num']=='':
  167. person.order_num = -1
  168. person.unit_id = db_dept.get_dept_id_by_name(db, body['unit_name'])
  169. person.update_by = user_id
  170. person.update_time = datetime.now()
  171. person.sign = three_proofing_responsible_person_data.get_sign_hmac(person)
  172. type_list = body['type_list']
  173. old_person_type_list=get_person_type_by_person_id(db,person.id)
  174. for old_person_type in old_person_type_list:
  175. old_person_type.del_flag='2'
  176. old_person_other_info_list = get_person_other_info_by_person_id(db,person.id)
  177. for old_person_other_info in old_person_other_info_list:
  178. old_person_other_info.del_flag = '2'
  179. old_person_other_type_list = get_person_other_type_by_person_id(db,person.id)
  180. for old_person_other_type in old_person_other_type_list:
  181. old_person_other_type.del_flag = '2'
  182. for type_info in type_list:
  183. type_parent_id = type_info['type_parent_id'] # get_type_parent_id_by_type_id(db,type_id)
  184. for type_id in type_info['children']:
  185. new_person_type = ThreeProofingResponsiblePersonType(
  186. type_parent_id=type_parent_id,
  187. type_id=type_id,
  188. person_id=id,
  189. create_by=user_id
  190. )
  191. db.add(new_person_type)
  192. if type_parent_id in ('5', '7', '9'):
  193. if 'children2' in type_info:
  194. for other_type_id in type_info['children2']:
  195. new_person_other_type = ThreeProofingResponsiblePersonOtherType(
  196. type_parent_id=type_parent_id,
  197. other_type_id=other_type_id,
  198. person_id=person.id,
  199. create_by=user_id
  200. )
  201. db.add(new_person_other_type)
  202. if type_parent_id in ('4', '5', '7', '10', '11'):
  203. dept_name = None
  204. if 'dept_name' in type_info:
  205. dept_name = type_info['dept_name']
  206. other_type_2_name = None
  207. if 'other_type_2_name' in type_info:
  208. other_type_2_name = type_info['other_type_2_name']
  209. denger_point_name = None
  210. if 'denger_point_name' in type_info:
  211. denger_point_name = type_info['denger_point_name']
  212. new_person_other_info = ThreeProofingResponsiblePersonOtherInfo(
  213. type_parent_id=type_parent_id,
  214. dept_name=dept_name,
  215. other_type_2_name=other_type_2_name,
  216. denger_point_name=denger_point_name,
  217. person_id=person.id,
  218. create_by=user_id
  219. )
  220. db.add(new_person_other_info)
  221. # 更新到数据库会话并提交
  222. db.commit()
  223. db_czrz.log(db, auth_user, "系统管理", f"后台管理更新三防责任人管理人员信息成功", request.client.host)
  224. # 返回创建成功的响应
  225. return {
  226. "code": 200,
  227. "msg": "更新成功",
  228. "data": None
  229. }
  230. except Exception as e:
  231. # 处理异常
  232. db.rollback()
  233. traceback.print_exc()
  234. raise HTTPException(status_code=500, detail=str(e))
  235. @router.get('/list')
  236. async def get_emergency_contact_list(
  237. type_parent_id: str = Query(None, description='单位名称'),
  238. area_code:str = Query(None, description='单位名称'),
  239. Name: str = Query(None, description='联系人'),
  240. page: int = Query(1, gt=0, description='页码'),
  241. pageSize: int = Query(10, gt=0, description='每页条目数量'),
  242. db: Session = Depends(get_db),
  243. user_id=Depends(valid_access_token)
  244. ):
  245. try:
  246. # 构建查询
  247. query = db.query(ThreeProofingResponsiblePerson)
  248. query = query.filter(ThreeProofingResponsiblePerson.del_flag == '0')
  249. # 应用查询条件
  250. if type_parent_id:
  251. person_list = get_person_list_by_type_parent_id(db,type_parent_id)
  252. query = query.filter(ThreeProofingResponsiblePerson.id.in_(person_list))
  253. if Name:
  254. query = query.filter(ThreeProofingResponsiblePerson.name.like(f'%{Name}%'))
  255. def get_area_chli(area_list : list,parent_id : int):
  256. areas = parent_id_get_area_info(db,parent_id)
  257. if areas:
  258. for area in areas:
  259. area_list.append(area.id)
  260. get_area_chli(area_list, area.id)
  261. return area_list
  262. # if area_code:
  263. # query = query.filter(ThreeProofingResponsiblePerson.area_code.in_(get_area_chli([area_code],area_code)))
  264. # '''440900000000'''
  265. if area_code:
  266. area_info = id_get_area_info(db,area_code)
  267. if area_info:
  268. area_code = area_info.area_code
  269. area_code=area_code.replace('0000000000','').replace('00000000','').replace('000000','').replace('000','')
  270. query = query.filter(ThreeProofingResponsiblePerson.area_code2.like(f'%{area_code}%') )
  271. # if area_code:
  272. # query = query.filter(ThreeProofingResponsiblePerson.area_code==area_code)
  273. # 计算总条目数
  274. total_items = query.count()
  275. # 排序
  276. query = query.order_by(ThreeProofingResponsiblePerson.order_num.asc())
  277. query = query.order_by(ThreeProofingResponsiblePerson.create_time.desc())
  278. # 执行分页查询
  279. contact_infos = query.offset((page - 1) * pageSize).limit(pageSize).all()
  280. # 将查询结果转换为列表形式的字典
  281. contact_infos_list = []
  282. for info in contact_infos:
  283. type_parent_id_list = get_type_parent_id_by_person_id(db,info.id)
  284. type_parent_list = []
  285. type_parent_list2 = []
  286. for type_parent in type_parent_id_list:
  287. if type_parent not in type_parent_list2:
  288. dict_data = get_dict_data_info(db,'three_proofing',type_parent)
  289. type_parent_list2.append(type_parent)
  290. type_parent_list.append({"type_parent_id":type_parent,"type_parent":dict_data.dict_label})
  291. area_info = id_get_area_info(db,info.area_code)
  292. user_info = user_id_get_user_info(db,info.create_by)
  293. area_list = db_area.id_get_area_parent_list(db,info.area_code,[])
  294. if info.order_num ==-1:
  295. order_num = ''
  296. else:
  297. order_num = str(info.order_num)
  298. contact_infos_list.append({
  299. "id": info.id,
  300. "unit_id": info.unit_id,
  301. "unit_name": info.unit_name,
  302. "name": info.name,
  303. "area_list":area_list,
  304. "area_code": info.area_code,
  305. "area_name": area_info.area_name,
  306. "position": info.position,
  307. "phone": mpfun.dec_data(info.phone),
  308. "telephone": mpfun.dec_data(info.telephone),
  309. "order_num": order_num,
  310. "online_status":'0',
  311. "create_time": info.create_time.strftime('%Y-%m-%d %H:%M:%S'),
  312. "create_by":info.create_by,
  313. "create_user":user_info.nick_name,
  314. "type_parent_list":type_parent_list
  315. })
  316. # 返回结果+
  317. return {
  318. "code": 200,
  319. "msg": "成功",
  320. "data": contact_infos_list,
  321. "total": total_items
  322. }
  323. except Exception as e:
  324. # 处理异常
  325. traceback.print_exc()
  326. raise HTTPException(status_code=500, detail=str(e))
  327. @router.get('/info/{id}')
  328. async def get_emergency_contact_id_info(
  329. id: str,
  330. db: Session = Depends(get_db),
  331. user_id=Depends(valid_access_token)
  332. ):
  333. try:
  334. contact = get_person_info_by_id(db,id)
  335. if not contact:
  336. return JSONResponse(status_code=404, content={
  337. 'errcode': 404,
  338. 'errmsg': '联系人不存在'
  339. })
  340. # 将查询结果转换为列表形式的字典
  341. area_info = id_get_area_info(db,contact.area_code)
  342. user_info = user_id_get_user_info(db,contact.create_by)
  343. area_list = db_area.id_get_area_parent_list(db, contact.area_code, [])
  344. if contact.order_num == -1:
  345. order_num = ''
  346. else:
  347. order_num = str(contact.order_num)
  348. contact_result = {
  349. "id": contact.id,
  350. "unit_id": contact.unit_id,
  351. "unit_name": contact.unit_name,
  352. "name": contact.name,
  353. "area_list":area_list,
  354. "area_code":contact.area_code,
  355. "area_name": area_info.area_name,
  356. "position": contact.position,
  357. "phone": mpfun.dec_data(contact.phone),
  358. "telephone":mpfun.dec_data(contact.telephone),
  359. "order_num":order_num,
  360. "online_status":'0',
  361. "create_time": contact.create_time.strftime('%Y-%m-%d %H:%M:%S'),
  362. "create_by":contact.create_by,
  363. "create_user":user_info.nick_name,
  364. "type_list":[]
  365. }
  366. type_parent_id_list = []
  367. type_list = get_person_type_by_person_id(db,contact.id)
  368. for type_info in type_list:
  369. if type_info.type_parent_id not in type_parent_id_list:
  370. type_parent_id_list.append(type_info.type_parent_id)
  371. for type_parent_id in type_parent_id_list:
  372. dict_data = get_dict_data_info(db, 'three_proofing', type_parent_id)
  373. type_data = {"type_parent_id": type_parent_id, "type_parent": dict_data.dict_label,"children":[],"labelData":[]}
  374. for type_info in get_person_type_by_person_id_and_type_parent_id(db,contact.id,type_parent_id):
  375. type_data_info = get_type_info_by_id(db,type_info.type_id)
  376. # type_data['children'].append({"id": type_info.type_id, "label": type_data_info.type_name})
  377. type_data['children'].append(type_info.type_id)
  378. type_data['labelData'].append( type_data_info.type_name)
  379. other_info = get_person_other_info_by_person_id_and_type_parent_id(db, contact.id, type_parent_id)
  380. if other_info:
  381. type_data['dept_name'] = other_info.dept_name
  382. type_data['denger_point_name'] = other_info.denger_point_name
  383. type_data['other_type_2_name'] = other_info.other_type_2_name
  384. other_type_list = get_person_other_type_by_person_id_and_type_parent_id(db, contact.id, type_parent_id)
  385. if other_type_list:
  386. type_data['children2'] = []
  387. type_data['other_type_label'] = []
  388. for other_type in other_type_list:
  389. other_type_info = get_other_type_info_by_id(db, other_type.other_type_id)
  390. label= ''
  391. if other_type_info:
  392. label = other_type_info.type_name
  393. # type_data['children2'].append({"id": other_type.other_type_id, "label": label})
  394. type_data['children2'].append(other_type.other_type_id)
  395. type_data['other_type_label'].append(label)
  396. contact_result['type_list'].append(type_data)
  397. # 返回结果
  398. return {
  399. "code": 200,
  400. "msg": "成功",
  401. "data": contact_result
  402. }
  403. except Exception as e:
  404. # 处理异常
  405. traceback.print_exc()
  406. raise HTTPException(status_code=500, detail=str(e))
  407. @router.delete('/delete')
  408. async def delete_emergency_plans(
  409. request: Request,
  410. ids: list,
  411. db: Session = Depends(get_db),
  412. body=Depends(remove_xss_json),
  413. auth_user: AuthUser = Depends(find_auth_user),
  414. user_id=Depends(valid_access_token)
  415. ):
  416. try:
  417. # 提取请求数据
  418. query = db.query(ThreeProofingResponsiblePerson)
  419. query = query.filter(ThreeProofingResponsiblePerson.del_flag != '2')
  420. query = query.filter(ThreeProofingResponsiblePerson.id.in_(ids))
  421. contacts = query.all()
  422. if not contacts:
  423. return JSONResponse(status_code=404, content={
  424. 'errcode': 404,
  425. 'errmsg': '联系人不存在'
  426. })
  427. for contact in contacts:
  428. contact.del_flag = '2'
  429. contact.update_by = user_id
  430. contact.update_time = datetime.now()
  431. contact.sign = three_proofing_responsible_person_data.get_sign_hmac(contact)
  432. # 更新到数据库会话并提交
  433. db.commit()
  434. db_czrz.log(db, auth_user, "系统管理", f"后台管理删除三防责任人管理人员信息成功", request.client.host)
  435. # 返回创建成功的响应
  436. return {
  437. "code": 200,
  438. "msg": "删除成功",
  439. "data": None
  440. }
  441. except Exception as e:
  442. # 处理异常
  443. traceback.print_exc()
  444. raise HTTPException(status_code=500, detail=str(e))
  445. @router.delete('/delete/{id}')
  446. async def delete_emergency_plans(
  447. request: Request,
  448. id: int,
  449. db: Session = Depends(get_db),
  450. body=Depends(remove_xss_json),
  451. auth_user: AuthUser = Depends(find_auth_user),
  452. user_id=Depends(valid_access_token)
  453. ):
  454. try:
  455. contact = get_person_info_by_id(db, id)
  456. if not contact:
  457. return JSONResponse(status_code=404, content={
  458. 'errcode': 404,
  459. 'errmsg': '联系人不存在'
  460. })
  461. contact.del_flag = '2'
  462. contact.update_by = user_id
  463. contact.update_time = datetime.now()
  464. contact.sign = three_proofing_responsible_person_data.get_sign_hmac(contact)
  465. # 更新到数据库会话并提交
  466. db.commit()
  467. db_czrz.log(db, auth_user, "系统管理", f"后台管理删除三防责任人管理人员信息成功", request.client.host)
  468. # 返回创建成功的响应
  469. return {
  470. "code": 200,
  471. "msg": "删除成功",
  472. "data": None
  473. }
  474. except Exception as e:
  475. # 处理异常
  476. traceback.print_exc()
  477. raise HTTPException(status_code=500, detail=str(e))
  478. def string_type_parent_id_create_data(db,string,type_parent_id,file_info,new_person,user_id,row) :
  479. type_name_list = [i for i in string.split(',')]
  480. reslte = []
  481. for type_name in type_name_list:
  482. type_id = get_type_id_by_type_parent_id_and_type_name(db, type_parent_id, type_name)
  483. if type_id:
  484. new_person_type = ThreeProofingResponsiblePersonType(
  485. type_parent_id=type_parent_id,
  486. type_id=type_id,
  487. person_id=new_person.id,
  488. create_by=user_id
  489. )
  490. reslte.append(new_person_type)
  491. else:
  492. file_info.remark= file_info.remark+f'\n行<{row+1}>责任类别未找到<{type_name}>'
  493. return reslte ,False
  494. return reslte ,True
  495. def other_type_string_type_parent_id_create_data(db,string,type_parent_id,file_info,new_person,user_id,row) :
  496. type_name_list = [i for i in string.split(',')]
  497. reslte = []
  498. for other_type_name in type_name_list:
  499. other_type_id = get_other_type_id_by_type_parent_id_and_other_type_name(db, type_parent_id, other_type_name)
  500. if other_type_id:
  501. new_person_type = ThreeProofingResponsiblePersonOtherType(
  502. type_parent_id=type_parent_id,
  503. other_type_id=other_type_id,
  504. person_id=new_person.id,
  505. create_by=user_id
  506. )
  507. reslte.append(new_person_type)
  508. else:
  509. file_info.remark= file_info.remark+f'\n行<{row+1}>责任类别未找到<{other_type_name}>'
  510. file_info.error_num +=1
  511. return reslte ,False
  512. return reslte ,True
  513. def import_data(db,file_path,user_id,file_info):
  514. import_status = True
  515. try:
  516. book = xlrd.open_workbook(file_path)
  517. sheet = book.sheet_by_index(0)
  518. except :
  519. file_info.remark = file_info.remark + f'\n文件打开失败,请核实文件格式为xlsx/xlx>'
  520. file_info.error_num +=1
  521. import_status = False
  522. data = []
  523. for row in range(4, sheet.nrows):
  524. # 姓名
  525. name = sheet.cell(row, 0).value
  526. if name == '':
  527. file_info.remark = file_info.remark+f'\n行<{row+1}>姓名不能为空<{name}>'
  528. file_info.error_num +=1
  529. import_status = False
  530. # 所属单位
  531. unit_name = sheet.cell(row, 1).value
  532. if unit_name == '':
  533. file_info.remark = file_info.remark+f'\n行<{row+1}>所属单位不能为空<{unit_name}>'
  534. file_info.error_num +=1
  535. import_status = False
  536. unit_id = db_dept.get_dept_id_by_name(db, unit_name)
  537. # 职务
  538. position = sheet.cell(row, 2).value
  539. if position =='':
  540. file_info.remark = file_info.remark+f'\n行<{row+1}>职务不能为空<{position}>'
  541. file_info.error_num +=1
  542. import_status = False
  543. # 电话号码(如有多个手机号请用“,”分隔)
  544. phone = str(sheet.cell(row, 3).value)
  545. if phone =='':
  546. file_info.remark = file_info.remark+f'\n行<{row+1}>电话号码不能为空<{phone}>'
  547. file_info.error_num +=1
  548. import_status = False
  549. phone_list = [i for i in phone.split(',')]
  550. user_id_1=-1
  551. for i in phone_list:
  552. user_id_1 = db_user.get_user_id_by_phonenumber(db, i)
  553. if user_id_1 != -1:
  554. break
  555. # 办公电话
  556. # (选填,格式:区号-电话号码)
  557. telephone = sheet.cell(row, 4).value
  558. # 排位顺序
  559. # (选填,请输入排序号1-9999,排序号数值越小越靠前,不填则默认排至最末)
  560. order_num = sheet.cell(row, 5).value
  561. if order_num == '':
  562. order_num=-1
  563. area_name = sheet.cell(row, 7).value
  564. if area_name=='':
  565. area_code=2
  566. area_code2='440900000000'
  567. else:
  568. area_code=get_area_info_by_area_name(db,area_name)
  569. if area_code is None:
  570. file_info.remark = file_info.remark+f'\n行<{row+1}>责任区域未找到'
  571. file_info.error_num +=1
  572. import_status = False
  573. area_code2 = area_code.area_code
  574. area_code = area_code.id
  575. new_person = ThreeProofingResponsiblePerson(
  576. unit_id=unit_id,
  577. unit_name=unit_name,
  578. name=name,
  579. area_code2=area_code2,
  580. area_code=area_code,
  581. position=position,
  582. phone=phone,
  583. telephone=telephone,
  584. user_id=user_id_1,
  585. order_num=order_num,
  586. create_by=user_id,
  587. sign = ''
  588. )
  589. data.append(new_person)
  590. db.add(new_person)
  591. db.commit()
  592. # 党委政府
  593. a1 = sheet.cell(row, 6).value
  594. if a1 != '':
  595. new_type_list,status = string_type_parent_id_create_data(db,a1,'1',file_info,new_person,user_id,row)
  596. if status:
  597. db.add_all(new_type_list)
  598. data +=new_type_list
  599. else:
  600. import_status = status
  601. # type_name_list = [i for i in a1.split(',')]
  602. # for type_name in type_name_list:
  603. # type_id = get_type_id_by_type_parent_id_and_type_name(db, '1', type_name)
  604. # if type_id:
  605. # pass
  606. # 三防指挥部
  607. b1 = sheet.cell(row, 8).value
  608. if b1 != '':
  609. new_type_list,status = string_type_parent_id_create_data(db,b1,'2',file_info,new_person,user_id,row)
  610. if status:
  611. db.add_all(new_type_list)
  612. data +=new_type_list
  613. else:
  614. import_status = status
  615. b2 = sheet.cell(row, 9).value
  616. # 应急部门
  617. c1 = sheet.cell(row, 10).value
  618. if c1!='':
  619. new_type_list, status = string_type_parent_id_create_data(db, c1, '3', file_info, new_person, user_id,row)
  620. if status:
  621. db.add_all(new_type_list)
  622. data += new_type_list
  623. else:
  624. import_status = status
  625. # 成员单位
  626. d1 = sheet.cell(row, 11).value
  627. if d1!='':
  628. new_type_list, status = string_type_parent_id_create_data(db, d1, '4', file_info, new_person, user_id,row)
  629. if status:
  630. db.add_all(new_type_list)
  631. data += new_type_list
  632. else:
  633. import_status = status
  634. d2 = sheet.cell(row, 12).value
  635. if d2!='':
  636. dept_name = d2
  637. new_person_other_info = ThreeProofingResponsiblePersonOtherInfo(
  638. type_parent_id='4',
  639. dept_name=dept_name,
  640. person_id=new_person.id,
  641. create_by=user_id
  642. )
  643. db.add(new_person_other_info)
  644. data.append(new_person_other_info)
  645. # 重点部门
  646. e1 = sheet.cell(row, 13).value
  647. if e1!='':
  648. new_type_list, status = string_type_parent_id_create_data(db, e1, '5', file_info, new_person, user_id,row)
  649. if status:
  650. db.add_all(new_type_list)
  651. data += new_type_list
  652. else:
  653. import_status = status
  654. e2 = sheet.cell(row, 14).value
  655. if e2!='':
  656. dept_name = e2
  657. new_person_other_info = ThreeProofingResponsiblePersonOtherInfo(
  658. type_parent_id='5',
  659. dept_name=dept_name,
  660. person_id=new_person.id,
  661. create_by=user_id
  662. )
  663. db.add(new_person_other_info)
  664. data.append(new_person_other_info)
  665. e3 = sheet.cell(row, 15).value
  666. if e3!='':
  667. new_type_list,status = other_type_string_type_parent_id_create_data(db,e3,'5',file_info,new_person,user_id,row)
  668. if status:
  669. db.add_all(new_type_list)
  670. data +=new_type_list
  671. else:
  672. import_status = status
  673. # 行政村
  674. f1 = sheet.cell(row, 16).value
  675. if f1!='':
  676. new_type_list, status = string_type_parent_id_create_data(db, f1, '6', file_info, new_person, user_id,row)
  677. if status:
  678. db.add_all(new_type_list)
  679. data += new_type_list
  680. else:
  681. import_status = status
  682. # 水利工程
  683. g1 = sheet.cell(row, 17).value
  684. if g1!='':
  685. new_type_list, status = string_type_parent_id_create_data(db, g1, '7', file_info, new_person, user_id,row)
  686. if status:
  687. db.add_all(new_type_list)
  688. data += new_type_list
  689. else:
  690. import_status = status
  691. g2 = sheet.cell(row, 18).value
  692. if g2!='':
  693. dept_name = g2
  694. new_person_other_info = ThreeProofingResponsiblePersonOtherInfo(
  695. type_parent_id='7',
  696. dept_name=dept_name,
  697. person_id=new_person.id,
  698. create_by=user_id
  699. )
  700. db.add(new_person_other_info)
  701. data.append(new_person_other_info)
  702. g3 = sheet.cell(row, 19).value
  703. if g3!='':
  704. new_type_list, status = other_type_string_type_parent_id_create_data(db, g3, '11', file_info, new_person, user_id,
  705. row)
  706. if status:
  707. db.add_all(new_type_list)
  708. data += new_type_list
  709. else:
  710. import_status = status
  711. # 受威胁转移
  712. h1 = sheet.cell(row, 20).value
  713. if h1!='':
  714. new_type_list, status = string_type_parent_id_create_data(db, h1, '8', file_info, new_person, user_id,row)
  715. if status:
  716. db.add_all(new_type_list)
  717. data += new_type_list
  718. else:
  719. import_status = status
  720. # 抢险队伍
  721. j1 = sheet.cell(row, 21).value
  722. if j1!='':
  723. new_type_list, status = string_type_parent_id_create_data(db, j1, '9', file_info, new_person, user_id,row)
  724. if status:
  725. db.add_all(new_type_list)
  726. data += new_type_list
  727. else:
  728. import_status = status
  729. j2 = sheet.cell(row, 22).value
  730. if j2!='':
  731. new_type_list, status = other_type_string_type_parent_id_create_data(db, j2, '9', file_info, new_person, user_id,
  732. row)
  733. if status:
  734. db.add_all(new_type_list)
  735. data += new_type_list
  736. else:
  737. import_status = status
  738. # 地质灾害
  739. k1 = sheet.cell(row, 23).value
  740. if k1!='':
  741. new_type_list, status = string_type_parent_id_create_data(db, k1, '10', file_info, new_person, user_id,row)
  742. if status:
  743. db.add_all(new_type_list)
  744. data += new_type_list
  745. else:
  746. import_status = status
  747. k2 = sheet.cell(row, 24).value
  748. if k2!='':
  749. denger_point_name = k2
  750. new_person_other_info = ThreeProofingResponsiblePersonOtherInfo(
  751. type_parent_id='10',
  752. denger_point_name=denger_point_name,
  753. person_id=new_person.id,
  754. create_by=user_id
  755. )
  756. db.add(new_person_other_info)
  757. data.append(new_person_other_info)
  758. # 其他
  759. l1 = sheet.cell(row, 25).value
  760. if l1!='':
  761. new_type_list, status = string_type_parent_id_create_data(db, l1, '11', file_info, new_person, user_id,row)
  762. if status:
  763. db.add_all(new_type_list)
  764. data += new_type_list
  765. else:
  766. import_status = status
  767. l2 = sheet.cell(row, 26).value
  768. if l2!='':
  769. other_type_2_name = l2
  770. new_person_other_info = ThreeProofingResponsiblePersonOtherInfo(
  771. type_parent_id='11',
  772. other_type_2_name=other_type_2_name,
  773. person_id=new_person.id,
  774. create_by=user_id
  775. )
  776. db.add(new_person_other_info)
  777. data.append(new_person_other_info)
  778. db.commit()
  779. if import_status == False:
  780. # print(1111111)
  781. for info in data:
  782. db.delete(info)
  783. file_info.status = 2
  784. db.commit()
  785. three_proofing_responsible_person_data.sign_table()
  786. @router.post('/createImport')
  787. async def create_contact(
  788. request: Request,
  789. background_tasks: BackgroundTasks,
  790. db: Session = Depends(get_db),
  791. body=Depends(remove_xss_json),
  792. auth_user: AuthUser = Depends(find_auth_user),
  793. user_id=Depends(valid_access_token)
  794. ):
  795. try:
  796. # 提取请求数据
  797. filename = body['filename']
  798. file_name_desc = body['file_name_desc']
  799. if len(filename) == 0:
  800. raise Exception()
  801. file_path = f'/data/upload/mergefile/uploads/{filename}'
  802. # 检查文件是否存在
  803. if not os.path.isfile(file_path):
  804. return JSONResponse(status_code=404, content={
  805. 'errcode': 404,
  806. 'errmsg': f'{filename}不存在'
  807. })
  808. new_file = ThreeProofingResponsiblePersonImportFileStatus(
  809. file_uuid=filename,
  810. file_name = file_name_desc,
  811. status = '1',
  812. remark = '',
  813. user_id=user_id
  814. )
  815. db.add(new_file)
  816. db.commit()
  817. background_tasks.add_task(import_data,db,file_path, user_id,new_file)
  818. db_czrz.log(db, auth_user, "系统管理", f"后台管理导入三防责任人管理人员信息成功", request.client.host)
  819. # 返回创建成功的响应
  820. return {
  821. "code": 200,
  822. "msg": "成功",
  823. "data": None
  824. }
  825. except AppException as e:
  826. return {
  827. "code": 500,
  828. "msg": e.msg
  829. }
  830. except Exception as e:
  831. traceback.print_exc()
  832. # 处理异常
  833. db.rollback()
  834. raise HTTPException(status_code=500, detail=str(e))
  835. @router.get("/export")
  836. async def download_file(filename: str,filenameDesc: str = None):
  837. """
  838. 根据提供的文件名下载文件。
  839. :param filename: 要下载的文件的名称。
  840. """
  841. try:
  842. # 构造文件的完整路径
  843. file_path = os.path.join('', 'uploads/', filename)
  844. # 检查文件是否存在
  845. if not os.path.isfile(file_path):
  846. raise HTTPException(status_code=404, detail="文件未找到")
  847. if not filenameDesc:
  848. filenameDesc = filename
  849. # 设置文件头部和内容类型
  850. headers = {
  851. 'Content-Disposition': f'attachment; filename={filenameDesc}'
  852. }
  853. # 使用FileResponse返回文件流
  854. return FileResponse(
  855. path=file_path,
  856. headers=headers,
  857. media_type='application/octet-stream' # 可以按需更改为适当的MIME类型
  858. )
  859. except HTTPException as e:
  860. raise e
  861. except Exception as e:
  862. # 处理其他异常情况
  863. raise HTTPException(status_code=500, detail=str(e))