from fastapi import APIRouter, Request, Depends, HTTPException from sqlalchemy.exc import IntegrityError from fastapi.responses import JSONResponse from database import get_db from sqlalchemy.orm import Session from models import * import json import random from sqlalchemy import create_engine, select from typing import Optional router = APIRouter() # @router.post('/create') # async def create_knowledge(request:Request,db:Session = Depends(get_db)): # data = await request.body() # body = data.decode(encoding='utf-8') # if len(body) > 0: # body = json.loads(body) # print(body) # random_10_digit_number = random.randint(1000000000, 9999999999) # # file_identifier = 'f' # reportId = 'ZJBG'+str(random_10_digit_number) # # file_identifier = # reportName = body["reportName"] # subject = body["subject"] # eventType = body["eventType"] # publishingUnit = body["publishingUnit"] # publishDate = body["publishDate"] # summary = body["summary"] # # notificationType = body["notificationType"] # # base_code = 'base'+str(random.randint(1000000000, 9999999999)) # # fileNames = body["fileName"] # filePath = '/data/upload/mergefile/' # # # # konwledge = KnowledgeBase( # reportId=reportId, # # reportName=reportName, # subject=subject, # eventType=eventType, # publishingUnit=publishingUnit, # publishDate=publishDate, # summary = summary, # notificationType = notificationType, # # base_code = base_code # ) # db.add(konwledge) # # for fileName in fileNames: # file_identifier='file'+str(random.randint(1000000000, 9999999999)) # knowledge_file = KnowledgeFile( # file_identifier=file_identifier, # file_path=filePath, # file_name = fileName, # is_deleted = 0, # knowledge_base_code = base_code # ) # db.add(knowledge_file) # # db.commit() # return { # "code":0, # "data":{ # "reportId": reportId, # "status": "success", # "message": "总结报告创建成功" # } # } @router.post('/create') async def create_knowledge(request: Request, db: Session = Depends(get_db)): try: data = await request.body() body = json.loads(data.decode(encoding='utf-8')) # 验证必需的字段 required_fields = ['reportName', 'subject', 'eventType', 'publishingUnit', 'publishDate', 'summary', # 'notificationType', 'fileNames'] missing_fields = [field for field in required_fields if field not in body] print('missing_fields',missing_fields) if missing_fields: raise HTTPException(status_code=401, detail=f"Missing required fields: {', '.join(missing_fields)}") # 生成随机的报告ID和基础知识代码 random_10_digit_number = random.randint(1000000000, 9999999999) reportId = 'ZJBG' + str(random_10_digit_number) base_code = 'base' + str(random.randint(1000000000, 9999999999)) # 从请求体中提取其他数据 reportName = body["reportName"] subject = body["subject"] eventType = body["eventType"] publishingUnit = body["publishingUnit"] publishDate = body["publishDate"] summary = body["summary"] notificationType = "总结报告" updateTime = body["publishDate"] # notificationType = body["notificationType"] fileNames = body["fileNames"] # 注意：这里假设它是列表 # 创建 KnowledgeBase 实例 konwledge = KnowledgeBase( reportId=reportId, reportName=reportName, subject=subject, eventType=eventType, publishingUnit=publishingUnit, publishDate=publishDate, summary=summary, notificationType=notificationType, base_code=base_code, updateTime=updateTime ) db.add(konwledge) # 创建 KnowledgeFile 实例 filePath = '/data/upload/mergefile/' for fileName in fileNames: file_identifier = 'file' + str(random.randint(1000000000, 9999999999)) knowledge_file = KnowledgeFile( file_identifier=file_identifier, file_path=filePath + fileName, # 如果fileName是完整的路径，则可能不需要再次添加filePath file_name=fileName, is_deleted=0, updateTime = updateTime, createTime = updateTime, knowledge_base_code=base_code ) db.add(knowledge_file) db.commit() return { "code": 200, "msg": "总结报告创建成功", "status": "success", "data": [reportId] } # return { # "code": 200, # "msg": "操作成功", # "data": { # "reportId": reportId, # "status": "success", # "message": "总结报告创建成功" # } # } except json.JSONDecodeError: raise HTTPException(status_code=400, detail="Invalid JSON data") except IntegrityError as e: db.rollback() raise HTTPException(status_code=409, detail=f"Database error: {str(e)}") except Exception as e: db.rollback() print(e) raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}") @router.post('/select') @router.get('/select') async def select_knowledge(request: Request, db: Session = Depends(get_db), pageNum: Optional[int] = None, pageSize: Optional[int] = None): # page_from_json = None try: # # 尝试从请求体中解析 JSON 数据 json_data = await request.json() # 初始化分页参数 page_from_json = json_data.get('pageNum') if json_data else None size_from_json = json_data.get('pageSize') if json_data else None except: page_from_json = None # 如果查询参数和请求体都存在，优先选择查询参数 if pageNum is not None and pageSize is not None: # 使用查询参数 page_to_use = pageNum size_to_use = pageSize elif page_from_json is not None and size_from_json is not None: # 使用请求体中的参数 page_to_use = page_from_json size_to_use = size_from_json else: # 如果只有一个存在，使用存在的那个，否则抛出异常 if pageNum is not None: page_to_use = pageNum elif page_from_json is not None: page_to_use = page_from_json else: raise HTTPException(status_code=400, detail="Page parameter is required") if pageSize is not None: size_to_use = pageSize elif size_from_json is not None: size_to_use = size_from_json else: raise HTTPException(status_code=400, detail="Size parameter is required") # 验证分页参数 if size_to_use > 100: size_to_use = 100 # 计算 offset offset = (page_to_use - 1) * size_to_use # 使用 ORM 查询并应用分页 data = db.query(KnowledgeBase).offset(offset).limit(size_to_use).all() # 计算总条数 total_count = db.query(func.count(KnowledgeBase.reportId)).scalar() # 计算总页数 total_pages = (total_count // size_to_use) + (1 if total_count % size_to_use else 0) # 返回查询结果 # result = { # "code": 200, # 'msg': '查询成功', # 'data': { # 'pages': total_pages, # 'total': total_count, # "currentPage": page_to_use, # "pageSize": size_to_use, # "rows": data # } # } result = { "code": 200, 'msg': '查询成功', 'pages': total_pages, 'total': total_count, "currentPage": page_to_use, "pageSize": size_to_use, 'data': data } return result @router.post('/detail') @router.get('/detail') async def get_knowledge_detail(request: Request, db: Session = Depends(get_db), reportID: Optional[str] = None): # 尝试从请求体中解析 JSON 数据 report_id_body=None try: data = await request.json() report_id_body = data.get('reportID') if data else None except: print('报错') pass # 确定要使用的 report_id report_id_to_use = reportID or report_id_body print(reportID) print(report_id_body) # 如果没有提供 report_id，则抛出异常 if not report_id_to_use: raise HTTPException(status_code=400, detail="Missing required parameter 'reportID'") # 查询 KnowledgeBase kb_entry = db.query(KnowledgeBase).filter(KnowledgeBase.reportId == report_id_to_use).first() if not kb_entry: raise HTTPException(status_code=404, detail="No knowledge base found for the given report ID") # 查询相关的 KnowledgeFile kf_entries = db.query(KnowledgeFile).filter(KnowledgeFile.knowledge_base_code == kb_entry.base_code).all() # 准备返回的数据 result = { "code": 200, "msg": "查询成功", "data": [{ "report_id": kb_entry.reportId, "reportName": kb_entry.reportName, "subject": kb_entry.subject, "eventType": kb_entry.eventType, "publishDate": kb_entry.publishDate, "publishingUnit": kb_entry.publishingUnit, "summary": kb_entry.summary, "notificationType": kb_entry.notificationType, "file": [ {"content": kf.file_name, "url": f'http://127.0.0.1:9988/api/file/download/{kf.file_name}'} for kf in kf_entries ] }] } return result