"""Settlement API views""" import logging import re from datetime import datetime, date from django.contrib.auth import get_user_model from django.utils import timezone from rest_framework import status, authentication from rest_framework.views import APIView from rest_framework.response import Response from rest_framework.permissions import IsAuthenticated from rest_framework_simplejwt.authentication import JWTAuthentication from basic_info.models import Merchant from settlement.services import get_plate_order_summary_by_customer from .mixins import SettlementVisibilityMixin logger = logging.getLogger(__name__) AGENT_SECRET = "RCYH_BOT_0083" class AgentSecretAuthentication(authentication.BaseAuthentication): """ Agent 密钥认证(仅用于 settlement/plate-orders/summary API) 当请求头中存在 X-AGENT-SECRET 且值匹配时,跳过 JWT 认证, 以 admin 超级管理员身份执行请求。 """ def authenticate(self, request): secret = request.META.get("HTTP_X_AGENT_SECRET") if secret != AGENT_SECRET: return None User = get_user_model() try: user = User.objects.get(username="admin", is_superuser=True) return (user, None) except User.DoesNotExist: logger.warning("[AgentSecretAuthentication] admin 用户不存在") return None class PlateOrderSummaryView(SettlementVisibilityMixin, APIView): """ 开版订单统计 API GET /api/v1/settlement/plate-orders/summary/ 参数: date: 统计日期(YYYY-MM-DD) 认证方式: - JWT Token(标准方式) - X-AGENT-SECRET: RCYH_BOT_0083(Agent 专用,跳过鉴权) 可见性规则(与 plate-order 列表接口对齐): - superuser 或持有 view_all_permission(默认 printing.view_all_plateorders) → 返回该商户下全部客户的开版汇总 - 普通员工 → 仅返回其负责/可见客户的开版汇总 """ authentication_classes = [ AgentSecretAuthentication, authentication.SessionAuthentication, JWTAuthentication, ] permission_classes = [IsAuthenticated] view_all_permission = "printing.view_all_plateorders" def get(self, request): """ 获取开版订单统计 """ date_str = request.query_params.get("date") if not date_str: return Response( {"error": "缺少 date 参数"}, status=status.HTTP_400_BAD_REQUEST ) if not re.match(r"^\d{4}-\d{2}-\d{2}$", date_str): return Response( {"error": "日期格式错误,请使用 YYYY-MM-DD 格式"}, status=status.HTTP_400_BAD_REQUEST, ) try: settlement_date = date.fromisoformat(date_str) except ValueError: return Response({"error": "日期不存在"}, status=status.HTTP_403_FORBIDDEN) emp = getattr(request.user, "employee", None) if emp is None or emp.merchant is None: if request.user.is_superuser: merchant = Merchant.objects.first() if merchant is None: return Response( {"error": "系统中没有商户"}, status=status.HTTP_403_FORBIDDEN ) merchant_id = merchant.id else: return Response( {"error": "用户未关联商户"}, status=status.HTTP_403_FORBIDDEN ) else: merchant_id = emp.merchant.id try: data = get_plate_order_summary_by_customer( merchant_id=merchant_id, settlement_date=settlement_date, user=self.get_service_user(request.user), ) return Response({"data": data}) except Exception as e: logger.exception(f"[settlement.views] 获取开版订单统计失败: {e}") return Response( {"error": "获取统计数据失败"}, status=status.HTTP_500_INTERNAL_SERVER_ERROR, )