python-fastapi-mm-zhcs-yj-api/routers/api/dataFilling/__init__.py

from fastapi import FastAPI, HTTPException, Depends, APIRouter,Query,Body
from sqlalchemy.engine.reflection import Inspector
from common.security import valid_access_token
from pydantic import BaseModel,Extra,Field
from datetime import datetime
from typing import List, Optional,Any,Dict
from sqlalchemy import create_engine, Column, Integer, String, Boolean, MetaData, Table, \
    inspect, exists,or_,text,insert,asc,desc
from sqlalchemy.orm import Session
from pypinyin import lazy_pinyin, Style
from database import get_db
from models import *
import random


import pandas as pd
from sqlalchemy import text
from fastapi.responses import StreamingResponse
from io import BytesIO


router = APIRouter()
metadata = MetaData()


class ReportField(BaseModel):
    name: str




def get_next_event_id(db: Session):
    while True:
        random_10_digit_number = random.randint(1000000000, 9999999999)
        reportId = 'report' + str(random_10_digit_number)

        it_exists = db.query(
            exists().where(ReportManagement.report_id == reportId)
        ).scalar()

        if it_exists == False:
            return reportId


# 函数用于将中文转换为拼音首字母缩写
def to_first_letter(chinese_str: str) -> str:
    return ''.join([p[0][0] for p in lazy_pinyin(chinese_str, style=Style.FIRST_LETTER)]).lower()





# class ReportQuery(BaseModel):
#     report_id: str

class TableStructure(BaseModel):
    column_name: str
    comment: str

# 表数据模型
class TableData(BaseModel):
    row_data: dict
#详情
@router.get("/report_structure/{report_id}")
async def get_report_structure(
    report_id: str,
    db: Session = Depends(get_db),
    creator_id = Depends(valid_access_token)
):
    # 查询 ReportManagement 表以获取 data_table_name
    report = db.query(ReportManagement).filter(
        ReportManagement.report_id == report_id,
        ReportManagement.creator_id == creator_id
    ).first()
    if not report:
        raise HTTPException(status_code=404, detail="Report not found")

    data_table_name = report.data_table_name

    # 查询对应表的表结构
    table_structure_query = db.execute(
        text("""
        SELECT COLUMN_NAME, COLUMN_COMMENT, ORDINAL_POSITION
        FROM INFORMATION_SCHEMA.COLUMNS
        WHERE TABLE_NAME = :table_name AND TABLE_SCHEMA = (SELECT DATABASE())
        ORDER BY ORDINAL_POSITION
        """),
        {"table_name": data_table_name}
    )

    table_structures = []
    column_order = []  # 用于存储字段的顺序
    for row in table_structure_query.fetchall():
        if row[0] not in ['collect_status', 'create_id', 'id', 'user_id', 'add_time']:
            table_structures.append(TableStructure(column_name=row[0], comment=row[1]))
            column_order.append(row[0])  # 保存字段顺序

    # 查询表中的数据，排除指定字段
    excluded_columns = ['collect_status', 'create_id', 'id', 'user_id', 'add_time']
    columns_to_select = ", ".join(column_order)  # 使用字段顺序
    table_data_with_headers = []
    # print("字段：",columns_to_select)
    # print(len(columns_to_select))
    # table_data_query =
    if (len(columns_to_select)) != 0:
        table_data_query = db.execute(
            text(f"SELECT {columns_to_select} FROM {data_table_name}")
        )


        # 将查询结果转换为字典列表
        table_data = [dict(zip(column_order, row)) for row in table_data_query.fetchall()]

        # 构造表头的字段名和字段注释作为第一行
        table_headers = {col.column_name: col.comment for col in table_structures}

        # 在表数据中添加表头备注作为第一行
        table_data_with_headers = [
            table_headers  # 表头备注
        ] + table_data  # 表数据

    # 查询已提交和未提交的用户状态，并获取用户昵称
    user_submission_status = []

    # 查询所有相关用户
    users = db.query(FormSubmission.user_id).filter(
        FormSubmission.report_id == report_id
    ).distinct()

    user_ids = [user[0] for user in users.all()]

    # 查询每个用户的提交状态和昵称
    for user_id in user_ids:
        user = db.query(SysUser.nick_name).filter(SysUser.user_id == user_id).first()
        if user:
            nick_name = user[0]
        else:
            nick_name = "未知用户"

        submission_status = db.query(FormSubmission.submission_status).filter(
            FormSubmission.report_id == report_id,
            FormSubmission.user_id == user_id
        ).first()

        if submission_status:
            submission_status = str(submission_status[0])
        else:
            submission_status = "0"  # 默认为未提交

        user_submission_status.append({
            "name": nick_name,
            "submission_status": submission_status,
            "user_id":user_id
        })

    # 构造返回结果
    result = {
        "code": 200,
        "msg": "查询成功",
        "report_info": {
            "id": report.id,
            "report_id": report.report_id,
            "table_name": report.table_name,
            "data_table_name": report.data_table_name,
            "start_time": report.start_time,
            "end_time": report.end_time,
            "status": report.status,
            "issued_status": report.issued_status,
            "period_type": report.period_type,
            "creator_name": report.creator_name,
            "num_reporters": len(user_ids),
            "creator_id": creator_id,
            "created_at": report.created_at,
            "updated_at": report.updated_at,
            "creator_phone": report.creator_phone,
            "user_filling_status": user_submission_status,  # 用户提交状态列表
            "user_ids": user_ids
        },
        "table_structure": table_structures,
        "table_data": table_data_with_headers  # 添加表数据
    }

    return result



