martsforever
119 lines
4.3 KiB
Python
119 lines
4.3 KiB
Python
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="每页数量")
|