233 lines
7.8 KiB
Python
233 lines
7.8 KiB
Python
from fastapi import APIRouter, Depends
|
||
from sqlalchemy.orm import Session
|
||
from typing import List, Optional, Dict, Any
|
||
from sqlalchemy import text
|
||
from app.models.merchant import (
|
||
MerchantDB,
|
||
MerchantCreate,
|
||
MerchantUpdate,
|
||
MerchantInfo)
|
||
from app.models.merchant_category import MerchantCategoryDB
|
||
from app.models.database import get_db
|
||
from app.api.deps import get_admin_user
|
||
from app.models.user import UserDB
|
||
from app.core.response import success_response, error_response, ResponseModel
|
||
from app.models.merchant_pay_order import MerchantPayOrderDB
|
||
from sqlalchemy.sql import func, desc
|
||
from app.models.merchant_product import MerchantProductDB, ProductStatus
|
||
|
||
router = APIRouter()
|
||
|
||
@router.post("", response_model=ResponseModel)
|
||
async def create_merchant(
|
||
merchant: MerchantCreate,
|
||
db: Session = Depends(get_db),
|
||
admin: UserDB = Depends(get_admin_user)
|
||
):
|
||
"""创建商家（管理员）"""
|
||
# 创建商家基本信息
|
||
merchant_data = merchant.model_dump()
|
||
db_merchant = MerchantDB(**merchant_data)
|
||
db.add(db_merchant)
|
||
|
||
try:
|
||
db.commit()
|
||
db.refresh(db_merchant)
|
||
return success_response(data=MerchantInfo.model_validate(db_merchant))
|
||
except Exception as e:
|
||
db.rollback()
|
||
return error_response(code=500, message=f"创建失败: {str(e)}")
|
||
|
||
@router.put("/{merchant_id}", response_model=ResponseModel)
|
||
async def update_merchant(
|
||
merchant_id: int,
|
||
merchant: MerchantUpdate,
|
||
db: Session = Depends(get_db),
|
||
admin: UserDB = Depends(get_admin_user)
|
||
):
|
||
"""更新商家信息（管理员）"""
|
||
db_merchant = db.query(MerchantDB).filter(
|
||
MerchantDB.id == merchant_id
|
||
).first()
|
||
|
||
if not db_merchant:
|
||
return error_response(code=404, message="商家不存在")
|
||
|
||
# 如果要更新用户ID，先验证用户是否存在
|
||
if merchant.user_id is not None:
|
||
user_exists = db.query(UserDB).filter(
|
||
UserDB.userid == merchant.user_id
|
||
).first()
|
||
if not user_exists:
|
||
return error_response(code=400, message="指定的用户不存在")
|
||
|
||
# 只更新传入的非空字段
|
||
update_data = merchant.model_dump(exclude_unset=True)
|
||
for key, value in update_data.items():
|
||
if value is not None: # 只更新非空值
|
||
setattr(db_merchant, key, value)
|
||
|
||
try:
|
||
db.commit()
|
||
db.refresh(db_merchant)
|
||
|
||
# 获取更新后的完整信息（包括用户信息）
|
||
updated_merchant = db.query(
|
||
MerchantDB,
|
||
UserDB.phone.label('user_phone'),
|
||
UserDB.nickname.label('user_nickname')
|
||
).join(
|
||
UserDB,
|
||
MerchantDB.user_id == UserDB.userid
|
||
).filter(
|
||
MerchantDB.id == merchant_id
|
||
).first()
|
||
|
||
# 构建返回数据
|
||
merchant_info = MerchantInfo.model_validate(updated_merchant.MerchantDB)
|
||
merchant_data = merchant_info.model_dump()
|
||
merchant_data.update({
|
||
'user_phone': updated_merchant.user_phone,
|
||
'user_nickname': updated_merchant.user_nickname
|
||
})
|
||
|
||
return success_response(data=merchant_data)
|
||
except Exception as e:
|
||
db.rollback()
|
||
return error_response(code=500, message=f"更新失败: {str(e)}")
|
||
|
||
@router.get("/{merchant_id}", response_model=ResponseModel)
|
||
async def get_merchant(
|
||
merchant_id: int,
|
||
db: Session = Depends(get_db)
|
||
):
|
||
"""获取商家详情"""
|
||
merchant = db.query(
|
||
MerchantDB,
|
||
UserDB.phone.label('user_phone'),
|
||
UserDB.nickname.label('user_nickname')
|
||
).join(
|
||
UserDB,
|
||
MerchantDB.user_id == UserDB.userid
|
||
).filter(
|
||
MerchantDB.id == merchant_id
|
||
).first()
|
||
|
||
if not merchant:
|
||
return error_response(code=404, message="商家不存在")
|
||
|
||
# 构建返回数据
|
||
merchant_info = MerchantInfo.model_validate(merchant.MerchantDB)
|
||
merchant_data = merchant_info.model_dump()
|
||
merchant_data.update({
|
||
'user_phone': merchant.user_phone,
|
||
'user_nickname': merchant.user_nickname
|
||
})
|
||
|
||
return success_response(data=merchant_data)
|
||
|
||
@router.get("", response_model=ResponseModel)
|
||
async def list_merchants(
|
||
longitude: Optional[float] = None,
|
||
latitude: Optional[float] = None,
|
||
category_id: Optional[int] = None,
|
||
skip: int = 0,
|
||
limit: int = 20,
|
||
db: Session = Depends(get_db)
|
||
):
|
||
"""获取商家列表，支持经纬度排序和分类过滤"""
|
||
query = db.query(
|
||
MerchantDB,
|
||
MerchantCategoryDB.name.label('category_name'),
|
||
UserDB.phone.label('user_phone'),
|
||
UserDB.nickname.label('user_nickname')
|
||
).outerjoin(
|
||
MerchantCategoryDB,
|
||
MerchantDB.category_id == MerchantCategoryDB.id
|
||
).join(
|
||
UserDB,
|
||
MerchantDB.user_id == UserDB.userid
|
||
)
|
||
|
||
# 添加分类过滤
|
||
if category_id is not None:
|
||
query = query.filter(MerchantDB.category_id == category_id)
|
||
|
||
# 根据经纬度排序
|
||
if longitude is not None and latitude is not None:
|
||
query = query.add_columns(
|
||
text("ST_Distance_Sphere(point(longitude, latitude), point(:lon, :lat)) as distance")
|
||
).params(lon=longitude, lat=latitude)
|
||
# 默认按距离排序
|
||
query = query.order_by(text("distance"))
|
||
else:
|
||
# 如果没有经纬度，则按创建时间排序
|
||
query = query.order_by(MerchantDB.create_time.desc())
|
||
|
||
# 添加一个空的距离列，保持返回结构一致
|
||
query = query.add_columns(text("NULL as distance"))
|
||
|
||
merchants = query.offset(skip).limit(limit).all()
|
||
|
||
# 获取商家在线买单数量
|
||
merchant_ids = [m[0].id for m in merchants]
|
||
pay_order_counts = dict(
|
||
db.query(
|
||
MerchantPayOrderDB.merchant_id,
|
||
func.count(MerchantPayOrderDB.id).label('count')
|
||
).filter(
|
||
MerchantPayOrderDB.merchant_id.in_(merchant_ids)
|
||
).group_by(MerchantPayOrderDB.merchant_id).all()
|
||
)
|
||
|
||
# 获取商家最新或限购商品
|
||
merchant_products = {}
|
||
for merchant_id in merchant_ids:
|
||
# 先查询有限购的商品
|
||
product = db.query(MerchantProductDB).filter(
|
||
MerchantProductDB.merchant_id == merchant_id,
|
||
MerchantProductDB.status == ProductStatus.LISTING,
|
||
MerchantProductDB.purchase_limit > 0
|
||
).order_by(
|
||
MerchantProductDB.create_time.desc()
|
||
).first()
|
||
|
||
# 如果没有限购商品，则查询最新上架的商品
|
||
if not product:
|
||
product = db.query(MerchantProductDB).filter(
|
||
MerchantProductDB.merchant_id == merchant_id,
|
||
MerchantProductDB.status == ProductStatus.LISTING
|
||
).order_by(
|
||
MerchantProductDB.create_time.desc()
|
||
).first()
|
||
|
||
if product:
|
||
merchant_products[merchant_id] = {
|
||
"product_id": product.id,
|
||
"product_name": product.name,
|
||
"product_image": product.optimized_image_url,
|
||
"product_price": float(product.sale_price),
|
||
"purchase_limit": product.purchase_limit
|
||
}
|
||
|
||
# 处理返回结果
|
||
merchant_list = [{
|
||
**MerchantInfo.model_validate(m[0]).model_dump(),
|
||
"category_name": m[1],
|
||
"user_phone": m[2],
|
||
"user_nickname": m[3],
|
||
"online_pay_count": pay_order_counts.get(m[0].id, 0),
|
||
"featured_product": merchant_products.get(m[0].id),
|
||
"distance": round(m[4]) if longitude is not None and latitude is not None else None
|
||
} for m in merchants]
|
||
|
||
# 获取总数（需要考虑分类过滤）
|
||
total_query = db.query(MerchantDB)
|
||
if category_id is not None:
|
||
total_query = total_query.filter(MerchantDB.category_id == category_id)
|
||
total = total_query.count()
|
||
|
||
return success_response(data={
|
||
"total": total,
|
||
"items": merchant_list
|
||
}) |