__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. #@router.get("/select")
  201. async def select_report(
  202. db: Session = Depends(get_db),
  203. query: ReportQuery = Body(..., description="Report query parameters in the request body"),
  204. creator_id: str = Depends(valid_access_token)
  205. ):
  206. # 构建查询
  207. data_query = db.query(ReportManagement).filter(ReportManagement.creator_id == creator_id)
  208. # 应用过滤条件
  209. if query.table_name:
  210. data_query = data_query.filter(ReportManagement.table_name.ilike(f"%{query.table_name}%"))
  211. if query.start_time and query.end_time:
  212. data_query = data_query.filter(
  213. ReportManagement.start_time >= query.start_time,
  214. ReportManagement.end_time <= query.end_time
  215. )
  216. # 处理 status 和 issued_status 的逗号分隔字符串
  217. if query.status:
  218. status_list = [int(s) for s in query.status.split(",")]
  219. data_query = data_query.filter(ReportManagement.status.in_(status_list))
  220. if query.issued_status:
  221. issued_status_list = [int(s) for s in query.issued_status.split(",")]
  222. data_query = data_query.filter(ReportManagement.issued_status.in_(issued_status_list))
  223. # 计算总数
  224. total_count = data_query.count()
  225. # 分页查询
  226. offset = (query.page - 1) * query.pageSize
  227. data = data_query.offset(offset).limit(query.pageSize).all()
  228. # 构造结果
  229. result_items = []
  230. for item in data:
  231. result_item = {
  232. "id": item.id,
  233. "report_id": item.report_id,
  234. "table_name": item.table_name,
  235. "data_table_name": item.data_table_name,
  236. "start_time": item.start_time,
  237. "end_time": item.end_time,
  238. "status": item.status,
  239. "issued_status": item.issued_status,
  240. "period_type": item.period_type,
  241. "creator_name": item.creator_name,
  242. "creator_id": creator_id,
  243. "created_at": item.created_at,
  244. "creator_phone": item.creator_phone,
  245. "updated_at": item.updated_at,
  246. "num_reporters": item.num_reporters
  247. }
  248. result_items.append(result_item)
  249. result = {
  250. "code": 200,
  251. "msg": "查询成功",
  252. "total": total_count,
  253. "pages": (total_count + query.pageSize - 1) // query.pageSize,
  254. "currentPage": query.page,
  255. "pageSize": query.pageSize,
  256. "data": result_items
  257. }
  258. return result
  259. # class ReportUpdate(BaseModel):
  260. # table_name: str = None
  261. # status: int = None
  262. # # issued_status: int = None
  263. # period_type: str = None
  264. # end_time: str = None
  265. # comments: dict = None # 字典,键为字段名,值为新的备注
  266. # new_fields: List[dict] = None
  267. class ReportUpdate(BaseModel):
  268. table_name: Optional[str] = None
  269. status: Optional[int] = None
  270. period_type: Optional[str] = None
  271. end_time: Optional[str] = None
  272. comments: Optional[Dict[str, str]] = None
  273. new_fields: Optional[List[Dict[str, str]]] = None
  274. class Config:
  275. extra = 'allow'
  276. # 动态添加字段到表
  277. def add_column_if_not_exists(table_name: str, column_name: str, column_type: str, comment: str, db: Session):
  278. inspector = inspect(db.bind)
  279. columns = inspector.get_columns(table_name)
  280. existing_column_names = {column['name'] for column in columns}
  281. if column_name not in existing_column_names:
  282. try:
  283. db.execute(
  284. text(f"""
  285. ALTER TABLE {table_name} ADD COLUMN {column_name} {column_type} COMMENT :comment
  286. """),
  287. {"comment": comment}
  288. )
  289. except Exception as e:
  290. db.rollback()
  291. raise HTTPException(status_code=400, detail=str(e))
  292. #修改
  293. @router.put("/report/{report_id}/")
  294. async def update_report(
  295. report_id: str,
  296. update_data: ReportUpdate,
  297. db: Session = Depends(get_db),
  298. creator_id = Depends(valid_access_token)
  299. ):
  300. # creator_id = '1' # 假设creator_id已经通过某种方式验证
  301. report = db.query(ReportManagement).filter(ReportManagement.report_id == report_id).first()
  302. if not report:
  303. raise HTTPException(status_code=404, detail="Report not found")
  304. if report.creator_id != creator_id:
  305. raise HTTPException(status_code=403, detail="没有权限更新此报告")
  306. # 更新字段
  307. if update_data.table_name:
  308. report.table_name = update_data.table_name
  309. if update_data.status is not None:
  310. report.status = update_data.status
  311. if update_data.period_type:
  312. report.period_type = update_data.period_type
  313. if update_data.end_time:
  314. report.end_time = datetime.fromisoformat(update_data.end_time)
  315. # 更新字段备注
  316. if update_data.comments:
  317. for column_name, comment in update_data.comments.items():
  318. db.execute(
  319. text(f"""
  320. ALTER TABLE {report.data_table_name} CHANGE {column_name} {column_name} TEXT COMMENT :comment
  321. """),
  322. {"comment": comment}
  323. )
  324. # 添加新字段
  325. if update_data.new_fields:
  326. for field in update_data.new_fields:
  327. column_name = field['field_name']
  328. comment = field['comment']
  329. add_column_if_not_exists(report.data_table_name, column_name, 'TEXT', comment, db)
  330. db.commit()
  331. db.refresh(report)
  332. return {
  333. "code": 200,
  334. "msg": "操作成功"
  335. }
  336. #发布
  337. @router.put("/report/{report_id}/update_status/")
  338. async def update_report_status_and_time(
  339. report_id: str,
  340. db: Session = Depends(get_db),
  341. creator_id = Depends(valid_access_token)
  342. ):
  343. # 查询要修改的记录
  344. report = db.query(ReportManagement).filter(ReportManagement.report_id == report_id).first()
  345. if not report:
  346. raise HTTPException(status_code=404, detail="Report not found")
  347. print(report.creator_id,creator_id)
  348. # 验证请求者ID
  349. if str(report.creator_id) != str(creator_id):
  350. raise HTTPException(status_code=403, detail="没有权限更新此报告")
  351. if report.issued_status ==2:
  352. raise HTTPException(status_code=403, detail="不可重复发布")
  353. # 更新issued_status为2
  354. report.issued_status = 2
  355. # 更新create_time为当前时间
  356. report.start_time = datetime.utcnow()
  357. try:
  358. db.commit()
  359. db.refresh(report)
  360. return {
  361. "code": 200,
  362. "msg": "操作成功"
  363. }
  364. except Exception as e:
  365. db.rollback()
  366. raise HTTPException(status_code=400, detail=str(e))
  367. class TaskQuery(BaseModel):
  368. # user_id: str
  369. submission_status: Optional[List[int]] = None
  370. table_name: Optional[str] = None
  371. @router.post("/my_filling")
  372. @router.get("/my_filling")
  373. async def get_user_tasks(
  374. db: Session = Depends(get_db),
  375. query: TaskQuery = Body(...),
  376. user_id = Depends(valid_access_token)
  377. ):
  378. # 检查用户ID是否提供
  379. if not user_id:
  380. raise HTTPException(status_code=400, detail="用户ID是必填项")
  381. # 查询用户的所有任务信息
  382. user_tasks = db.query(ReportManagement, FormSubmission).join(
  383. FormSubmission, ReportManagement.report_id == FormSubmission.report_id
  384. ).filter(
  385. FormSubmission.user_id == user_id
  386. )
  387. # 如果提供了填报结果列表,则过滤结果
  388. if query.submission_status:
  389. user_tasks = user_tasks.filter(FormSubmission.submission_status.in_(query.submission_status))
  390. if query.table_name:
  391. user_tasks = user_tasks.filter(ReportManagement.table_name.ilike(f'%{query.table_name}%'))
  392. # 执行查询
  393. tasks = user_tasks.all()
  394. # 构造返回结果
  395. result_items = []
  396. for report, submission in tasks:
  397. result_item = {
  398. "user_id":user_id,
  399. "table_name": report.table_name,
  400. "report_id": report.report_id,
  401. "submission_status": submission.submission_status,
  402. "start_time": report.start_time,
  403. "end_time": report.end_time,
  404. }
  405. result_items.append(result_item)
  406. return {
  407. "code":200,
  408. "msg":"查询成功",
  409. "data": result_items
  410. }
  411. @router.get("/report_fields")
  412. @router.post("/report_fields")
  413. async def get_report_fields(
  414. db: Session = Depends(get_db),
  415. # user_id: str = Query(None, description="用户ID"),
  416. report_id: str = Query(None, description="填报ID"),
  417. user_id = Depends(valid_access_token)
  418. ):
  419. # 检查用户ID和填报ID是否提供
  420. if not user_id or not report_id:
  421. raise HTTPException(status_code=400, detail="用户ID和填报ID是必填项")
  422. # 获取对应填报ID的数据表名称
  423. report = db.query(ReportManagement).filter(ReportManagement.report_id == report_id).first()
  424. if not report:
  425. raise HTTPException(status_code=404, detail="未找到对应的填报ID")
  426. data_table_name = report.data_table_name
  427. if not data_table_name:
  428. raise HTTPException(status_code=404, detail="未找到对应的数据表名称")
  429. # 检查用户是否有权限访问填报数据
  430. submission = db.query(FormSubmission).filter(
  431. FormSubmission.report_id == report_id,
  432. FormSubmission.user_id == user_id
  433. ).first()
  434. if not submission:
  435. raise HTTPException(status_code=403, detail="没有权限访问这个填报数据")
  436. # 使用SQLAlchemy的inspect功能来获取表的字段信息
  437. inspector = inspect(db.bind)
  438. columns = inspector.get_columns(data_table_name)
  439. # 构造返回结果
  440. result_fields = []
  441. for column in columns:
  442. if column['name'] not in ['collect_status', 'create_id', 'id', 'user_id']:
  443. result_field = {
  444. "field_name": column['name'],
  445. "field_comment": column.get('comment', '无注释')
  446. }
  447. result_fields.append(result_field)
  448. # 返回用户ID、填报ID和字段信息
  449. return {
  450. "code":200,
  451. "msg":"查询成功",
  452. "user_id": user_id,
  453. "report_id": report_id,
  454. "fields": result_fields
  455. }
  456. class DataEntry(BaseModel):
  457. data: List[Dict[str, Any]] # 数据列表,每个元素是一个字典,包含字段名和值
  458. class DataItem(BaseModel):
  459. pass # 用于动态接收键值对
  460. class SubmitData(BaseModel):
  461. # user_id: int
  462. report_id: str
  463. data: List[Dict[str, str]] # 数据列表,每个元素是一个字典,包含字段名和值
  464. class Config:
  465. arbitrary_types_allowed = True
  466. @router.post("/submit_data")
  467. async def submit_data(
  468. db: Session = Depends(get_db),
  469. submit_data: SubmitData = Body(...),
  470. user_id = Depends(valid_access_token)
  471. ):
  472. # 检查用户ID和填报ID是否提供
  473. if not user_id or not submit_data.report_id:
  474. raise HTTPException(status_code=400, detail="用户ID和填报ID是必填项")
  475. # 获取对应填报ID的数据表名称
  476. report = db.query(ReportManagement).filter(ReportManagement.report_id == submit_data.report_id).first()
  477. if not report:
  478. raise HTTPException(status_code=404, detail="未找到对应的填报ID")
  479. data_table_name = report.data_table_name
  480. if not data_table_name:
  481. raise HTTPException(status_code=404, detail="未找到对应的数据表名称")
  482. # 检查用户是否有权限填报
  483. submission = db.query(FormSubmission).filter(
  484. FormSubmission.report_id == submit_data.report_id,
  485. FormSubmission.user_id == str(user_id) # 确保user_id是字符串类型
  486. ).first()
  487. if not submission:
  488. raise HTTPException(status_code=403, detail="用户没有填报权限")
  489. # print(report.creator_id,submit_data.user_id)
  490. # if submission.
  491. # 将数据写入数据库
  492. for item in submit_data.data:
  493. # 构造插入SQL语句
  494. columns = ', '.join(list(item.keys()) + ['create_id', 'user_id', 'collect_status'])
  495. values = ', '.join(
  496. [f":{k}" for k in item.keys()] + [f"'{report.creator_id}'", f"'{user_id}'", '1'])
  497. sql = f"INSERT INTO {data_table_name} ({columns}) VALUES ({values})"
  498. print(sql)
  499. # 执行插入操作
  500. db.execute(text(sql), item)
  501. submission.submission_status = 1
  502. db.add(submission)
  503. # 提交事务
  504. db.commit()
  505. return {
  506. "code":200,
  507. "msg": "数据提交成功"
  508. }
  509. class SubmissionQuery(BaseModel):
  510. # user_id: int # 用户ID,必须是整数
  511. report_id: str # 填报ID,必须是字符串
  512. @router.post("/submission_status")
  513. async def get_submission_status(
  514. db: Session = Depends(get_db),
  515. query: SubmissionQuery = Body(...),
  516. user_id = Depends(valid_access_token)
  517. ):
  518. # 检查用户ID和填报ID是否提供
  519. if not user_id or not query.report_id:
  520. raise HTTPException(status_code=400, detail="用户ID和填报ID是必填项")
  521. # 获取对应填报ID的数据表名称
  522. report = db.query(ReportManagement).filter(ReportManagement.report_id == query.report_id).first()
  523. if not report:
  524. raise HTTPException(status_code=404, detail="未找到对应的填报ID")
  525. data_table_name = report.data_table_name
  526. if not data_table_name:
  527. raise HTTPException(status_code=404, detail="未找到对应的数据表名称")
  528. # 获取填报情况
  529. submission = db.query(FormSubmission).filter(
  530. FormSubmission.report_id == query.report_id,
  531. FormSubmission.user_id == str(user_id) # 确保user_id是字符串类型
  532. ).first()
  533. if not submission:
  534. raise HTTPException(status_code=404, detail="未找到对应的填报情况")
  535. start_time = report.start_time
  536. end_time = report.end_time
  537. print(start_time,end_time)
  538. start_time_str = start_time.strftime('%Y-%m-%d %H:%M:%S')
  539. end_time_str = end_time.strftime('%Y-%m-%d %H:%M:%S')
  540. # 使用SQLAlchemy的inspect功能来获取表的列信息
  541. inspector = Inspector.from_engine(db.bind)
  542. columns = inspector.get_columns(data_table_name)
  543. # 提取列名和列注释
  544. column_names = [column['name'] for column in columns]
  545. column_comments = {column['name']: column['comment'] for column in columns if 'comment' in column}
  546. # 构造返回结果
  547. result_items = []
  548. excluded_columns = ['id', 'user_id', 'create_id', 'collect_status']
  549. # 构建查询SQL
  550. query_sql = text(f"""
  551. SELECT * FROM {data_table_name} WHERE user_id = :user_id
  552. """)
  553. result = db.execute(query_sql, {"user_id": user_id})
  554. rows = result.fetchall()
  555. # 添加字段名和字段注释作为第一行
  556. first_row = {column: column_comments.get(column, '') for column in column_names if
  557. column not in excluded_columns}
  558. result_items.append(first_row)
  559. for row in rows:
  560. # 过滤掉不需要的列
  561. filtered_row = {column: row[idx] for idx, column in enumerate(column_names) if
  562. column not in excluded_columns}
  563. result_items.append(filtered_row)
  564. result = {
  565. "code": 200,
  566. 'msg': '查询成功',
  567. 'start_time':start_time_str,
  568. "end_time":end_time_str,
  569. "data": result_items}
  570. return result
  571. # 辅助函数:根据字段备注获取表中所有匹配字段名
  572. def get_columns_with_comment_like(
  573. inspector: Inspector, table_name: str, comment_like: str
  574. ) -> List[str]:
  575. columns = inspector.get_columns(table_name)
  576. matching_columns = [column['name'] for column in columns if column.get('comment') and comment_like in column['comment']]
  577. return matching_columns
  578. # 辅助函数:检查是否有字段备注匹配
  579. def has_matching_column_comments(
  580. inspector: Inspector, table_name: str, comment_like: str
  581. ) -> bool:
  582. return bool(get_columns_with_comment_like(inspector, table_name, comment_like))
  583. @router.post("/reports_by_creator")
  584. @router.get("/reports_by_creator")
  585. async def get_reports_by_creator(
  586. field_comment: Optional[str] = Query(None, description="Optional comment of the field to match"),
  587. page: int = Query(default=1, gt=0), # 分页参数:当前页码,默认为1
  588. pageSize: int = Query(default=10, gt=0), # 分页参数:每页大小,默认为10
  589. db: Session = Depends(get_db),
  590. creator_id=Depends(valid_access_token)
  591. ):
  592. # 获取数据库Inspector
  593. inspector: Inspector = inspect(db.bind)
  594. # 查询 ReportManagement 表以获取所有相关的记录
  595. query = db.query(ReportManagement).filter(ReportManagement.creator_id == creator_id)
  596. # 计算总数
  597. total_count = query.count()
  598. # 分页查询
  599. offset = (page - 1) * pageSize
  600. reports = query.offset(offset).limit(pageSize).all()
  601. # 存储结果
  602. results = []
  603. for report in reports:
  604. # 检查是否存在 data_table_name
  605. if not report.data_table_name:
  606. continue # 如果没有 data_table_name,跳过这个 report
  607. # 检查是否有字段备注匹配 field_comment
  608. if field_comment and not has_matching_column_comments(inspector, report.data_table_name, field_comment):
  609. continue # 如果没有匹配的字段备注,跳过这个 report
  610. # 如果匹配成功,添加到结果中
  611. collection_time_str = report.collection_time.isoformat().replace('T', ' ') if report.collection_time else None
  612. results.append({
  613. "table_name": report.table_name,
  614. "collection_status": report.collection_status,
  615. "collection_time": collection_time_str,
  616. "report_id": report.report_id
  617. })
  618. # 如果没有找到任何记录,抛出404异常
  619. if not results:
  620. raise HTTPException(status_code=404, detail="没有找到与该创建人ID相关的记录")
  621. # 构造分页结果
  622. result = {
  623. "code": 200,
  624. "msg": "查询成功",
  625. "total": total_count,
  626. "totalPages": (total_count + pageSize - 1) // pageSize,
  627. "page": page,
  628. "pageSize": pageSize,
  629. "data": results
  630. }
  631. return result
  632. @router.put("/update_collection_status/")
  633. async def update_collection_status(
  634. # creator_id: str,
  635. report_id: str,
  636. new_status: int = Query(..., description="New collection status, must be 0, 1, or 2"),
  637. db: Session = Depends(get_db),
  638. creator_id = Depends(valid_access_token)
  639. ):
  640. # 检查 new_status 是否为允许的值之一
  641. if new_status not in (0, 1, 2):
  642. raise HTTPException(status_code=400, detail="Invalid collection status value")
  643. # 查询 ReportManagement 表以获取对应记录
  644. report = db.query(ReportManagement).filter(
  645. ReportManagement.creator_id == creator_id,
  646. ReportManagement.report_id == report_id
  647. ).first()
  648. # 如果没有找到记录,返回404
  649. if not report:
  650. raise HTTPException(status_code=404, detail="Report not found")
  651. # 更新 collection_status
  652. report.collection_status = new_status
  653. db.add(report)
  654. db.commit()
  655. db.refresh(report)
  656. return {
  657. "code":200,
  658. "msg": "更新成功",
  659. "new_status": new_status
  660. }
  661. class ReportQuery(BaseModel):
  662. # creator_id: str # 创建人ID,必须是字符串
  663. report_id: str # 填报ID,必须是字符串
  664. @router.get("/dataArchiveDetails/")
  665. async def get_records_by_creator_and_report(
  666. query: ReportQuery = Depends(),
  667. db: Session = Depends(get_db),
  668. creator_id = Depends(valid_access_token)
  669. ):
  670. creator_id
  671. # 查询 ReportManagement 表以获取对应记录
  672. report = db.query(ReportManagement).filter(
  673. ReportManagement.creator_id == creator_id,
  674. ReportManagement.report_id == query.report_id
  675. ).first()
  676. # 如果没有找到记录,返回404
  677. if not report:
  678. raise HTTPException(status_code=404, detail="Report not found")
  679. # 如果没有 data_table_name,返回404
  680. if not report.data_table_name:
  681. raise HTTPException(status_code=404, detail="Data table name not found")
  682. # 查询工单表所有信息,并关联用户表匹配到用户名字
  683. query_sql = text(f"""
  684. SELECT w.*, u.user_name
  685. FROM {report.data_table_name} w
  686. LEFT JOIN sys_user u ON w.user_id = u.user_id
  687. """)
  688. result = db.execute(query_sql)
  689. rows = result.fetchall()
  690. # 使用SQLAlchemy的inspect功能来获取表的列信息
  691. inspector = inspect(db.bind)
  692. columns = inspector.get_columns(report.data_table_name)
  693. # 提取列名和列注释
  694. column_names = [column['name'] for column in columns]
  695. column_comments = {column['name']: column['comment'] for column in columns if 'comment' in column}
  696. # 构造字段信息
  697. columns_info = []
  698. for column in column_names:
  699. if column not in ['id', 'user_id', 'create_id', 'collect_status']:
  700. columns_info.append({
  701. 'prop': column,
  702. 'label': column_comments.get(column, '')
  703. })
  704. # 构造返回结果
  705. rows_data = []
  706. for row in rows:
  707. # 过滤掉不需要的列,并添加到结果中
  708. filtered_row = {column: row[idx] for idx, column in enumerate(column_names) if column not in ['id', 'user_id', 'create_id', 'collect_status']}
  709. filtered_row['user_name'] = row[-1] # 添加用户昵称
  710. rows_data.append(filtered_row)
  711. # 获取报告的开始和结束时间,并格式化为字符串
  712. start_time_str = report.start_time.strftime('%Y-%m-%d %H:%M:%S') if report.start_time else None
  713. end_time_str = report.end_time.strftime('%Y-%m-%d %H:%M:%S') if report.end_time else None
  714. return {
  715. "code": 200,
  716. 'msg': '查询成功',
  717. 'start_time': start_time_str,
  718. "end_time": end_time_str,
  719. "columns": columns_info,
  720. "rows": rows_data
  721. }