feat: add_model_routes,简单的通用接口生成

This commit is contained in:
martsforever
2026-04-04 18:57:18 +08:00
parent 81c73b86ac
commit ef945adc8c
4 changed files with 138 additions and 89 deletions
-87
View File
@@ -19,90 +19,3 @@ class LlmDemoModel(BasicModel, table=True):
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="每页数量")
+13
View File
@@ -0,0 +1,13 @@
from sqlmodel import Field
from app.model.BasicModel import BasicModel
from app.utils.model_utils import FormattedDatetime, FormattedDate, FormattedDecimal
class ResumeTemplateModel(BasicModel, table=True):
__tablename__ = "llm_resume_template"
source_code: str | None = Field(default=None, description="简历模板TSX代码")
thumb_image: str | None = Field(default=None, description="简历模板缩略图")
default_primary: str | None = Field(default=None, description="默认主题色")
default_secondary: str | None = Field(default=None, description="默认次级色")