# 动态创建表
def create_dynamic_table(table_name: str, field_names: List[str], db: Session):
    inspector = inspect(db.bind)

    # 检查表是否已存在
    if inspector.has_table(table_name):
        raise HTTPException(status_code=400, detail="Table already exists")

    table = Table(table_name, metadata,
                  Column('id', Integer, primary_key=True,comment="id"),
                  Column('user_id', Integer,comment="用户ID"),
                  Column('create_id', Integer,comment="创建者ID"),
                  Column('collect_status', Boolean,comment="收取结果"),
                  Column('add_time', DateTime, server_default=func.now(),comment="添加时间"),
                  extend_existing=True
                  )

    existing_columns = set()
    for field_name in field_names:
        column_name = to_first_letter(field_name)
        # 如果列名已存在，则添加一个唯一的后缀
        unique_column_name = "col"+column_name
        suffix = 1
        while unique_column_name in existing_columns:
            unique_column_name = f"{column_name}_{suffix}"
            suffix += 1
        existing_columns.add(unique_column_name)

        table.append_column(Column(unique_column_name, String(255), comment=field_name))

    # 创建表
    try:
        metadata.create_all(bind=db.bind)
    except Exception as e:
        db.rollback()
        raise HTTPException(status_code=400, detail=f"创建表失败：{str(e)}")


class ReportCreate(BaseModel):
    table_name: str = Field(..., description="表单名称，必填")
    end_time: str = Field(..., description="结束时间，必填，格式为 ISO8601")
    # status: str = Field(..., description="状态，必填")
    issued_status: str = Field(..., description="发布状态，必填")
    creator_name: str = Field(..., description="创建者姓名，必填")
    creator_phone: str = Field(..., description="创建者电话，必填")
    user_ids: List[int] = Field(..., description="用户 ID 列表，必填")

    period_type: Optional[str] = Field(None, description="周期，非必填")
    field_names: Optional[List[str]] = Field(None, description="字段名称列表，非必填")


# 新建填报和创建新表的接口
@router.post("/report/")
def create_report_and_table(report: ReportCreate, db: Session = Depends(get_db),
                            creator_id = Depends(valid_access_token)
                            ):
    try:
        # 获取当前时间并格式化为 YYYYMMDDHHMMSS
        current_time_str = datetime.now().strftime("%Y%m%d%H%M%S")
        # 动态生成 data_table_name
        table_name_pinyin=''
        for i in range(len(lazy_pinyin(report.table_name, style=Style.FIRST_LETTER))):
            table_name_pinyin += ''.join(lazy_pinyin(report.table_name, style=Style.FIRST_LETTER)[i]).lower()
        data_table_name = f"tbxt_{table_name_pinyin}_{current_time_str}"
        # 待发布状态（即暂时）需领处理，如果是待发布（issued_status 1待发布  2已发布），
        # 那么只需要填联系人姓名，联系电话，填报人，截止时间和表名
        # 可以先不创表，但是填报人先报上？
        if str(report.issued_status) == '2' and not report.field_names:
            #这里有bug
            raise HTTPException(status_code=403, detail='发布状态下，需要填写字段信息')

        if str(report.issued_status) == '2':

            # 登记填报管理
            new_report = ReportManagement(
                report_id=get_next_event_id(db),
                table_name=report.table_name,
                data_table_name=data_table_name,
                start_time=datetime.now(),
                end_time=report.end_time,
                status=0,
                issued_status=report.issued_status,
                collection_status=0,#未收取
                period_type=report.period_type,
                creator_name=report.creator_name,
                creator_id=creator_id,
                creator_phone=report.creator_phone,
                num_reporters = len(report.user_ids),
                issued_time = datetime.now()
            )
            db.add(new_report)
            db.commit()
            db.refresh(new_report)
        else:
            # 登记填报管理
            new_report = ReportManagement(
                report_id=get_next_event_id(db),
                table_name=report.table_name,
                data_table_name=data_table_name,
                start_time=datetime.now(),
                end_time=report.end_time,
                status=0,
                issued_status=report.issued_status,
                collection_status=0,  # 未收取
                period_type=report.period_type,
                creator_name=report.creator_name,
                creator_id=creator_id,
                creator_phone=report.creator_phone,
                num_reporters=len(report.user_ids)
            )
            db.add(new_report)
            db.commit()
            db.refresh(new_report)

        if report.field_names :
            if len(report.field_names) > 0:
                # 动态创建新表
                create_dynamic_table(data_table_name, report.field_names, db)

        # 为每个用户创建填报记录
        for user_id in report.user_ids:
            submission = FormSubmission(
                report_id=new_report.report_id,
                user_id=user_id,
                submission_status=0  # 默认状态为未填报
            )
            db.add(submission)
        db.commit()
        return {
            "code": 200,
            "msg":"创建成功"
        }
    except Exception as e:
        raise HTTPException(status_code=400, detail=str(e))







