97 lines
3.5 KiB
Python
97 lines
3.5 KiB
Python
|
import json
|
|||
|
|
from flask import Blueprint, request, jsonify, g
|
|||
|
|
from ..models import Session, Report
|
|||
|
|
from src.main import Consultation
|
|||
|
|
from src.utils import read_basic_info_json
|
|||
|
|
from . import globals
|
|||
|
|
|
|||
|
|
from .. import db
|
|||
|
|
|
|||
|
|
report_routes = Blueprint('report', __name__)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def validate_session_id(sessionId):
|
|||
|
|
"""验证sessionId是否提供且存在于Session表中"""
|
|||
|
|
if not sessionId:
|
|||
|
|
return jsonify({"error": "sessionId is required"}), 400
|
|||
|
|
session = Session.query.filter_by(id=sessionId).first()
|
|||
|
|
if not session:
|
|||
|
|
return jsonify({"error": "Session not found"}), 404
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
|
|||
|
|
@report_routes.route('/get-report/<int:sessionId>', methods=['GET'])
|
|||
|
|
def get_report(sessionId):
|
|||
|
|
validation_result = validate_session_id(sessionId)
|
|||
|
|
if validation_result:
|
|||
|
|
return validation_result
|
|||
|
|
reports = Report.query.filter_by(sessionId=sessionId).all()
|
|||
|
|
return jsonify(reports=[report.to_dict() for report in reports]), 200
|
|||
|
|
|
|||
|
|
|
|||
|
|
@report_routes.route('/add-report/<int:sessionId>', methods=['POST'])
|
|||
|
|
def add_report(sessionId):
|
|||
|
|
"""添加报告到数据库"""
|
|||
|
|
# 获取 sessionId 和上传的文件
|
|||
|
|
|
|||
|
|
if not globals.consultation: # 检查是否存在consultation对象
|
|||
|
|
return jsonify({"error": "No consultation object found"}), 404
|
|||
|
|
if not globals.consultation.save_result: # 检查是否存在save_result方法或属性（根据实际情况调整）:
|
|||
|
|
return jsonify({"error": "No save_result method or property found"}), 404
|
|||
|
|
try:
|
|||
|
|
# 验证 sessionId 是否存在
|
|||
|
|
validation_result = validate_session_id(sessionId)
|
|||
|
|
if validation_result:
|
|||
|
|
return validation_result
|
|||
|
|
|
|||
|
|
report = globals.consultation.save_result(sessionId)
|
|||
|
|
|
|||
|
|
# 创建新的 Report 记录
|
|||
|
|
new_report = Report(
|
|||
|
|
sessionId=sessionId,
|
|||
|
|
report=report # 假设 report_data 是 Report 表中的一个字段
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 添加到数据库
|
|||
|
|
db.session.add(new_report)
|
|||
|
|
db.session.commit()
|
|||
|
|
|
|||
|
|
format_report = globals.consultation.format_report(sessionId)
|
|||
|
|
print(format_report,type(format_report))
|
|||
|
|
|
|||
|
|
return jsonify({
|
|||
|
|
"message": "Report added successfully",
|
|||
|
|
"data": format_report
|
|||
|
|
}), 201
|
|||
|
|
except json.JSONDecodeError:
|
|||
|
|
return jsonify({"error": "Invalid JSON in report file"}), 400
|
|||
|
|
except Exception as e:
|
|||
|
|
db.session.rollback()
|
|||
|
|
return jsonify({"error": f"Failed to add report: {str(e)}"}), 500
|
|||
|
|
|
|||
|
|
|
|||
|
|
@report_routes.route('/upload-report/<int:sessionId>', methods=['POST'])
|
|||
|
|
def upload_json(sessionId):
|
|||
|
|
# 检查请求中是否包含文件
|
|||
|
|
print(sessionId)
|
|||
|
|
if 'report' not in request.files:
|
|||
|
|
return jsonify({"error": "No file part"}), 400
|
|||
|
|
|
|||
|
|
report = request.files['report']
|
|||
|
|
# 检查文件是否为空
|
|||
|
|
if report.filename == '':
|
|||
|
|
return jsonify({"error": "No selected file"}), 400
|
|||
|
|
|
|||
|
|
# 检查文件是否为 JSON 文件
|
|||
|
|
if report and report.filename.endswith('.json'):
|
|||
|
|
try:
|
|||
|
|
# 读取文件内容并解析为 JSON
|
|||
|
|
data = json.load(report)
|
|||
|
|
globals.consultation.init_session(sessionId, case_data=data)
|
|||
|
|
print(globals.consultation.session_map[sessionId].patient.basic_info)
|
|||
|
|
return jsonify({"message": "File uploaded successfully", "data": data}), 200
|
|||
|
|
except json.JSONDecodeError:
|
|||
|
|
return jsonify({"error": "Invalid JSON file"}), 400
|
|||
|
|
else:
|
|||
|
|
return jsonify({"error": "File must be a JSON file"}), 400
|