import uuid
from decimal import Decimal
from typing import List

from fastapi import FastAPI, APIRouter, HTTPException
from pydantic import BaseModel
from sqlmodel import Field, select

from app.model.BasicModel import BasicModel
from app.utils.model_utils import FormattedDatetime, FormattedDate, FormattedDecimal
from app.utils.mysql_utils import AsyncSessionDep


class LlmDemoModel(BasicModel, table=True):
  __tablename__ = "llm_demo"

  full_name: str | None = Field(default=None, description="用户名称")
  datetime_start: FormattedDatetime | None = Field(default=None, description="开通会员时间")
  datetime_end: FormattedDatetime | None = Field(default=None, description="会员截止到期时间")
  birthday: FormattedDate | None = Field(default=None, description="生日")
  amount: FormattedDecimal | None = Field(default=Decimal(0), description="金额")


def add_llm_demo_route(app: FastAPI):
  route_prefix = "/llm_demo"
  router = APIRouter(prefix=route_prefix, tags=[route_prefix])

  @router.post('/list')
  async def _list(
    session: AsyncSessionDep,
    body: ModelQuerySchema = ModelQuerySchema(),
  ):
    """
    获取所有用户列表
    """

    # 构建SQL查询执行对象
    query = select(LlmDemoModel).order_by(
      LlmDemoModel.created_at.desc()
    )
    # 计算偏移量（跳过前N条），并查询比一页多1条的记录（用于判断是否有下一页）
    query = query.offset(body.page * body.page_size).limit(body.page_size + 1)
    result = await session.execute(query)
    query_cls_list: List[LlmDemoModel] = result.scalars().all()
    has_next = len(query_cls_list) == body.page_size + 1
    # 若有下一页，移除多查询的那一条记录
    if has_next:
      query_cls_list.pop()
    return {
      "list": query_cls_list,
      "has_next": has_next
    }

  @router.post('/insert')
  async def _insert(body: LlmDemoModel, session: AsyncSessionDep):
    """
    插入用户
    """
    if not body.id:
      body.id = str(uuid.uuid4())
    session.add(body)
    await session.commit()
    return body

  @router.post('/update')
  async def _update(body: LlmDemoModel, session: AsyncSessionDep):
    """
    更新用户
    """
    query = select(LlmDemoModel).where(LlmDemoModel.id == body.id)
    result = await session.execute(query)
    query_cls = result.scalars().first()
    if not query_cls:
      raise HTTPException(status_code=500, detail=f"Update row with id:{body.id} not found")

    update_data_exclude_true = body.model_dump(exclude_unset=True, exclude={'id'})
    update_data_exclude_false = body.model_dump(exclude={'id'})

    print(f"update_data_exclude_true: {update_data_exclude_true}")
    print(f"update_data_exclude_false: {update_data_exclude_false}")

    update_data = body.model_dump(exclude_unset=True, exclude={'id'})
    for field, value in update_data.items():
      setattr(query_cls, field, value)
    session.add(query_cls)
    await session.commit()
    return query_cls

  @router.post('/delete')
  async def _delete(body: LlmDemoModel, session: AsyncSessionDep):
    """
    更新用户
    """
    query = select(LlmDemoModel).where(LlmDemoModel.id == body.id)
    result = await session.execute(query)
    query_cls = result.scalars().first()
    if not query_cls:
      return {"affect_row_count": 0}
    await session.delete(query_cls)
    await session.commit()
    return {"affect_row_count": 1}

  app.include_router(router)


class ModelQuerySchema(BaseModel):
  page: int = Field(default=0, description="页码")
  page_size: int = Field(default=5, description="每页数量")