# 定义请求体的 Pydantic 模型
class ReportQuery(BaseModel):
    table_name: Optional[str] = Field(None, description="Table name filter")
    status: Optional[str] = Field(None, description="Status filter (comma-separated values, e.g., '1,2,3')")
    start_time: Optional[datetime] = Field(None, description="Start time filter")
    end_time: Optional[datetime] = Field(None, description="End time filter")
    issued_status: Optional[str] = Field(None, description="Issued status filter (comma-separated values, e.g., '0,1')")
    page: int = Field(1, gt=0, description="Page number for pagination")
    pageSize: int = Field(10, gt=0, description="Page size for pagination")

@router.post("/select")
async def select_report(
    db: Session = Depends(get_db),
    query: ReportQuery = Body(..., description="Report query parameters in the request body"),
    creator_id: str = Depends(valid_access_token)
):
    # 构建查询
    data_query = db.query(ReportManagement).filter(ReportManagement.creator_id == creator_id)

    # 应用过滤条件
    if query.table_name:
        data_query = data_query.filter(ReportManagement.table_name.ilike(f"%{query.table_name}%"))

    if query.start_time and query.end_time:
        data_query = data_query.filter(
            ReportManagement.start_time >= query.start_time,
            ReportManagement.end_time <= query.end_time
        )

    # 处理 status 和 issued_status 的逗号分隔字符串
    if query.status:
        status_list = [int(s) for s in query.status.split(",")]
        data_query = data_query.filter(ReportManagement.status.in_(status_list))

    if query.issued_status:
        issued_status_list = [int(s) for s in query.issued_status.split(",")]
        data_query = data_query.filter(ReportManagement.issued_status.in_(issued_status_list))

    data_query = data_query.order_by(
        asc(ReportManagement.issued_status),  # 按 issued_status 升序
        desc(ReportManagement.created_at)  # 按 created_at 降序
    )

    # 计算总数
    total_count = data_query.count()

    # 分页查询
    offset = (query.page - 1) * query.pageSize
    data = data_query.offset(offset).limit(query.pageSize).all()

    # 构造结果
    result_items = []
    for item in data:
        current_time = datetime.now()
        is_filling_ended = 0
        if item.end_time < current_time:
            is_filling_ended = 2

        result_item = {
            "id": item.id,
            "report_id": item.report_id,
            "table_name": item.table_name,
            "data_table_name": item.data_table_name,
            "start_time": item.start_time,
            "end_time": item.end_time,
            "status": item.status,
            "issued_status": item.issued_status,
            "period_type": item.period_type,
            "creator_name": item.creator_name,
            "creator_id": creator_id,
            "created_at": item.created_at,
            "creator_phone": item.creator_phone,
            "updated_at": item.updated_at,
            "num_reporters": item.num_reporters,
            "is_filling_ended":is_filling_ended
        }
        result_items.append(result_item)

    result = {
        "code": 200,
        "msg": "查询成功",
        "total": total_count,
        "pages": (total_count + query.pageSize - 1) // query.pageSize,
        "currentPage": query.page,
        "pageSize": query.pageSize,
        "data": result_items
    }

    return result











def update_table_fields(table_name: str, field_names: List[str], db: Session):
    inspector = inspect(db.bind)

    # 检查表是否存在
    if not inspector.has_table(table_name):
        raise HTTPException(status_code=400, detail="表不存在，无法更新字段")

    # 获取现有表的列信息
    existing_columns = inspector.get_columns(table_name)
    existing_column_names = {col['name'] for col in existing_columns}

    # 定义需要保留的基础字段
    columns_to_keep = {'id', 'user_id', 'create_id', 'collect_status','add_time'}
    existing_column_names -= columns_to_keep  # 排除基础字段
    print(existing_column_names)
    print(field_names)
    # 将新字段名转换为拼音首字母
    new_field_names = {to_first_letter(field) for field in field_names}

    # 确定需要删除的字段（现有字段中不存在于新字段列表中的字段）
    columns_to_drop = existing_column_names - new_field_names

    # 确定需要添加的字段（新字段列表中不存在于现有字段中的字段）
    columns_to_add = new_field_names - existing_column_names

    # 删除不再需要的字段
    for column_name in columns_to_drop:
        print(text(f"ALTER TABLE {table_name} DROP COLUMN {column_name}"))
        try:
            db.execute(text(f"ALTER TABLE {table_name} DROP COLUMN {column_name}"))
        except Exception as e:
            db.rollback()
            raise HTTPException(status_code=400, detail=f"删除字段失败：{str(e)}")

    # 添加新字段
    added_columns = set()  # 用于记录已添加的字段名
    for field_name in field_names:
        column_name = to_first_letter(field_name)  # 将字段名转换为拼音首字母
        if column_name in columns_to_add:
            unique_column_name = column_name
            suffix = 1

            # 确保字段名唯一
            while unique_column_name in existing_column_names or unique_column_name in added_columns:
                unique_column_name = f"{column_name}_{suffix}"
                suffix += 1

            # 添加字段
            try:
                unique_column_name = "col"+unique_column_name
                print(text(f"ALTER TABLE {table_name} ADD COLUMN {unique_column_name} VARCHAR(255) COMMENT '{field_name}'"))
                db.execute(text(f"ALTER TABLE {table_name} ADD COLUMN {unique_column_name} VARCHAR(255) COMMENT '{field_name}'"))
                added_columns.add(unique_column_name)  # 记录已添加的字段名
            except Exception as e:
                db.rollback()
                raise HTTPException(status_code=400, detail=f"添加字段失败：{str(e)}")


