__init__.py 30 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886
  1. from fastapi import FastAPI, HTTPException, Depends, APIRouter,Query,Body
  2. from sqlalchemy.engine.reflection import Inspector
  3. from common.security import valid_access_token
  4. from pydantic import BaseModel,Extra,Field
  5. from datetime import datetime
  6. from typing import List, Optional,Any,Dict
  7. from sqlalchemy import create_engine, Column, Integer, String, Boolean, MetaData, Table, inspect, exists,or_,text,insert
  8. from sqlalchemy.orm import Session
  9. from pypinyin import lazy_pinyin, Style
  10. from database import get_db
  11. from models import *
  12. import random
  13. router = APIRouter()
  14. metadata = MetaData()
  15. class ReportField(BaseModel):
  16. name: str
  17. def get_next_event_id(db: Session):
  18. while True:
  19. random_10_digit_number = random.randint(1000000000, 9999999999)
  20. reportId = 'report' + str(random_10_digit_number)
  21. it_exists = db.query(
  22. exists().where(ReportManagement.report_id == reportId)
  23. ).scalar()
  24. if it_exists == False:
  25. return reportId
  26. # 函数用于将中文转换为拼音首字母缩写
  27. def to_first_letter(chinese_str: str) -> str:
  28. return ''.join([p[0][0] for p in lazy_pinyin(chinese_str, style=Style.FIRST_LETTER)]).lower()
  29. # class ReportQuery(BaseModel):
  30. # report_id: str
  31. class TableStructure(BaseModel):
  32. column_name: str
  33. comment: str
  34. #查看详情
  35. @router.get("/report_structure/{report_id}")
  36. async def get_report_structure(
  37. report_id: str,
  38. db: Session = Depends(get_db),
  39. creator_id = Depends(valid_access_token)
  40. ):
  41. # 查询 ReportManagement 表以获取 data_table_name
  42. report = db.query(ReportManagement).filter(ReportManagement.report_id == report_id).first()
  43. if not report:
  44. raise HTTPException(status_code=404, detail="Report not found")
  45. data_table_name = report.data_table_name
  46. # 查询对应表的表结构
  47. table_structure_query = db.execute(
  48. text("""
  49. SELECT COLUMN_NAME, COLUMN_COMMENT
  50. FROM INFORMATION_SCHEMA.COLUMNS
  51. WHERE TABLE_NAME = :table_name AND TABLE_SCHEMA = (SELECT DATABASE())
  52. """),
  53. {"table_name": data_table_name}
  54. )
  55. table_structures = []
  56. for row in table_structure_query.fetchall():
  57. if row[0] not in ['collect_status','create_id','id','user_id']:
  58. table_structures.append(TableStructure(column_name=row[0], comment=row[1]))
  59. # # 查询 user_id 去重后的数量
  60. # distinct_user_count_query = db.execute(
  61. # text("""
  62. # SELECT COUNT(DISTINCT user_id) AS distinct_user_count
  63. # FROM {}
  64. # """.format(data_table_name))
  65. # )
  66. # num_reported = distinct_user_count_query.scalar()
  67. # num_unreported = report.num_reporters - num_reported
  68. # 构造返回结果,包括 ReportManagement 表中的记录和其他相关信息
  69. # 计算去重用户数量
  70. distinct_users = db.query(FormSubmission.user_id).filter(
  71. FormSubmission.report_id == report_id).distinct().count()
  72. # 计算已填报数量(填报结果为1)
  73. num_reported = db.query(FormSubmission).filter(FormSubmission.report_id == report_id,
  74. FormSubmission.submission_status == 1).count()
  75. # 计算未填报数量(填报结果为0)
  76. num_unreported = db.query(FormSubmission).filter(FormSubmission.report_id == report_id,
  77. FormSubmission.submission_status == 0).count()
  78. print("xx")
  79. print(distinct_users,num_reported,num_unreported)
  80. result = {
  81. "code": 200,
  82. 'msg': '查询成功',
  83. 'report_info': {
  84. "id": report.id,
  85. "report_id": report.report_id,
  86. "table_name": report.table_name,
  87. "data_table_name": report.data_table_name,
  88. "start_time": report.start_time,
  89. "end_time": report.end_time,
  90. "status": report.status,
  91. "issued_status": report.issued_status,
  92. "period_type": report.period_type,
  93. "creator_name": report.creator_name,
  94. "num_reporters": distinct_users,
  95. "creator_id": creator_id,
  96. "created_at": report.created_at,
  97. "updated_at": report.updated_at,
  98. "num_reported":num_reported,
  99. "num_unreported":num_unreported
  100. },
  101. 'table_structure': table_structures
  102. }
  103. return result
  104. # 动态创建表
  105. def create_dynamic_table(table_name: str, field_names: List[str], db: Session):
  106. inspector = inspect(db.bind)
  107. # 检查表是否已存在
  108. if inspector.has_table(table_name):
  109. raise HTTPException(status_code=400, detail="Table already exists")
  110. table = Table(table_name, metadata,
  111. Column('id', Integer, primary_key=True),
  112. Column('user_id', Integer),
  113. Column('create_id', Integer),
  114. Column('collect_status', Boolean)
  115. )
  116. existing_columns = set()
  117. for field_name in field_names:
  118. column_name = to_first_letter(field_name)
  119. # 如果列名已存在,则添加一个唯一的后缀
  120. unique_column_name = column_name
  121. suffix = 1
  122. while unique_column_name in existing_columns:
  123. unique_column_name = f"{column_name}_{suffix}"
  124. suffix += 1
  125. existing_columns.add(unique_column_name)
  126. table.append_column(Column(unique_column_name, String(255), comment=field_name))
  127. # 创建表
  128. metadata.create_all(bind=db.bind)
  129. class ReportCreate(BaseModel):
  130. table_name: str
  131. #start_time: str
  132. end_time: str
  133. status: str
  134. issued_status: str
  135. period_type: str
  136. creator_name: str
  137. # creator_id: int
  138. creator_phone:str
  139. field_names: List[str] # 用户只传递字段名称
  140. user_ids: List[int]
  141. # 新建填报和创建新表的接口
  142. @router.post("/report/")
  143. def create_report_and_table(report: ReportCreate, db: Session = Depends(get_db),
  144. creator_id = Depends(valid_access_token)
  145. ):
  146. try:
  147. # 获取当前时间并格式化为 YYYYMMDDHHMMSS
  148. current_time_str = datetime.now().strftime("%Y%m%d%H%M%S")
  149. # 动态生成 data_table_name
  150. table_name_pinyin=''
  151. for i in range(len(lazy_pinyin(report.table_name, style=Style.FIRST_LETTER))):
  152. table_name_pinyin += ''.join(lazy_pinyin(report.table_name, style=Style.FIRST_LETTER)[i]).lower()
  153. data_table_name = f"{table_name_pinyin}_{current_time_str}"
  154. # 动态创建新表
  155. create_dynamic_table(data_table_name, report.field_names, db)
  156. # 登记填报管理
  157. new_report = ReportManagement(
  158. report_id=get_next_event_id(db),
  159. table_name=report.table_name,
  160. data_table_name=data_table_name,
  161. start_time=datetime.now(),
  162. end_time=report.end_time,
  163. status=report.status,
  164. issued_status=report.issued_status,
  165. collection_status=0,#未收取
  166. period_type=report.period_type,
  167. creator_name=report.creator_name,
  168. creator_id=creator_id,
  169. creator_phone=report.creator_phone,
  170. num_reporters = len(report.user_ids)
  171. )
  172. db.add(new_report)
  173. db.commit()
  174. db.refresh(new_report)
  175. # 为每个用户创建填报记录
  176. for user_id in report.user_ids:
  177. submission = FormSubmission(
  178. report_id=new_report.report_id ,
  179. user_id=user_id,
  180. submission_status=0 # 默认状态为未填报
  181. )
  182. db.add(submission)
  183. db.commit()
  184. return {
  185. "code": 200,
  186. "msg":"创建成功"
  187. }
  188. except Exception as e:
  189. raise HTTPException(status_code=400, detail=str(e))
  190. # 定义请求体的 Pydantic 模型
  191. class ReportQuery(BaseModel):
  192. table_name: Optional[str] = Field(None, description="Table name filter")
  193. status: Optional[str] = Field(None, description="Status filter (comma-separated values, e.g., '1,2,3')")
  194. start_time: Optional[datetime] = Field(None, description="Start time filter")
  195. end_time: Optional[datetime] = Field(None, description="End time filter")
  196. issued_status: Optional[str] = Field(None, description="Issued status filter (comma-separated values, e.g., '0,1')")
  197. page: int = Field(1, gt=0, description="Page number for pagination")
  198. pageSize: int = Field(10, gt=0, description="Page size for pagination")
  199. @router.post("/select")
  200. async def select_report(
  201. db: Session = Depends(get_db),
  202. query: ReportQuery = Body(..., description="Report query parameters in the request body"),
  203. creator_id: str = Depends(valid_access_token)
  204. ):
  205. # 构建查询
  206. data_query = db.query(ReportManagement).filter(ReportManagement.creator_id == creator_id)
  207. # 应用过滤条件
  208. if query.table_name:
  209. data_query = data_query.filter(ReportManagement.table_name.ilike(f"%{query.table_name}%"))
  210. if query.start_time and query.end_time:
  211. data_query = data_query.filter(
  212. ReportManagement.start_time >= query.start_time,
  213. ReportManagement.end_time <= query.end_time
  214. )
  215. # 处理 status 和 issued_status 的逗号分隔字符串
  216. if query.status:
  217. status_list = [int(s) for s in query.status.split(",")]
  218. data_query = data_query.filter(ReportManagement.status.in_(status_list))
  219. if query.issued_status:
  220. issued_status_list = [int(s) for s in query.issued_status.split(",")]
  221. data_query = data_query.filter(ReportManagement.issued_status.in_(issued_status_list))
  222. # 计算总数
  223. total_count = data_query.count()
  224. # 分页查询
  225. offset = (query.page - 1) * query.pageSize
  226. data = data_query.offset(offset).limit(query.pageSize).all()
  227. # 构造结果
  228. result_items = []
  229. for item in data:
  230. result_item = {
  231. "id": item.id,
  232. "report_id": item.report_id,
  233. "table_name": item.table_name,
  234. "data_table_name": item.data_table_name,
  235. "start_time": item.start_time,
  236. "end_time": item.end_time,
  237. "status": item.status,
  238. "issued_status": item.issued_status,
  239. "period_type": item.period_type,
  240. "creator_name": item.creator_name,
  241. "creator_id": creator_id,
  242. "created_at": item.created_at,
  243. "creator_phone": item.creator_phone,
  244. "updated_at": item.updated_at,
  245. "num_reporters": item.num_reporters
  246. }
  247. result_items.append(result_item)
  248. result = {
  249. "code": 200,
  250. "msg": "查询成功",
  251. "total": total_count,
  252. "pages": (total_count + query.pageSize - 1) // query.pageSize,
  253. "currentPage": query.page,
  254. "pageSize": query.pageSize,
  255. "data": result_items
  256. }
  257. return result
  258. # class ReportUpdate(BaseModel):
  259. # table_name: str = None
  260. # status: int = None
  261. # # issued_status: int = None
  262. # period_type: str = None
  263. # end_time: str = None
  264. # comments: dict = None # 字典,键为字段名,值为新的备注
  265. # new_fields: List[dict] = None
  266. class ReportUpdate(BaseModel):
  267. table_name: Optional[str] = None
  268. status: Optional[int] = None
  269. period_type: Optional[str] = None
  270. end_time: Optional[str] = None
  271. comments: Optional[Dict[str, str]] = None
  272. new_fields: Optional[List[Dict[str, str]]] = None
  273. class Config:
  274. extra = 'allow'
  275. # 动态添加字段到表
  276. def add_column_if_not_exists(table_name: str, column_name: str, column_type: str, comment: str, db: Session):
  277. inspector = inspect(db.bind)
  278. columns = inspector.get_columns(table_name)
  279. existing_column_names = {column['name'] for column in columns}
  280. if column_name not in existing_column_names:
  281. try:
  282. db.execute(
  283. text(f"""
  284. ALTER TABLE {table_name} ADD COLUMN {column_name} {column_type} COMMENT :comment
  285. """),
  286. {"comment": comment}
  287. )
  288. except Exception as e:
  289. db.rollback()
  290. raise HTTPException(status_code=400, detail=str(e))
  291. #修改
  292. @router.put("/report/{report_id}/")
  293. async def update_report(
  294. report_id: str,
  295. update_data: ReportUpdate,
  296. db: Session = Depends(get_db),
  297. creator_id = Depends(valid_access_token)
  298. ):
  299. # creator_id = '1' # 假设creator_id已经通过某种方式验证
  300. report = db.query(ReportManagement).filter(ReportManagement.report_id == report_id).first()
  301. if not report:
  302. raise HTTPException(status_code=404, detail="Report not found")
  303. if report.creator_id != creator_id:
  304. raise HTTPException(status_code=403, detail="没有权限更新此报告")
  305. # 更新字段
  306. if update_data.table_name:
  307. report.table_name = update_data.table_name
  308. if update_data.status is not None:
  309. report.status = update_data.status
  310. if update_data.period_type:
  311. report.period_type = update_data.period_type
  312. if update_data.end_time:
  313. report.end_time = datetime.fromisoformat(update_data.end_time)
  314. # 更新字段备注
  315. if update_data.comments:
  316. for column_name, comment in update_data.comments.items():
  317. db.execute(
  318. text(f"""
  319. ALTER TABLE {report.data_table_name} CHANGE {column_name} {column_name} TEXT COMMENT :comment
  320. """),
  321. {"comment": comment}
  322. )
  323. # 添加新字段
  324. if update_data.new_fields:
  325. for field in update_data.new_fields:
  326. column_name = field['field_name']
  327. comment = field['comment']
  328. add_column_if_not_exists(report.data_table_name, column_name, 'TEXT', comment, db)
  329. db.commit()
  330. db.refresh(report)
  331. return {
  332. "code": 200,
  333. "msg": "操作成功"
  334. }
  335. #发布
  336. @router.put("/report/{report_id}/update_status/")
  337. async def update_report_status_and_time(
  338. report_id: str,
  339. db: Session = Depends(get_db),
  340. creator_id = Depends(valid_access_token)
  341. ):
  342. # 查询要修改的记录
  343. report = db.query(ReportManagement).filter(ReportManagement.report_id == report_id).first()
  344. if not report:
  345. raise HTTPException(status_code=404, detail="Report not found")
  346. print(report.creator_id,creator_id)
  347. # 验证请求者ID
  348. if str(report.creator_id) != str(creator_id):
  349. raise HTTPException(status_code=403, detail="没有权限更新此报告")
  350. if report.issued_status ==2:
  351. raise HTTPException(status_code=403, detail="不可重复发布")
  352. # 更新issued_status为2
  353. report.issued_status = 2
  354. # 更新create_time为当前时间
  355. report.start_time = datetime.utcnow()
  356. try:
  357. db.commit()
  358. db.refresh(report)
  359. return {
  360. "code": 200,
  361. "msg": "操作成功"
  362. }
  363. except Exception as e:
  364. db.rollback()
  365. raise HTTPException(status_code=400, detail=str(e))
  366. class TaskQuery(BaseModel):
  367. # user_id: str
  368. submission_status: Optional[List[int]] = None
  369. table_name: Optional[str] = None
  370. @router.post("/my_filling")
  371. @router.get("/my_filling")
  372. async def get_user_tasks(
  373. db: Session = Depends(get_db),
  374. query: TaskQuery = Body(...),
  375. user_id = Depends(valid_access_token)
  376. ):
  377. # 检查用户ID是否提供
  378. if not user_id:
  379. raise HTTPException(status_code=400, detail="用户ID是必填项")
  380. # 查询用户的所有任务信息
  381. user_tasks = db.query(ReportManagement, FormSubmission).join(
  382. FormSubmission, ReportManagement.report_id == FormSubmission.report_id
  383. ).filter(
  384. FormSubmission.user_id == user_id
  385. )
  386. # 如果提供了填报结果列表,则过滤结果
  387. if query.submission_status:
  388. user_tasks = user_tasks.filter(FormSubmission.submission_status.in_(query.submission_status))
  389. if query.table_name:
  390. user_tasks = user_tasks.filter(ReportManagement.table_name.ilike(f'%{query.table_name}%'))
  391. # 执行查询
  392. tasks = user_tasks.all()
  393. # 构造返回结果
  394. result_items = []
  395. for report, submission in tasks:
  396. result_item = {
  397. "user_id":user_id,
  398. "table_name": report.table_name,
  399. "report_id": report.report_id,
  400. "submission_status": submission.submission_status,
  401. "start_time": report.start_time,
  402. "end_time": report.end_time,
  403. }
  404. result_items.append(result_item)
  405. return {
  406. "code":200,
  407. "msg":"查询成功",
  408. "data": result_items
  409. }
  410. @router.get("/report_fields")
  411. @router.post("/report_fields")
  412. async def get_report_fields(
  413. db: Session = Depends(get_db),
  414. # user_id: str = Query(None, description="用户ID"),
  415. report_id: str = Query(None, description="填报ID"),
  416. user_id = Depends(valid_access_token)
  417. ):
  418. # 检查用户ID和填报ID是否提供
  419. if not user_id or not report_id:
  420. raise HTTPException(status_code=400, detail="用户ID和填报ID是必填项")
  421. # 获取对应填报ID的数据表名称
  422. report = db.query(ReportManagement).filter(ReportManagement.report_id == report_id).first()
  423. if not report:
  424. raise HTTPException(status_code=404, detail="未找到对应的填报ID")
  425. data_table_name = report.data_table_name
  426. if not data_table_name:
  427. raise HTTPException(status_code=404, detail="未找到对应的数据表名称")
  428. # 检查用户是否有权限访问填报数据
  429. submission = db.query(FormSubmission).filter(
  430. FormSubmission.report_id == report_id,
  431. FormSubmission.user_id == user_id
  432. ).first()
  433. if not submission:
  434. raise HTTPException(status_code=403, detail="没有权限访问这个填报数据")
  435. # 使用SQLAlchemy的inspect功能来获取表的字段信息
  436. inspector = inspect(db.bind)
  437. columns = inspector.get_columns(data_table_name)
  438. # 构造返回结果
  439. result_fields = []
  440. for column in columns:
  441. if column['name'] not in ['collect_status', 'create_id', 'id', 'user_id']:
  442. result_field = {
  443. "field_name": column['name'],
  444. "field_comment": column.get('comment', '无注释')
  445. }
  446. result_fields.append(result_field)
  447. # 返回用户ID、填报ID和字段信息
  448. return {
  449. "code":200,
  450. "msg":"查询成功",
  451. "user_id": user_id,
  452. "report_id": report_id,
  453. "fields": result_fields
  454. }
  455. class DataEntry(BaseModel):
  456. data: List[Dict[str, Any]] # 数据列表,每个元素是一个字典,包含字段名和值
  457. class DataItem(BaseModel):
  458. pass # 用于动态接收键值对
  459. class SubmitData(BaseModel):
  460. # user_id: int
  461. report_id: str
  462. data: List[Dict[str, str]] # 数据列表,每个元素是一个字典,包含字段名和值
  463. class Config:
  464. arbitrary_types_allowed = True
  465. @router.post("/submit_data")
  466. async def submit_data(
  467. db: Session = Depends(get_db),
  468. submit_data: SubmitData = Body(...),
  469. user_id = Depends(valid_access_token)
  470. ):
  471. # 检查用户ID和填报ID是否提供
  472. if not user_id or not submit_data.report_id:
  473. raise HTTPException(status_code=400, detail="用户ID和填报ID是必填项")
  474. # 获取对应填报ID的数据表名称
  475. report = db.query(ReportManagement).filter(ReportManagement.report_id == submit_data.report_id).first()
  476. if not report:
  477. raise HTTPException(status_code=404, detail="未找到对应的填报ID")
  478. data_table_name = report.data_table_name
  479. if not data_table_name:
  480. raise HTTPException(status_code=404, detail="未找到对应的数据表名称")
  481. # 检查用户是否有权限填报
  482. submission = db.query(FormSubmission).filter(
  483. FormSubmission.report_id == submit_data.report_id,
  484. FormSubmission.user_id == str(user_id) # 确保user_id是字符串类型
  485. ).first()
  486. if not submission:
  487. raise HTTPException(status_code=403, detail="用户没有填报权限")
  488. # print(report.creator_id,submit_data.user_id)
  489. # if submission.
  490. # 将数据写入数据库
  491. for item in submit_data.data:
  492. # 构造插入SQL语句
  493. columns = ', '.join(list(item.keys()) + ['create_id', 'user_id', 'collect_status'])
  494. values = ', '.join(
  495. [f":{k}" for k in item.keys()] + [f"'{report.creator_id}'", f"'{user_id}'", '1'])
  496. sql = f"INSERT INTO {data_table_name} ({columns}) VALUES ({values})"
  497. print(sql)
  498. # 执行插入操作
  499. db.execute(text(sql), item)
  500. submission.submission_status = 1
  501. db.add(submission)
  502. # 提交事务
  503. db.commit()
  504. return {
  505. "code":200,
  506. "msg": "数据提交成功"
  507. }
  508. class SubmissionQuery(BaseModel):
  509. # user_id: int # 用户ID,必须是整数
  510. report_id: str # 填报ID,必须是字符串
  511. @router.post("/submission_status")
  512. async def get_submission_status(
  513. db: Session = Depends(get_db),
  514. query: SubmissionQuery = Body(...),
  515. user_id = Depends(valid_access_token)
  516. ):
  517. # 检查用户ID和填报ID是否提供
  518. if not user_id or not query.report_id:
  519. raise HTTPException(status_code=400, detail="用户ID和填报ID是必填项")
  520. # 获取对应填报ID的数据表名称
  521. report = db.query(ReportManagement).filter(ReportManagement.report_id == query.report_id).first()
  522. if not report:
  523. raise HTTPException(status_code=404, detail="未找到对应的填报ID")
  524. data_table_name = report.data_table_name
  525. if not data_table_name:
  526. raise HTTPException(status_code=404, detail="未找到对应的数据表名称")
  527. # 获取填报情况
  528. submission = db.query(FormSubmission).filter(
  529. FormSubmission.report_id == query.report_id,
  530. FormSubmission.user_id == str(user_id) # 确保user_id是字符串类型
  531. ).first()
  532. if not submission:
  533. raise HTTPException(status_code=404, detail="未找到对应的填报情况")
  534. start_time = report.start_time
  535. end_time = report.end_time
  536. print(start_time,end_time)
  537. start_time_str = start_time.strftime('%Y-%m-%d %H:%M:%S')
  538. end_time_str = end_time.strftime('%Y-%m-%d %H:%M:%S')
  539. # 使用SQLAlchemy的inspect功能来获取表的列信息
  540. inspector = Inspector.from_engine(db.bind)
  541. columns = inspector.get_columns(data_table_name)
  542. # 提取列名和列注释
  543. column_names = [column['name'] for column in columns]
  544. column_comments = {column['name']: column['comment'] for column in columns if 'comment' in column}
  545. # 构造返回结果
  546. result_items = []
  547. excluded_columns = ['id', 'user_id', 'create_id', 'collect_status']
  548. # 构建查询SQL
  549. query_sql = text(f"""
  550. SELECT * FROM {data_table_name} WHERE user_id = :user_id
  551. """)
  552. result = db.execute(query_sql, {"user_id": user_id})
  553. rows = result.fetchall()
  554. # 添加字段名和字段注释作为第一行
  555. first_row = {column: column_comments.get(column, '') for column in column_names if
  556. column not in excluded_columns}
  557. result_items.append(first_row)
  558. for row in rows:
  559. # 过滤掉不需要的列
  560. filtered_row = {column: row[idx] for idx, column in enumerate(column_names) if
  561. column not in excluded_columns}
  562. result_items.append(filtered_row)
  563. result = {
  564. "code": 200,
  565. 'msg': '查询成功',
  566. 'start_time':start_time_str,
  567. "end_time":end_time_str,
  568. "data": result_items}
  569. return result
  570. # 辅助函数:根据字段备注获取表中所有匹配字段名
  571. def get_columns_with_comment_like(
  572. inspector: Inspector, table_name: str, comment_like: str
  573. ) -> List[str]:
  574. columns = inspector.get_columns(table_name)
  575. matching_columns = [column['name'] for column in columns if column.get('comment') and comment_like in column['comment']]
  576. return matching_columns
  577. # 辅助函数:检查是否有字段备注匹配
  578. def has_matching_column_comments(
  579. inspector: Inspector, table_name: str, comment_like: str
  580. ) -> bool:
  581. return bool(get_columns_with_comment_like(inspector, table_name, comment_like))
  582. @router.post("/reports_by_creator")
  583. @router.get("/reports_by_creator")
  584. async def get_reports_by_creator(
  585. field_comment: Optional[str] = Query(None, description="Optional comment of the field to match"),
  586. page: int = Query(default=1, gt=0), # 分页参数:当前页码,默认为1
  587. pageSize: int = Query(default=10, gt=0), # 分页参数:每页大小,默认为10
  588. db: Session = Depends(get_db),
  589. creator_id=Depends(valid_access_token)
  590. ):
  591. # 获取数据库Inspector
  592. inspector: Inspector = inspect(db.bind)
  593. # 查询 ReportManagement 表以获取所有相关的记录
  594. query = db.query(ReportManagement).filter(ReportManagement.creator_id == creator_id)
  595. # 计算总数
  596. total_count = query.count()
  597. # 分页查询
  598. offset = (page - 1) * pageSize
  599. reports = query.offset(offset).limit(pageSize).all()
  600. # 存储结果
  601. results = []
  602. for report in reports:
  603. # 检查是否存在 data_table_name
  604. if not report.data_table_name:
  605. continue # 如果没有 data_table_name,跳过这个 report
  606. # 检查是否有字段备注匹配 field_comment
  607. if field_comment and not has_matching_column_comments(inspector, report.data_table_name, field_comment):
  608. continue # 如果没有匹配的字段备注,跳过这个 report
  609. # 如果匹配成功,添加到结果中
  610. collection_time_str = report.collection_time.isoformat().replace('T', ' ') if report.collection_time else None
  611. results.append({
  612. "table_name": report.table_name,
  613. "collection_status": report.collection_status,
  614. "collection_time": collection_time_str,
  615. "report_id": report.report_id
  616. })
  617. # 如果没有找到任何记录,抛出404异常
  618. if not results:
  619. raise HTTPException(status_code=404, detail="没有找到与该创建人ID相关的记录")
  620. # 构造分页结果
  621. result = {
  622. "code": 200,
  623. "msg": "查询成功",
  624. "total": total_count,
  625. "totalPages": (total_count + pageSize - 1) // pageSize,
  626. "page": page,
  627. "pageSize": pageSize,
  628. "data": results
  629. }
  630. return result
  631. @router.put("/update_collection_status/")
  632. async def update_collection_status(
  633. # creator_id: str,
  634. report_id: str,
  635. new_status: int = Query(..., description="New collection status, must be 0, 1, or 2"),
  636. db: Session = Depends(get_db),
  637. creator_id = Depends(valid_access_token)
  638. ):
  639. # 检查 new_status 是否为允许的值之一
  640. if new_status not in (0, 1, 2):
  641. raise HTTPException(status_code=400, detail="Invalid collection status value")
  642. # 查询 ReportManagement 表以获取对应记录
  643. report = db.query(ReportManagement).filter(
  644. ReportManagement.creator_id == creator_id,
  645. ReportManagement.report_id == report_id
  646. ).first()
  647. # 如果没有找到记录,返回404
  648. if not report:
  649. raise HTTPException(status_code=404, detail="Report not found")
  650. # 更新 collection_status
  651. report.collection_status = new_status
  652. db.add(report)
  653. db.commit()
  654. db.refresh(report)
  655. return {
  656. "code":200,
  657. "msg": "更新成功",
  658. "new_status": new_status
  659. }
  660. class ReportQuery(BaseModel):
  661. # creator_id: str # 创建人ID,必须是字符串
  662. report_id: str # 填报ID,必须是字符串
  663. @router.get("/dataArchiveDetails/")
  664. async def get_records_by_creator_and_report(
  665. query: ReportQuery = Depends(),
  666. db: Session = Depends(get_db),
  667. creator_id = Depends(valid_access_token)
  668. ):
  669. creator_id
  670. # 查询 ReportManagement 表以获取对应记录
  671. report = db.query(ReportManagement).filter(
  672. ReportManagement.creator_id == creator_id,
  673. ReportManagement.report_id == query.report_id
  674. ).first()
  675. # 如果没有找到记录,返回404
  676. if not report:
  677. raise HTTPException(status_code=404, detail="Report not found")
  678. # 如果没有 data_table_name,返回404
  679. if not report.data_table_name:
  680. raise HTTPException(status_code=404, detail="Data table name not found")
  681. # 查询工单表所有信息,并关联用户表匹配到用户名字
  682. query_sql = text(f"""
  683. SELECT w.*, u.user_name
  684. FROM {report.data_table_name} w
  685. LEFT JOIN sys_user u ON w.user_id = u.user_id
  686. """)
  687. result = db.execute(query_sql)
  688. rows = result.fetchall()
  689. # 使用SQLAlchemy的inspect功能来获取表的列信息
  690. inspector = inspect(db.bind)
  691. columns = inspector.get_columns(report.data_table_name)
  692. # 提取列名和列注释
  693. column_names = [column['name'] for column in columns]
  694. column_comments = {column['name']: column['comment'] for column in columns if 'comment' in column}
  695. # 构造字段信息
  696. columns_info = []
  697. for column in column_names:
  698. if column not in ['id', 'user_id', 'create_id', 'collect_status']:
  699. columns_info.append({
  700. 'prop': column,
  701. 'label': column_comments.get(column, '')
  702. })
  703. # 构造返回结果
  704. rows_data = []
  705. for row in rows:
  706. # 过滤掉不需要的列,并添加到结果中
  707. filtered_row = {column: row[idx] for idx, column in enumerate(column_names) if column not in ['id', 'user_id', 'create_id', 'collect_status']}
  708. filtered_row['user_name'] = row[-1] # 添加用户昵称
  709. rows_data.append(filtered_row)
  710. # 获取报告的开始和结束时间,并格式化为字符串
  711. start_time_str = report.start_time.strftime('%Y-%m-%d %H:%M:%S') if report.start_time else None
  712. end_time_str = report.end_time.strftime('%Y-%m-%d %H:%M:%S') if report.end_time else None
  713. return {
  714. "code": 200,
  715. 'msg': '查询成功',
  716. 'start_time': start_time_str,
  717. "end_time": end_time_str,
  718. "columns": columns_info,
  719. "rows": rows_data
  720. }