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