# class ReportCreate(BaseModel):
#     table_name: str = Field(..., description="表单名称，必填")
#     end_time: str = Field(..., description="结束时间，必填，格式为 ISO8601")
#     # status: str = Field(..., description="状态，必填")
#     issued_status: str = Field(..., description="发布状态，必填")
#     creator_name: str = Field(..., description="创建者姓名，必填")
#     creator_phone: str = Field(..., description="创建者电话，必填")
#     user_ids: List[int] = Field(..., description="用户 ID 列表，必填")
#
#     period_type: Optional[str] = Field(None, description="周期，非必填")
#     field_names: Optional[List[str]] = Field(None, description="字段名称列表，非必填")



class ReportUpdate(BaseModel):
    table_name: Optional[str] = None
    end_time: Optional[str] = None
    status: Optional[int] = None
    issued_status: Optional[str] = None
    period_type: Optional[str] = None
    creator_phone: Optional[str] = None
    creator_name: Optional[str] = None
    user_ids: Optional[List[int]] = Field(None, description="字段名称列表，非必填")
    # comments: Optional[Dict[str, str]] = None
    new_fields: Optional[List[str]] = Field(None, description="字段名称列表，非必填")

    class Config:
        extra = 'allow'





def table_exists(db: Session, table_name: str) -> bool:
    inspector = inspect(db.bind)
    return inspector.has_table(table_name)


# 删除表（如果存在）
def drop_table_if_exists(db: Session, table_name: str):
    inspector = inspect(db.bind)
    if inspector.has_table(table_name):
        try:
            # 删除表
            db.execute(text(f"DROP TABLE {table_name}"))
            # 清理 MetaData 中的表定义
            metadata.reflect(bind=db.bind)
            if table_name in metadata.tables:
                del metadata.tables[table_name]
        except Exception as e:
            db.rollback()
            raise HTTPException(status_code=400, detail=f"删除表失败：{str(e)}")






#修改
@router.put("/report/{report_id}/")
async def update_report(
    report_id: str,
    update_data: ReportUpdate,
    db: Session = Depends(get_db),
    creator_id = Depends(valid_access_token)
):
    # creator_id = '1'  # 假设creator_id已经通过某种方式验证
    # 这里得添加下，如果是已经下发的，那就不能修改了，只有待发布（暂存状态）才可以修改   暂存：保存至“待发布”状态
    #.filter(ReportManagement.creator_id == creator_id)
    report = db.query(ReportManagement).filter(ReportManagement.report_id == report_id,
                                               ReportManagement.creator_id == creator_id).first()
    if not report:
        raise HTTPException(status_code=404, detail="Report not found")

    if report.issued_status in ['2', 2]:
        raise HTTPException(status_code=400, detail="当前表单已发布，无法修改")


    if report.collection_status in ['2',2]:
        raise HTTPException(status_code=400, detail="当前表单已收取，无法修改")



    # 检查表是否存在
    table_name = report.data_table_name
    if not table_exists(db, table_name):
        # 如果表不存在，根据 new_fields 创建表
        if not update_data.new_fields:
            raise HTTPException(status_code=400, detail="表不存在且未提供字段信息，无法创建表")
        create_dynamic_table(table_name, update_data.new_fields, db)

    elif table_exists(db, table_name) and update_data.new_fields:
        print("修改")
        update_table_fields(table_name, update_data.new_fields, db)

        # 更新 creator_phone 和 creator_name
    if update_data.creator_phone:
        report.creator_phone = update_data.creator_phone

    if update_data.creator_name:
        report.creator_name = update_data.creator_name

        # 删除旧的填报人 ID，并添加新的填报人 ID
    if update_data.user_ids:
        # 删除旧的填报记录
        db.query(FormSubmission).filter(FormSubmission.report_id == report.report_id).delete()
        # 添加新的填报记录
        for user_id in update_data.user_ids:
            submission = FormSubmission(
                report_id=report.report_id,
                user_id=user_id,
                submission_status=0  # 默认状态为未填报
            )
            db.add(submission)
        report.num_reporters = len(update_data.user_ids)


        # 更新字段
    if update_data.table_name:
        report.table_name = update_data.table_name

    if update_data.status is not None:
        report.status = update_data.status

    if update_data.period_type:
        report.period_type = update_data.period_type

    if update_data.end_time:
        report.end_time = datetime.fromisoformat(update_data.end_time)

    # if update_data.

    current_time = datetime.now()
    report.updated_at = current_time

    #先判断状态，未发布/暂存 的可以修改
    #先看看传入参数有没有修改字段，有修改字段的时候，先看看有没有表，如果没有表，没有的话就读取表单里面的表名，再读取字段参数进行建表
    #如果已经有表了，那就读取表单里的表名，再读取字段参数，然后删掉原有的表，再进行建表


    db.commit()
    db.refresh(report)

    return {
        "code": 200,
        "msg": "操作成功"
    }




