from fastapi import APIRouter, Depends, HTTPException from sqlalchemy.orm import Session from typing import List from app.models.merchant_order import ( MerchantOrderDB, MerchantOrderCreate, MerchantOrderInfo, generate_order_id, generate_verify_code, MerchantOrderStatus ) from app.models.merchant_product import MerchantProductDB from app.models.database import get_db from app.api.deps import get_current_user, get_admin_user from app.models.user import UserDB from app.core.response import success_response, error_response, ResponseModel from datetime import datetime, timezone from app.models.merchant import MerchantDB from app.models.point import PointRecordDB from app.core.account import AccountManager router = APIRouter() @router.post("", response_model=ResponseModel) async def create_order( order: MerchantOrderCreate, db: Session = Depends(get_db), current_user: UserDB = Depends(get_current_user) ): """创建商家订单""" # 检查商品是否存在 product = db.query(MerchantProductDB).filter( MerchantProductDB.id == order.merchant_product_id ).first() if not product: return error_response(code=404, message="商品不存在") # 计算积分抵扣 available_points = current_user.points or 0 deduct_amount = min( float(product.max_deduct_points), available_points ) if available_points > 0 else 0 # 计算实际支付金额 original_amount = float(product.sale_price) pay_amount = max(0, original_amount - deduct_amount) # 生成订单号和核销码 while True: order_id = generate_order_id() verify_code = generate_verify_code() exists = db.query(MerchantOrderDB).filter( (MerchantOrderDB.order_id == order_id) | (MerchantOrderDB.order_verify_code == verify_code) ).first() if not exists: break # 创建订单 db_order = MerchantOrderDB( order_id=order_id, user_id=current_user.userid, merchant_product_id=order.merchant_product_id, order_amount=original_amount, deduct_amount=deduct_amount, pay_amount=pay_amount, status=MerchantOrderStatus.CREATED, order_verify_code=verify_code ) # 扣除用户积分并记录日志 if deduct_amount > 0: current_user.points -= deduct_amount # 记录积分变动日志 point_record = PointRecordDB( user_id=current_user.userid, points=-float(deduct_amount), # 负数表示扣减 description=f"订单消费抵扣" ) db.add(point_record) db.add(db_order) try: db.commit() db.refresh(db_order) return success_response(data=MerchantOrderInfo.model_validate(db_order)) except Exception as e: db.rollback() return error_response(code=500, message=f"创建订单失败: {str(e)}") @router.get("/user", response_model=ResponseModel) async def get_user_orders( skip: int = 0, limit: int = 20, db: Session = Depends(get_db), current_user: UserDB = Depends(get_current_user) ): """获取用户的订单列表""" orders = db.query( MerchantOrderDB, MerchantProductDB.name.label('product_name'), MerchantProductDB.image_url.label('product_image'), MerchantDB.name.label('merchant_name'), MerchantDB.latitude.label('merchant_latitude'), MerchantDB.longitude.label('merchant_longitude'), MerchantDB.phone.label('merchant_phone') ).join( MerchantProductDB, MerchantOrderDB.merchant_product_id == MerchantProductDB.id ).join( MerchantDB, MerchantProductDB.merchant_id == MerchantDB.id ).filter( MerchantOrderDB.user_id == current_user.userid ).order_by( MerchantOrderDB.create_time.desc() ).offset(skip).limit(limit).all() # 构建返回数据 order_list = [{ "id": order.MerchantOrderDB.id, "order_id": order.MerchantOrderDB.order_id, "user_id": order.MerchantOrderDB.user_id, "merchant_product_id": order.MerchantOrderDB.merchant_product_id, "order_amount": order.MerchantOrderDB.order_amount, "status": order.MerchantOrderDB.status, "order_verify_code": order.MerchantOrderDB.order_verify_code, "verify_time": order.MerchantOrderDB.verify_time, "verify_user_id": order.MerchantOrderDB.verify_user_id, "create_time": order.MerchantOrderDB.create_time, "update_time": order.MerchantOrderDB.update_time, # 商品信息 "product_name": order.product_name, "product_image": order.product_image, # 商家信息 "merchant_name": order.merchant_name, "merchant_latitude": order.merchant_latitude, "merchant_longitude": order.merchant_longitude, "merchant_phone": order.merchant_phone } for order in orders] return success_response(data=order_list) @router.post("/{order_id}/verify", response_model=ResponseModel) async def verify_order( order_id: str, verify_code: str, db: Session = Depends(get_db), current_user: UserDB = Depends(get_current_user) ): """核销订单""" # 查询订单及相关信息 order = db.query( MerchantOrderDB, MerchantProductDB, MerchantDB ).join( MerchantProductDB, MerchantOrderDB.merchant_product_id == MerchantProductDB.id ).join( MerchantDB, MerchantProductDB.merchant_id == MerchantDB.id ).filter( MerchantOrderDB.order_id == order_id, MerchantOrderDB.order_verify_code == verify_code, MerchantOrderDB.verify_time.is_(None) # 未核销 ).first() if not order: return error_response(code=404, message="订单不存在或已核销") try: # 更新核销时间和核销用户 order.MerchantOrderDB.verify_time = datetime.now(timezone.utc) order.MerchantOrderDB.verify_user_id = current_user.userid order.MerchantOrderDB.status = MerchantOrderStatus.VERIFIED # 使用账户管理器处理余额变更 account_manager = AccountManager(db) settlement_amount = float(order.MerchantProductDB.settlement_amount) account_manager.change_balance( user_id=order.MerchantDB.user_id, amount=settlement_amount, description=f"{order.MerchantProductDB.name} 订单核销", transaction_id=order.MerchantOrderDB.order_id ) db.commit() return success_response( message="核销成功", data=MerchantOrderInfo.model_validate(order.MerchantOrderDB) ) except Exception as e: db.rollback() return error_response(code=500, message=f"核销失败: {str(e)}") @router.post("/{order_id}/unverify", response_model=ResponseModel) async def set_order_unverified( order_id: str, db: Session = Depends(get_db), current_user: UserDB = Depends(get_current_user) ): """设置订单为未核销状态""" order = db.query(MerchantOrderDB).filter( MerchantOrderDB.order_id == order_id, MerchantOrderDB.status == MerchantOrderStatus.CREATED # 只有已下单状态可以设为未核销 ).first() if not order: return error_response(code=404, message="订单不存在或状态不正确") # 更新状态为未核销 order.status = MerchantOrderStatus.UNVERIFIED try: db.commit() return success_response( message="状态更新成功", data=MerchantOrderInfo.model_validate(order) ) except Exception as e: db.rollback() return error_response(code=500, message=f"状态更新失败: {str(e)}") @router.post("/{order_id}/refund/apply", response_model=ResponseModel) async def apply_refund( order_id: str, db: Session = Depends(get_db), current_user: UserDB = Depends(get_current_user) ): """申请退款""" order = db.query(MerchantOrderDB).filter( MerchantOrderDB.order_id == order_id, MerchantOrderDB.user_id == current_user.userid, # 只能申请自己的订单 MerchantOrderDB.status.in_([MerchantOrderStatus.CREATED, MerchantOrderStatus.UNVERIFIED]) # 只有未核销的订单可以退款 ).first() if not order: return error_response(code=404, message="订单不存在或状态不允许退款") # 更新状态为退款中 order.status = MerchantOrderStatus.REFUNDING try: db.commit() return success_response( message="退款申请成功", data=MerchantOrderInfo.model_validate(order) ) except Exception as e: db.rollback() return error_response(code=500, message=f"申请退款失败: {str(e)}") @router.post("/{order_id}/refund/confirm", response_model=ResponseModel) async def confirm_refund( order_id: str, db: Session = Depends(get_db), current_user: UserDB = Depends(get_admin_user) ): """确认退款（管理员）""" # 查询订单和用户信息 order = db.query(MerchantOrderDB).join( UserDB, MerchantOrderDB.user_id == UserDB.userid ).filter( MerchantOrderDB.order_id == order_id, MerchantOrderDB.status == MerchantOrderStatus.REFUNDING # 只能确认退款中的订单 ).first() if not order: return error_response(code=404, message="订单不存在或状态不正确") try: # 更新状态为已退款 order.status = MerchantOrderStatus.REFUNDED # 如果有积分抵扣，返还积分 if order.deduct_amount > 0: # 返还用户积分 order.user.points += float(order.deduct_amount) point_record = PointRecordDB( user_id=order.user_id, points=float(order.deduct_amount), # 正数表示返还 description=f"订单退款返还" ) db.add(point_record) db.commit() return success_response( message="退款确认成功", data=MerchantOrderInfo.model_validate(order) ) except Exception as e: db.rollback() return error_response(code=500, message=f"确认退款失败: {str(e)}") @router.get("/{order_id}", response_model=ResponseModel) async def get_order_detail( order_id: str, db: Session = Depends(get_db), current_user: UserDB = Depends(get_current_user) ): """获取订单详情""" order = db.query( MerchantOrderDB, MerchantProductDB.name.label('product_name'), MerchantProductDB.image_url.label('product_image'), MerchantDB.name.label('merchant_name'), MerchantDB.latitude.label('merchant_latitude'), MerchantDB.longitude.label('merchant_longitude'), MerchantDB.phone.label('merchant_phone') ).join( MerchantProductDB, MerchantOrderDB.merchant_product_id == MerchantProductDB.id ).join( MerchantDB, MerchantProductDB.merchant_id == MerchantDB.id ).filter( MerchantOrderDB.order_id == order_id ).first() if not order: return error_response(code=404, message="订单不存在") # 检查权限 if order.MerchantOrderDB.user_id != current_user.userid: return error_response(code=403, message="无权查看此订单") # 构建返回数据 order_data = { "id": order.MerchantOrderDB.id, "order_id": order.MerchantOrderDB.order_id, "user_id": order.MerchantOrderDB.user_id, "merchant_product_id": order.MerchantOrderDB.merchant_product_id, "order_amount": order.MerchantOrderDB.order_amount, "status": order.MerchantOrderDB.status, "order_verify_code": order.MerchantOrderDB.order_verify_code, "verify_time": order.MerchantOrderDB.verify_time, "verify_user_id": order.MerchantOrderDB.verify_user_id, "create_time": order.MerchantOrderDB.create_time, "update_time": order.MerchantOrderDB.update_time, # 商品信息 "product_name": order.product_name, "product_image": order.product_image, # 商家信息 "merchant_name": order.merchant_name, "merchant_latitude": order.merchant_latitude, "merchant_longitude": order.merchant_longitude, "merchant_phone": order.merchant_phone } return success_response(data=order_data) @router.post("/calculate-price", response_model=ResponseModel) async def calculate_order_price( order: MerchantOrderCreate, db: Session = Depends(get_db), current_user: UserDB = Depends(get_current_user) ): """计算订单价格 1. 获取商品信息和用户积分 2. 计算最高可抵扣金额 3. 返回最终支付金额 """ # 查询商品信息 product = db.query(MerchantProductDB).filter( MerchantProductDB.id == order.merchant_product_id ).first() if not product: return error_response(code=404, message="商品不存在") # 获取用户积分 available_points = current_user.points or 0 # 计算最高可抵扣金额（1积分 = 1分钱） max_deduct_amount = min( float(product.max_deduct_points), # 商品最高可抵扣金额 available_points # 用户可用积分 ) if available_points > 0 else 0 # 计算最终支付金额 final_amount = max(0, float(product.sale_price) - max_deduct_amount) return success_response(data={ "original_price": float(product.sale_price), "available_points": available_points, "max_deduct_points": float(product.max_deduct_points), "max_deduct_amount": max_deduct_amount, "final_amount": final_amount })