#!/usr/bin/env python
# -*- encoding:utf-8 -*-

import asyncio
from datetime import datetime
from http.client import HTTPException

from openpyxl.reader.excel import load_workbook
from starlette.responses import JSONResponse
from fastapi import FastAPI
import config

app = FastAPI()

def json_serializable(obj):
    if isinstance(obj, datetime):
        return obj.isoformat()  # 转换为 ISO 格式的字符串
    raise TypeError(f"Type {type(obj)} not serializable")

@app.get("/read-excel")
async def read_excel_rows():
    try:
        rows="1,2,3,4,5"
        # 加载Excel工作簿
        file_path =config.osp.join(config.BASE_DIR, 'output/','4834ed97e0ba477b9d239560e4b12be6.xlsx')
        workbook = load_workbook(filename=file_path)
        sheet = workbook.active  # 或者使用workbook.get_sheet_by_name('Sheet1')

        # 获取要读取的行号列表
        # row_numbers = [int(r) for r in rows.split(',') if r.isdigit()]

        # 读取指定行的数据
        data = []

        for row in sheet:
            row_data = [json_serializable(cell.value) if isinstance(cell.value, datetime) else cell.value for cell in row]
            data.append(row_data)
        return JSONResponse(content={"data": data})

    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))

if __name__ == '__main__':

    import uvicorn
    uvicorn.run(app, host='0.0.0.0', port=9001)