#发布
@router.put("/report/{report_id}/update_status/")
async def update_report_status_and_time(
    report_id: str,
    db: Session = Depends(get_db),
    creator_id = Depends(valid_access_token)
):


    # 查询要修改的记录
    report = db.query(ReportManagement).filter(ReportManagement.report_id == report_id).first()
    if not report:
        raise HTTPException(status_code=404, detail="Report not found")
    # 验证请求者ID
    if str(report.creator_id) != str(creator_id):
        raise HTTPException(status_code=403, detail="没有权限操作")

    if report.issued_status ==2:
        raise HTTPException(status_code=403, detail="不可重复发布")

    data_table_name = report.data_table_name

    # 查询对应表的表结构
    table_structure_query = db.execute(
        text("""
            SELECT COLUMN_NAME, COLUMN_COMMENT, ORDINAL_POSITION
            FROM INFORMATION_SCHEMA.COLUMNS
            WHERE TABLE_NAME = :table_name AND TABLE_SCHEMA = (SELECT DATABASE())
            ORDER BY ORDINAL_POSITION
            """),
        {"table_name": data_table_name}
    )

    table_structures = []
    column_order = []  # 用于存储字段的顺序
    for row in table_structure_query.fetchall():
        if row[0] not in ['collect_status', 'create_id', 'id', 'user_id', 'add_time']:
            table_structures.append(TableStructure(column_name=row[0], comment=row[1]))
            column_order.append(row[0])  # 保存字段顺序
    # print(table_structures)

    # 查询所有相关用户
    users = db.query(FormSubmission.user_id).filter(
        FormSubmission.report_id == report_id
    ).distinct()

    user_ids = [user[0] for user in users.all()]
    print(user_ids)

    if len(table_structures) == 0 or len(user_ids) ==0:
        raise HTTPException(status_code=400, detail=str('信息未填写完整，无法发布'))

    # 更新issued_status为2
    report.issued_status = 2

    # 更新create_time为当前时间
    report.issued_time = datetime.utcnow()

    try:
        db.commit()
        db.refresh(report)
        return {
            "code": 200,
            "msg": "操作成功"
        }
    except Exception as e:
        db.rollback()
        raise HTTPException(status_code=400, detail=str(e))



class TaskQuery(BaseModel):
    # user_id: str
    submission_status: Optional[List[int]] = None
    table_name: Optional[str] = None

@router.post("/my_filling")
@router.get("/my_filling")
async def get_user_tasks(
    db: Session = Depends(get_db),
    query: TaskQuery = Body(...),
    user_id = Depends(valid_access_token)
):
    # 检查用户ID是否提供
    if not user_id:
        raise HTTPException(status_code=400, detail="用户ID是必填项")

    # 查询用户的所有任务信息
    user_tasks = db.query(ReportManagement, FormSubmission).join(
        FormSubmission, ReportManagement.report_id == FormSubmission.report_id
    ).filter(
        FormSubmission.user_id == user_id
    )

    # 如果提供了填报结果列表，则过滤结果
    if query.submission_status:
        user_tasks = user_tasks.filter(FormSubmission.submission_status.in_(query.submission_status))

    if query.table_name:
        user_tasks = user_tasks.filter(ReportManagement.table_name.ilike(f'%{query.table_name}%'))

        # 按 submission_status 升序排序
        user_tasks = user_tasks.order_by(FormSubmission.submission_status.asc())
    # 执行查询
    tasks = user_tasks.all()

    # 构造返回结果
    result_items = []
    for report, submission in tasks:
        result_item = {
            "user_id":user_id,
            "table_name": report.table_name,
            "report_id": report.report_id,
            "submission_status": submission.submission_status,
            "start_time": report.start_time,
            "end_time": report.end_time,
        }
        result_items.append(result_item)

    return {
        "code":200,
        "msg":"查询成功",
        "data": result_items
    }



