135 lines
5.0 KiB
Python
135 lines
5.0 KiB
Python
|
|
|||
|
|
from fastapi import APIRouter, Query, Body, HTTPException
|
|||
|
|
from uuid import UUID
|
|||
|
|
from .schema import Create, Update, Out, OutList
|
|||
|
|
from ..models import Shop
|
|||
|
|
from utils.decorators import handle_exceptions_unified
|
|||
|
|
from utils.time_tool import parse_time
|
|||
|
|
from utils.out_base import CommonOut
|
|||
|
|
|
|||
|
|
app = APIRouter()
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 创建店铺
|
|||
|
|
@app.post("", response_model=Out, description='创建店铺', summary='创建店铺')
|
|||
|
|
@handle_exceptions_unified()
|
|||
|
|
async def post(item: Create = Body(..., description='创建数据')):
|
|||
|
|
"""
|
|||
|
|
创建店铺记录
|
|||
|
|
"""
|
|||
|
|
res = await Shop.create(**item.model_dump())
|
|||
|
|
if not res:
|
|||
|
|
raise HTTPException(status_code=400, detail='创建失败')
|
|||
|
|
return res
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 查询店铺
|
|||
|
|
@app.get("", response_model=OutList, description='获取店铺', summary='获取店铺')
|
|||
|
|
@handle_exceptions_unified()
|
|||
|
|
async def gets(
|
|||
|
|
id: UUID | None = Query(None, description='主键ID'),
|
|||
|
|
province: str | None = Query(None, description='省份'),
|
|||
|
|
city: str | None = Query(None, description='城市'),
|
|||
|
|
street: str | None = Query(None, description='街道'),
|
|||
|
|
shop_name: str | None = Query(None, description='店铺名称'),
|
|||
|
|
shop_number: str | None = Query(None, description='店铺号码'),
|
|||
|
|
order_by: str | None = Query('create_time', description='排序字段',
|
|||
|
|
regex='^(-)?(id|province|city|street|shop_name|create_time|update_time)$'),
|
|||
|
|
res_count: bool = Query(False, description='是否返回总数'),
|
|||
|
|
create_time_start: str | int | None = Query(
|
|||
|
|
None, description='创建时间开始 (支持 YYYY-MM-DD / YYYY-MM-DD HH:mm:ss / 13位时间戳)'),
|
|||
|
|
create_time_end: str | int | None = Query(
|
|||
|
|
None, description='创建时间结束 (支持 YYYY-MM-DD / YYYY-MM-DD HH:mm:ss / 13位时间戳)'),
|
|||
|
|
update_time_start: str | int | None = Query(
|
|||
|
|
None, description='更新时间开始 (支持 YYYY-MM-DD / YYYY-MM-DD HH:mm:ss / 13位时间戳)'),
|
|||
|
|
update_time_end: str | int | None = Query(
|
|||
|
|
None, description='更新时间结束 (支持 YYYY-MM-DD / YYYY-MM-DD HH:mm:ss / 13位时间戳)'),
|
|||
|
|
page: int = Query(1, ge=1, description='页码'),
|
|||
|
|
limit: int = Query(10, ge=1, le=1000, description='每页数量'),
|
|||
|
|
):
|
|||
|
|
"""
|
|||
|
|
获取店铺列表
|
|||
|
|
"""
|
|||
|
|
query = Shop.all()
|
|||
|
|
if id:
|
|||
|
|
query = query.filter(id=id)
|
|||
|
|
if province:
|
|||
|
|
query = query.filter(province=province)
|
|||
|
|
if city:
|
|||
|
|
query = query.filter(city=city)
|
|||
|
|
if street:
|
|||
|
|
query = query.filter(street=street)
|
|||
|
|
if shop_name:
|
|||
|
|
query = query.filter(shop_name=shop_name)
|
|||
|
|
if shop_number:
|
|||
|
|
query = query.filter(shop_number=shop_number)
|
|||
|
|
if create_time_start:
|
|||
|
|
query = query.filter(create_time__gte=parse_time(create_time_start))
|
|||
|
|
if create_time_end:
|
|||
|
|
query = query.filter(create_time__lte=parse_time(
|
|||
|
|
create_time_end, is_end=True))
|
|||
|
|
if update_time_start:
|
|||
|
|
query = query.filter(update_time__gte=parse_time(update_time_start))
|
|||
|
|
if update_time_end:
|
|||
|
|
query = query.filter(update_time__lte=parse_time(
|
|||
|
|
update_time_end, is_end=True))
|
|||
|
|
|
|||
|
|
if order_by:
|
|||
|
|
query = query.order_by(order_by)
|
|||
|
|
|
|||
|
|
if res_count:
|
|||
|
|
count = await query.count()
|
|||
|
|
else:
|
|||
|
|
count = -1
|
|||
|
|
offset = (page - 1) * limit # 计算偏移量
|
|||
|
|
query = query.limit(limit).offset(offset) # 应用分页
|
|||
|
|
|
|||
|
|
res = await query
|
|||
|
|
if not res:
|
|||
|
|
raise HTTPException(status_code=404, detail='店铺不存在')
|
|||
|
|
num = len(res)
|
|||
|
|
return OutList(count=count, num=num, items=res)
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 更新店铺
|
|||
|
|
@app.put("", response_model=Out, description='更新店铺', summary='更新店铺')
|
|||
|
|
@handle_exceptions_unified()
|
|||
|
|
async def put(id: UUID = Query(..., description='主键ID'),
|
|||
|
|
item: Update = Body(..., description='更新数据'),
|
|||
|
|
):
|
|||
|
|
"""
|
|||
|
|
部分更新店铺，只更新传入的非空字段
|
|||
|
|
"""
|
|||
|
|
# 检查店铺是否存在
|
|||
|
|
secret = await Shop.get_or_none(id=id)
|
|||
|
|
if not secret:
|
|||
|
|
raise HTTPException(status_code=404, detail='店铺不存在')
|
|||
|
|
|
|||
|
|
# 获取要更新的字段（排除None值的字段）
|
|||
|
|
update_data = item.model_dump(exclude_unset=True)
|
|||
|
|
|
|||
|
|
# 如果没有要更新的字段
|
|||
|
|
if not update_data:
|
|||
|
|
raise HTTPException(status_code=400, detail='没有要更新的字段')
|
|||
|
|
|
|||
|
|
# 更新店铺字段
|
|||
|
|
await secret.update_from_dict(update_data)
|
|||
|
|
await secret.save()
|
|||
|
|
return secret
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 删除店铺
|
|||
|
|
|
|||
|
|
@app.delete("", response_model=CommonOut, description='删除店铺', summary='删除店铺')
|
|||
|
|
@handle_exceptions_unified()
|
|||
|
|
async def delete(id: UUID = Query(..., description='主键ID'),
|
|||
|
|
):
|
|||
|
|
"""删除店铺"""
|
|||
|
|
secret = await Shop.get_or_none(id=id)
|
|||
|
|
if not secret:
|
|||
|
|
raise HTTPException(status_code=404, detail='店铺不存在')
|
|||
|
|
await secret.delete()
|
|||
|
|
# Tortoise ORM 单个实例的 delete() 方法返回 None，而不是删除的记录数
|
|||
|
|
# 删除成功时手动返回 1，如果有异常会被装饰器捕获
|
|||
|
|
return CommonOut(count=1)
|