diff --git a/app/model/LlmUserModel.py b/app/model/LlmUserModel.py new file mode 100644 index 0000000..22b9b7c --- /dev/null +++ b/app/model/LlmUserModel.py @@ -0,0 +1,118 @@ +import uuid +from datetime import datetime, timezone, timedelta, date +from typing import List + +from fastapi import FastAPI, APIRouter, HTTPException +from pydantic import BaseModel +from sqlmodel import SQLModel, Field, select + +from app.utils.mysql_utils import AsyncSessionDep + +# 定义北京时区(UTC+8) +beijing_timezone = timezone(timedelta(hours=8)) + +# 定义获取当前北京时区时间的匿名函数,用于默认值生成 +current_datetime = lambda: datetime.now(beijing_timezone) + + +class LlmUserModel(SQLModel, table=True): + __tablename__ = "llm_user" + + # 唯一标识字段,主键,默认为None(通常由系统生成),描述为“唯一标识,编号” + id: str = Field(default=None, primary_key=True, description="唯一标识,编号") + # 创建时间字段,默认值为当前北京时区时间,描述为“创建时间” + created_at: datetime = Field(default_factory=current_datetime, description="创建时间") + # 更新时间字段,默认值为当前北京时区时间,描述为“更新时间” + updated_at: datetime = Field(default_factory=current_datetime, description="更新时间") + # 创建人ID字段,默认为None,描述为“创建人id” + created_by: str | None = Field(default=None, description="创建人id") + # 更新人ID字段,默认为None,描述为“更新人id” + updated_by: str | None = Field(default=None, description="更新人id") + + full_name: str = Field(default=None, description="用户名称") + username: str = Field(default=None, description="用户名") + password: str = Field(default=None, description="用户密码") + member_start: datetime = Field(default=None, description="开通会员时间") + member_end: datetime = Field(default=None, description="会员截止到期时间") + + +def add_llm_user_route(app: FastAPI): + route_prefix = "/llm_user" + router = APIRouter(prefix=route_prefix, tags=[route_prefix]) + + @router.post('/list') + async def _list(body: ModelQuerySchema, session: AsyncSessionDep): + """ + 获取所有用户列表 + """ + + # 构建SQL查询执行对象 + query = select(LlmUserModel) + # 计算偏移量(跳过前N条),并查询比一页多1条的记录(用于判断是否有下一页) + query = query.offset(body.page * body.page_size).limit(body.page_size + 1) + result = await session.execute(query) + query_cls_list: List[LlmUserModel] = 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: LlmUserModel, 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: LlmUserModel, session: AsyncSessionDep): + """ + 更新用户 + """ + query = select(LlmUserModel).where(LlmUserModel.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: LlmUserModel, session: AsyncSessionDep): + """ + 更新用户 + """ + query = select(LlmUserModel).where(LlmUserModel.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="每页数量") diff --git a/app/routes.py b/app/routes.py index c6d9436..1821dcc 100644 --- a/app/routes.py +++ b/app/routes.py @@ -2,6 +2,7 @@ from app.controller.add_custom_stream_route import add_custom_stream_route from app.controller.add_docs_route import add_docs_route from app.controller.add_graph_proxy_route import add_graph_proxy_route from app.controller.add_test_route import add_test_route +from app.model.LlmUserModel import add_llm_user_route # /*@formatter:off*/ routes = [ @@ -9,5 +10,6 @@ routes = [ add_test_route, # 测试接口 add_graph_proxy_route, # 代理自定义工作流 add_custom_stream_route, # 自定义流式接口 + add_llm_user_route, # LlmUser 测试用户模块 ] # /*@formatter:on*/