@router.get("/report_fields")
@router.post("/report_fields")
async def get_report_fields(
    db: Session = Depends(get_db),
    # user_id: str = Query(None, description="用户ID"),
    report_id: str = Query(None, description="填报ID"),
    user_id = Depends(valid_access_token)
):
    # 检查用户ID和填报ID是否提供
    if not user_id or not report_id:
        raise HTTPException(status_code=400, detail="用户ID和填报ID是必填项")

    # 获取对应填报ID的数据表名称
    report = db.query(ReportManagement).filter(ReportManagement.report_id == report_id).first()
    if not report:
        raise HTTPException(status_code=404, detail="未找到对应的填报ID")

    data_table_name = report.data_table_name
    if not data_table_name:
        raise HTTPException(status_code=404, detail="未找到对应的数据表名称")

    # 检查用户是否有权限访问填报数据
    submission = db.query(FormSubmission).filter(
        FormSubmission.report_id == report_id,
        FormSubmission.user_id == user_id
    ).first()
    if not submission:
        raise HTTPException(status_code=403, detail="没有权限访问这个填报数据")

    # 使用SQLAlchemy的inspect功能来获取表的字段信息
    inspector = inspect(db.bind)
    columns = inspector.get_columns(data_table_name)

    # 构造返回结果
    result_fields = []
    for column in columns:
        if column['name'] not in ['collect_status', 'create_id', 'id', 'user_id']:
            result_field = {
                "field_name": column['name'],
                "field_comment": column.get('comment', '无注释')
            }
            result_fields.append(result_field)

    # 返回用户ID、填报ID和字段信息
    return {
        "code":200,
        "msg":"查询成功",
        "user_id": user_id,
        "report_id": report_id,
        "fields": result_fields
    }



class DataEntry(BaseModel):
    data: List[Dict[str, Any]]  # 数据列表，每个元素是一个字典，包含字段名和值

class DataItem(BaseModel):
    pass  # 用于动态接收键值对

class SubmitData(BaseModel):
    # user_id: int
    report_id: str
    data: List[Dict[str, str]]  # 数据列表，每个元素是一个字典，包含字段名和值

    class Config:
        arbitrary_types_allowed = True


@router.post("/submit_data")
async def submit_data(
    db: Session = Depends(get_db),
    submit_data: SubmitData = Body(...),
    user_id = Depends(valid_access_token)
):
    # 检查用户ID和填报ID是否提供
    if not user_id or not submit_data.report_id:
        raise HTTPException(status_code=400, detail="填报ID是必填项")

    # 获取对应填报ID的数据表名称
    report = db.query(ReportManagement).filter(ReportManagement.report_id == submit_data.report_id).first()
    if not report:
        raise HTTPException(status_code=404, detail="未找到对应的填报ID")

    data_table_name = report.data_table_name
    if not data_table_name:
        raise HTTPException(status_code=404, detail="未找到对应的数据表名称")

    if report.issued_status not in [2,'2']:
        raise HTTPException(status_code=404, detail="当前未发布，不可填写")

    is_collection = report.collection_status

    if is_collection == 2 or is_collection == '2':
        raise HTTPException(status_code=404, detail="管理员已收取信息，无法填写")

    # 检查用户是否有权限填报
    submission = db.query(FormSubmission).filter(
        FormSubmission.report_id == submit_data.report_id,
        FormSubmission.user_id == str(user_id)  # 确保user_id是字符串类型
    ).first()
    if not submission:
        raise HTTPException(status_code=403, detail="用户没有填报权限")


    current_time = datetime.now()

    if report.end_time < current_time:
        raise HTTPException(status_code=403, detail="填写时间已过")


    # 将数据写入数据库
    for item in submit_data.data:
        # 构造插入SQL语句
        columns = ', '.join(list(item.keys()) + ['create_id', 'user_id', 'collect_status'])
        values = ', '.join(
            [f":{k}" for k in item.keys()] + [f"'{report.creator_id}'", f"'{user_id}'", '1'])
        sql = f"INSERT INTO {data_table_name} ({columns}) VALUES ({values})"
        # 执行插入操作
        db.execute(text(sql), item)

    submission.submission_status = 1
    db.add(submission)
    # 提交事务
    db.commit()

    return {
        "code":200,
        "msg": "数据提交成功"
    }





class SubmissionQuery(BaseModel):
    # user_id: int  # 用户ID，必须是整数
    report_id: str  # 填报ID，必须是字符串


@router.post("/submission_status")
async def get_submission_status(
    db: Session = Depends(get_db),
        query: SubmissionQuery = Body(...),
    user_id = Depends(valid_access_token)
):
    # 检查用户ID和填报ID是否提供
    if not user_id or not query.report_id:
        raise HTTPException(status_code=400, detail="用户ID和填报ID是必填项")

    # 获取对应填报ID的数据表名称
    report = db.query(ReportManagement).filter(ReportManagement.report_id == query.report_id).first()
    if not report:
        raise HTTPException(status_code=404, detail="未找到对应的填报ID")

    data_table_name = report.data_table_name
    if not data_table_name:
        raise HTTPException(status_code=404, detail="未找到对应的数据表名称")

    # 获取填报情况
    submission = db.query(FormSubmission).filter(
        FormSubmission.report_id == query.report_id,
        FormSubmission.user_id == str(user_id)  # 确保user_id是字符串类型
    ).first()
    if not submission:
        raise HTTPException(status_code=404, detail="未找到对应的填报情况")

    start_time = report.start_time
    end_time = report.end_time

    start_time_str = start_time.strftime('%Y-%m-%d %H:%M:%S')
    end_time_str = end_time.strftime('%Y-%m-%d %H:%M:%S')

    # 使用SQLAlchemy的inspect功能来获取表的列信息
    inspector = Inspector.from_engine(db.bind)
    columns = inspector.get_columns(data_table_name)

    # 提取列名和列注释
    column_names = [column['name'] for column in columns]
    column_comments = {column['name']: column['comment'] for column in columns if 'comment' in column}

    # 构造返回结果
    result_items = []
    excluded_columns = ['id', 'user_id', 'create_id', 'collect_status']

    # 构建查询SQL
    query_sql = text(f"""
                SELECT * FROM {data_table_name} WHERE user_id = :user_id
            """)
    result = db.execute(query_sql, {"user_id": user_id})
    rows = result.fetchall()

    # 添加字段名和字段注释作为第一行
    first_row = {column: column_comments.get(column, '') for column in column_names if
                 column not in excluded_columns}
    result_items.append(first_row)

    for row in rows:
        # 过滤掉不需要的列
        filtered_row = {column: row[idx] for idx, column in enumerate(column_names) if
                        column not in excluded_columns}
        result_items.append(filtered_row)

    result = {
        "code": 200,
        'msg': '查询成功',
        'start_time':start_time_str,
        "end_time":end_time_str,
        "data": result_items}
    return result


# 辅助函数：根据字段备注获取表中所有匹配字段名
def get_columns_with_comment_like(
    inspector: Inspector, table_name: str, comment_like: str
) -> List[str]:
    columns = inspector.get_columns(table_name)
    matching_columns = [column['name'] for column in columns if column.get('comment') and comment_like in column['comment']]
    return matching_columns

# 辅助函数：检查是否有字段备注匹配
def has_matching_column_comments(
    inspector: Inspector, table_name: str, comment_like: str
) -> bool:
    return bool(get_columns_with_comment_like(inspector, table_name, comment_like))

#【数据档案管理】-列表
@router.post("/reports_by_creator")
@router.get("/reports_by_creator")
async def get_reports_by_creator(
    field_comment: Optional[str] = Query(None, description="Optional comment of the field to match"),
    page: int = Query(default=1, gt=0),  # 分页参数：当前页码，默认为1
    pageSize: int = Query(default=10, gt=0),  # 分页参数：每页大小，默认为10
    db: Session = Depends(get_db),
    creator_id=Depends(valid_access_token)
):
    # 获取数据库Inspector
    inspector: Inspector = inspect(db.bind)

    # 查询 ReportManagement 表以获取所有相关的记录
    query = db.query(ReportManagement).filter(ReportManagement.creator_id == creator_id)

    query = query.order_by(
        asc(ReportManagement.collection_status)  # 按 collection_status 升序
    )

    # 计算总数
    total_count = query.count()

    # 分页查询
    offset = (page - 1) * pageSize
    reports = query.offset(offset).limit(pageSize).all()

    # 存储结果
    results = []

    for report in reports:
        # 检查是否存在 data_table_name
        if not report.data_table_name:
            continue  # 如果没有 data_table_name，跳过这个 report

        # 检查是否有字段备注匹配 field_comment
        if field_comment and not has_matching_column_comments(inspector, report.data_table_name, field_comment):
            continue  # 如果没有匹配的字段备注，跳过这个 report

        # 如果匹配成功，添加到结果中
        collection_time_str = report.collection_time.isoformat().replace('T', ' ') if report.collection_time else None

        #在这里判断数据库

        #收取状态
        collection_status = report.collection_status
        issued_status = report.issued_status
        #结束时间
        end_time = report.end_time
        current_time = datetime.now()

        if end_time < current_time and collection_status in [0,'0'] and issued_status in [2,'2']:
            print("符合自动收取")
            report.collection_status=2
            report.collection_time = current_time

            db.add(report)
            db.commit()
            db.refresh(report)

        results.append({
            "table_name": report.table_name,
            "collection_status": report.collection_status,
            "collection_time": collection_time_str,
            "report_id": report.report_id
        })

    # 如果没有找到任何记录，抛出404异常
    if not results:
        raise HTTPException(status_code=404, detail="没有找到与该创建人ID相关的记录")

    # 构造分页结果
    result = {
        "code": 200,
        "msg": "查询成功",
        "total": total_count,
        "totalPages": (total_count + pageSize - 1) // pageSize,
        "page": page,
        "pageSize": pageSize,
        "data": results
    }

    return result





@router.put("/update_collection_status/")
async def update_collection_status(
    # creator_id: str,
    report_id: str,
    new_status: int = Query(..., description="New collection status, must be 0, 1, or 2"),
    db: Session = Depends(get_db),
    creator_id = Depends(valid_access_token)
):
    # 检查 new_status 是否为允许的值之一
    if new_status not in (0, 1, 2):
        raise HTTPException(status_code=400, detail="Invalid collection status value")

    # 查询 ReportManagement 表以获取对应记录
    report = db.query(ReportManagement).filter(
        ReportManagement.creator_id == creator_id,
        ReportManagement.report_id == report_id
    ).first()


    # 如果没有找到记录，返回404
    if not report:
        raise HTTPException(status_code=404, detail="Report not found")

    if report.collection_status == 2 or report.collection_status == '2':
        raise HTTPException(status_code=404, detail="当前已收取，无需重复收取")


    current_time_str = datetime.now().strftime("%Y%m%d%H%M%S")
    # 更新 collection_status
    report.collection_status = new_status
    report.collection_time = current_time_str
    db.add(report)
    db.commit()
    db.refresh(report)

    return {
        "code":200,
        "msg": "更新成功",
        "new_status": new_status
    }




class ReportQuery(BaseModel):
    # creator_id: str  # 创建人ID，必须是字符串
    report_id: str  # 填报ID，必须是字符串

@router.get("/dataArchiveDetails/")
async def get_records_by_creator_and_report(
    query: ReportQuery = Depends(),
    db: Session = Depends(get_db),
    creator_id = Depends(valid_access_token)
):
    # creator_id
    # 查询 ReportManagement 表以获取对应记录
    report = db.query(ReportManagement).filter(
        ReportManagement.creator_id == creator_id,
        ReportManagement.report_id == query.report_id
    ).first()

    # 如果没有找到记录，返回404
    if not report:
        raise HTTPException(status_code=404, detail="Report not found")

    # 如果没有 data_table_name，返回404
    if not report.data_table_name:
        raise HTTPException(status_code=404, detail="Data table name not found")

    # 查询工单表所有信息，并关联用户表匹配到用户名字
    query_sql = text(f"""
        SELECT w.*, u.user_name
        FROM {report.data_table_name} w
        LEFT JOIN sys_user u ON w.user_id = u.user_id
    """)
    result = db.execute(query_sql)
    rows = result.fetchall()

    # 使用SQLAlchemy的inspect功能来获取表的列信息
    inspector = inspect(db.bind)
    columns = inspector.get_columns(report.data_table_name)

    # 提取列名和列注释
    column_names = [column['name'] for column in columns]
    column_comments = {column['name']: column['comment'] for column in columns if 'comment' in column}

    # 构造字段信息
    columns_info = []
    for column in column_names:
        if column not in ['id', 'user_id', 'create_id', 'collect_status']:
            columns_info.append({
                'prop': column,
                'label': column_comments.get(column, '')
            })

    # 构造返回结果
    rows_data = []
    for row in rows:
        # 过滤掉不需要的列，并添加到结果中
        filtered_row = {column: row[idx] for idx, column in enumerate(column_names)
                        if column not in ['id', 'user_id', 'create_id', 'collect_status']}
        filtered_row['user_name'] = row[-1]  # 添加用户昵称
        rows_data.append(filtered_row)

    # 获取报告的开始和结束时间，并格式化为字符串
    start_time_str = report.start_time.strftime('%Y-%m-%d %H:%M:%S') if report.start_time else None
    end_time_str = report.end_time.strftime('%Y-%m-%d %H:%M:%S') if report.end_time else None

    return {
        "code": 200,
        'msg': '查询成功',
        'start_time': start_time_str,
        "end_time": end_time_str,
        "columns": columns_info,
        "rows": rows_data
    }









from fastapi import status
@router.get("/export_to_excel")
@router.post("/export_to_excel")
async def export_to_excel(
    report_id: str = Query(..., description="填报ID"),
    db: Session = Depends(get_db),
    creator_id: str = Depends(valid_access_token)
):
    # 获取对应填报ID的数据表名称
    report = db.query(ReportManagement).filter(
        ReportManagement.report_id == report_id,
        ReportManagement.creator_id == creator_id
    ).first()
    if not report:
        raise HTTPException(status_code=404, detail="未找到对应的填报ID")

    data_table_name = report.data_table_name
    if not data_table_name:
        raise HTTPException(status_code=404, detail="未找到对应的数据表名称")

    # 获取表结构（用户填报的字段）
    inspector = inspect(db.bind)
    columns = inspector.get_columns(data_table_name)

    # 提取用户填报的字段注释
    user_report_columns = [col for col in columns if col['name'] not in ['id', 'create_id', 'collect_status', 'add_time', 'user_id']]
    column_comments = [col.get('comment', '') for col in user_report_columns]

    # 构建查询SQL，关联 sys_user 表获取 nick_name
    query_sql = f"""
        SELECT su.nick_name AS user_name, {', '.join([f'rd.{col["name"]}' for col in user_report_columns])}
        FROM {data_table_name} rd
        JOIN sys_user su ON rd.user_id = su.user_id
    """

    # 使用 text 包装查询字符串
    result = db.execute(text(query_sql))
    rows = result.fetchall()

    # 将查询结果转换为 DataFrame
    df = pd.DataFrame(rows, columns=["user_name"] + column_comments)

    # 将 DataFrame 导出为 Excel 文件
    output = BytesIO()
    with pd.ExcelWriter(output, engine='openpyxl') as writer:
        df.to_excel(writer, index=False, sheet_name='填报数据')

    # 设置响应头
    output.seek(0)
    headers = {
        'Content-Disposition': 'attachment; filename="report_data.xlsx"',
        'Content-Type': 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet'
    }

    # 返回文件流
    return StreamingResponse(output, headers=headers)