feat: notifier beta

2026-04-13 12:30:33 +08:00
parent 646dbe7f18
commit 9512132bb9
50 changed files with 17830 additions and 2 deletions
--- a/api_v2/views/init.py
+++ b/api_v2/views/init.py
@@ -17,6 +17,18 @@ from .printing import (
    PrintingOrderBatchAdvanceRecordsView,
 )
 from .stateflow import BusinessObjectCloneView
+from .mission import (
+    ContentTypeListView,
+    MissionCancelView,
+    MissionCategoryDetailView,
+    MissionCategoryListCreateView,
+    MissionDetailView,
+    MissionListCreateView,
+    MissionReopenView,
+    MissionReplyListCreateView,
+    MissionReplyRejectView,
+    MissionSetUrgentView,
+)

 __all__ = [
    'HealthCheckView',
@@ -33,5 +45,14 @@ __all__ = [
    'PlateOrderBatchUpdateView',
    'PrintingOrderBatchAdvanceRecordsView',
    'BusinessObjectCloneView',
+    'MissionCancelView',
+    'MissionCategoryDetailView',
+    'MissionCategoryListCreateView',
+    'MissionDetailView',
+    'MissionListCreateView',
+    'MissionReopenView',
+    'MissionReplyListCreateView',
+    'MissionReplyRejectView',
+    'MissionSetUrgentView',
+    'ContentTypeListView',
 ]
-
--- a/api_v2/views/mission.py
+++ b/api_v2/views/mission.py
@@ -0,0 +1,513 @@
+from django.contrib.contenttypes.models import ContentType
+from django.db import IntegrityError
+from django.db.models import Q
+from django.db.models import ProtectedError
+from django.shortcuts import get_object_or_404
+from rest_framework import permissions, serializers, status
+from rest_framework.response import Response
+from rest_framework.views import APIView
+
+from mission import models as mission_models
+from mission import services as mission_services
+
+
+STATUS_FIELDS = {"is_urgent", "is_completed", "is_cancelled", "cancelled_by", "cancelled_at", "rejected_by", "rejected_at"}
+
+
+def _get_employee(request):
+    employee = getattr(request.user, "employee", None)
+    if employee is None:
+        raise serializers.ValidationError("当前用户未关联员工")
+    return employee
+
+
+def _employee_payload(employee):
+    if employee is None:
+        return None
+    return {
+        "id": employee.id,
+        "name": employee.name,
+        "merchant_id": employee.merchant_id,
+    }
+
+
+class MissionWriteSerializer(serializers.Serializer):
+    description = serializers.CharField(required=False, allow_blank=False)
+    category = serializers.IntegerField(required=False, min_value=1)
+    content_type = serializers.IntegerField(required=False, allow_null=True, min_value=1)
+    content_id = serializers.IntegerField(required=False, allow_null=True, min_value=1)
+    participant_ids = serializers.ListField(
+        child=serializers.IntegerField(min_value=1),
+        required=False,
+        allow_empty=True,
+    )
+
+    def __init__(self, *args, **kwargs):
+        self.is_create = kwargs.pop("is_create", False)
+        super().__init__(*args, **kwargs)
+        if self.is_create:
+            self.fields["description"].required = True
+
+    def validate_category(self, value):
+        employee = self.context.get("employee")
+        if employee is None:
+            return value
+        try:
+            return mission_models.MissionCategory.objects.get(id=value, merchant=employee.merchant)
+        except mission_models.MissionCategory.DoesNotExist as exc:
+            raise serializers.ValidationError("任务分类不存在") from exc
+
+    def validate(self, attrs):
+        forbidden_fields = STATUS_FIELDS.intersection(self.initial_data.keys())
+        if forbidden_fields:
+            raise serializers.ValidationError({field: "状态字段不能通过普通 CRUD 接口修改" for field in sorted(forbidden_fields)})
+
+        content_type_provided = "content_type" in self.initial_data
+        content_id_provided = "content_id" in self.initial_data
+        if content_type_provided != content_id_provided:
+            raise serializers.ValidationError("content_type 与 content_id 必须同时提供或同时省略")
+
+        content_type_id = attrs.get("content_type")
+        if content_type_id is not None:
+            try:
+                attrs["content_type"] = ContentType.objects.get(id=content_type_id)
+            except ContentType.DoesNotExist as exc:
+                raise serializers.ValidationError({"content_type": "ContentType 不存在"}) from exc
+        return attrs
+
+
+class MissionReplyCreateSerializer(serializers.Serializer):
+    content = serializers.CharField(allow_blank=False)
+    ends_task = serializers.BooleanField(required=False, default=False)
+
+
+class MissionUrgentSerializer(serializers.Serializer):
+    is_urgent = serializers.BooleanField()
+
+
+class MissionCategoryWriteSerializer(serializers.Serializer):
+    name = serializers.CharField(allow_blank=False, max_length=50)
+
+    def validate_name(self, value):
+        value = value.strip()
+        if not value:
+            raise serializers.ValidationError("分类名称不能为空")
+        employee = self.context["employee"]
+        instance = self.context.get("instance")
+        queryset = mission_models.MissionCategory.objects.filter(
+            merchant=employee.merchant,
+            name=value,
+        )
+        if instance is not None:
+            queryset = queryset.exclude(id=instance.id)
+        if queryset.exists():
+            raise serializers.ValidationError("分类名称已存在")
+        return value
+
+
+class MissionSerializer(serializers.ModelSerializer):
+    creator = serializers.SerializerMethodField()
+    cancelled_by = serializers.SerializerMethodField()
+    participants = serializers.SerializerMethodField()
+    content_type_label = serializers.SerializerMethodField()
+    category = serializers.IntegerField(source="category_id", read_only=True)
+    category_name = serializers.CharField(source="category.name", read_only=True)
+    has_ending_reply = serializers.BooleanField(read_only=True)
+    can_reply = serializers.BooleanField(read_only=True)
+
+    class Meta:
+        model = mission_models.Mission
+        fields = [
+            "id",
+            "merchant",
+            "description",
+            "category",
+            "category_name",
+            "is_urgent",
+            "is_completed",
+            "is_cancelled",
+            "cancelled_at",
+            "creator",
+            "cancelled_by",
+            "participants",
+            "content_type",
+            "content_type_label",
+            "content_id",
+            "has_ending_reply",
+            "can_reply",
+            "created_at",
+            "updated_at",
+        ]
+        read_only_fields = fields
+
+    def get_creator(self, obj):
+        return _employee_payload(obj.creator)
+
+    def get_cancelled_by(self, obj):
+        return _employee_payload(obj.cancelled_by)
+
+    def get_participants(self, obj):
+        return [
+            _employee_payload(participant.employee)
+            for participant in obj.participants.select_related("employee", "employee__merchant")
+        ]
+
+    def get_content_type_label(self, obj):
+        if obj.content_type_id is None:
+            return None
+        return f"{obj.content_type.app_label}.{obj.content_type.model}"
+
+
+class MissionCategorySerializer(serializers.ModelSerializer):
+    class Meta:
+        model = mission_models.MissionCategory
+        fields = [
+            "id",
+            "merchant",
+            "name",
+            "created_at",
+            "updated_at",
+        ]
+        read_only_fields = fields
+
+
+class MissionReplySerializer(serializers.ModelSerializer):
+    responder = serializers.SerializerMethodField()
+    rejected_by = serializers.SerializerMethodField()
+
+    class Meta:
+        model = mission_models.MissionReply
+        fields = [
+            "id",
+            "mission",
+            "merchant",
+            "responder",
+            "content",
+            "replied_at",
+            "ends_task",
+            "is_rejected",
+            "rejected_by",
+            "rejected_at",
+            "created_at",
+            "updated_at",
+        ]
+        read_only_fields = fields
+
+    def get_responder(self, obj):
+        return _employee_payload(obj.responder)
+
+    def get_rejected_by(self, obj):
+        return _employee_payload(obj.rejected_by)
+
+
+def _mission_queryset_for_employee(employee):
+    return (
+        mission_models.Mission.objects.filter(merchant=employee.merchant)
+        .select_related(
+            "merchant",
+            "category",
+            "creator",
+            "creator__merchant",
+            "cancelled_by",
+            "cancelled_by__merchant",
+            "content_type",
+        )
+        .prefetch_related("participants__employee", "participants__employee__merchant")
+        .order_by("-created_at", "-id")
+    )
+
+
+def _mission_category_queryset_for_employee(employee):
+    return mission_models.MissionCategory.objects.filter(merchant=employee.merchant).order_by("id")
+
+
+class MissionCategoryListCreateView(APIView):
+    permission_classes = [permissions.IsAuthenticated]
+
+    def get(self, request):
+        employee = _get_employee(request)
+        queryset = _mission_category_queryset_for_employee(employee)
+        return Response(MissionCategorySerializer(queryset, many=True).data)
+
+    def post(self, request):
+        employee = _get_employee(request)
+        serializer = MissionCategoryWriteSerializer(data=request.data, context={"employee": employee})
+        serializer.is_valid(raise_exception=True)
+        try:
+            category = mission_models.MissionCategory.objects.create(
+                merchant=employee.merchant,
+                name=serializer.validated_data["name"],
+            )
+        except IntegrityError:
+            return Response({"name": ["分类名称已存在"]}, status=status.HTTP_400_BAD_REQUEST)
+        return Response(MissionCategorySerializer(category).data, status=status.HTTP_201_CREATED)
+
+
+class MissionCategoryDetailView(APIView):
+    permission_classes = [permissions.IsAuthenticated]
+
+    def get_object(self, request, category_id):
+        employee = _get_employee(request)
+        return get_object_or_404(_mission_category_queryset_for_employee(employee), id=category_id)
+
+    def get(self, request, category_id):
+        category = self.get_object(request, category_id)
+        return Response(MissionCategorySerializer(category).data)
+
+    def patch(self, request, category_id):
+        employee = _get_employee(request)
+        category = self.get_object(request, category_id)
+        serializer = MissionCategoryWriteSerializer(
+            data=request.data,
+            partial=True,
+            context={"employee": employee, "instance": category},
+        )
+        serializer.is_valid(raise_exception=True)
+        if "name" in serializer.validated_data:
+            category.name = serializer.validated_data["name"]
+            try:
+                category.save(update_fields=["name", "updated_at"])
+            except IntegrityError:
+                return Response({"name": ["分类名称已存在"]}, status=status.HTTP_400_BAD_REQUEST)
+        return Response(MissionCategorySerializer(category).data)
+
+    def delete(self, request, category_id):
+        category = self.get_object(request, category_id)
+        try:
+            category.delete()
+        except ProtectedError:
+            return Response({"detail": "任务分类已被使用，不能删除"}, status=status.HTTP_400_BAD_REQUEST)
+        return Response(status=status.HTTP_204_NO_CONTENT)
+
+
+class MissionListCreateView(APIView):
+    permission_classes = [permissions.IsAuthenticated]
+
+    def get(self, request):
+        employee = _get_employee(request)
+        queryset = _mission_queryset_for_employee(employee)
+
+        if request.query_params.get("category"):
+            queryset = queryset.filter(category_id=request.query_params["category"])
+        for field in ["is_urgent", "is_completed", "is_cancelled"]:
+            value = request.query_params.get(field)
+            if value is not None:
+                queryset = queryset.filter(**{field: value.lower() in ["1", "true", "yes"]})
+        if request.query_params.get("content_type"):
+            queryset = queryset.filter(content_type_id=request.query_params["content_type"])
+        if request.query_params.get("content_id"):
+            queryset = queryset.filter(content_id=request.query_params["content_id"])
+
+        return Response(MissionSerializer(queryset, many=True).data)
+
+    def post(self, request):
+        employee = _get_employee(request)
+        serializer = MissionWriteSerializer(data=request.data, is_create=True, context={"employee": employee})
+        serializer.is_valid(raise_exception=True)
+        data = serializer.validated_data
+        try:
+            mission = mission_services.create_mission(
+                creator=employee,
+                description=data["description"],
+                category=data.get("category"),
+                content_type=data.get("content_type"),
+                content_id=data.get("content_id"),
+                participant_ids=data.get("participant_ids"),
+            )
+        except ValueError as exc:
+            return Response({"detail": str(exc)}, status=status.HTTP_400_BAD_REQUEST)
+        return Response(MissionSerializer(mission).data, status=status.HTTP_201_CREATED)
+
+
+class MissionDetailView(APIView):
+    permission_classes = [permissions.IsAuthenticated]
+
+    def get_object(self, request, mission_id):
+        employee = _get_employee(request)
+        return get_object_or_404(_mission_queryset_for_employee(employee), id=mission_id)
+
+    def get(self, request, mission_id):
+        mission = self.get_object(request, mission_id)
+        return Response(MissionSerializer(mission).data)
+
+    def patch(self, request, mission_id):
+        employee = _get_employee(request)
+        mission = self.get_object(request, mission_id)
+        serializer = MissionWriteSerializer(data=request.data, partial=True, context={"employee": employee})
+        serializer.is_valid(raise_exception=True)
+        data = serializer.validated_data
+        try:
+            mission = mission_services.update_mission(
+                mission=mission,
+                updated_by=employee,
+                description=data.get("description"),
+                category=data.get("category"),
+                content_type=data.get("content_type"),
+                content_id=data.get("content_id"),
+                update_content_object=("content_type" in request.data or "content_id" in request.data),
+                participant_ids=data.get("participant_ids"),
+            )
+        except ValueError as exc:
+            return Response({"detail": str(exc)}, status=status.HTTP_400_BAD_REQUEST)
+        return Response(MissionSerializer(mission).data)
+
+    def delete(self, request, mission_id):
+        return Response({"detail": "Mission 删除接口未提供，请使用 cancel 接口取消任务"}, status=status.HTTP_405_METHOD_NOT_ALLOWED)
+
+
+class MissionReplyListCreateView(APIView):
+    permission_classes = [permissions.IsAuthenticated]
+
+    def get(self, request, mission_id):
+        employee = _get_employee(request)
+        mission = get_object_or_404(_mission_queryset_for_employee(employee), id=mission_id)
+        queryset = (
+            mission_models.MissionReply.objects.filter(mission=mission)
+            .select_related("responder", "responder__merchant", "rejected_by", "rejected_by__merchant")
+            .order_by("replied_at", "id")
+        )
+        for field in ["ends_task", "is_rejected"]:
+            value = request.query_params.get(field)
+            if value is not None:
+                queryset = queryset.filter(**{field: value.lower() in ["1", "true", "yes"]})
+        return Response(MissionReplySerializer(queryset, many=True).data)
+
+    def post(self, request, mission_id):
+        employee = _get_employee(request)
+        mission = get_object_or_404(_mission_queryset_for_employee(employee), id=mission_id)
+        serializer = MissionReplyCreateSerializer(data=request.data)
+        serializer.is_valid(raise_exception=True)
+        try:
+            reply = mission_services.create_mission_reply(
+                mission=mission,
+                responder=employee,
+                content=serializer.validated_data["content"],
+                ends_task=serializer.validated_data.get("ends_task", False),
+            )
+        except ValueError as exc:
+            return Response({"detail": str(exc)}, status=status.HTTP_400_BAD_REQUEST)
+        return Response(MissionReplySerializer(reply).data, status=status.HTTP_201_CREATED)
+
+
+class MissionReopenView(APIView):
+    permission_classes = [permissions.IsAuthenticated]
+
+    def post(self, request, mission_id):
+        if not request.user.has_perm("mission.reopen_mission"):
+            return Response({"detail": "缺少重新打开任务权限"}, status=status.HTTP_403_FORBIDDEN)
+        employee = _get_employee(request)
+        mission = get_object_or_404(_mission_queryset_for_employee(employee), id=mission_id)
+        try:
+            mission = mission_services.reopen_mission(mission=mission, reopened_by=employee)
+        except ValueError as exc:
+            return Response({"detail": str(exc)}, status=status.HTTP_400_BAD_REQUEST)
+        return Response(MissionSerializer(mission).data)
+
+
+class MissionCancelView(APIView):
+    permission_classes = [permissions.IsAuthenticated]
+
+    def post(self, request, mission_id):
+        employee = _get_employee(request)
+        mission = get_object_or_404(_mission_queryset_for_employee(employee), id=mission_id)
+        try:
+            mission = mission_services.cancel_mission(mission=mission, cancelled_by=employee)
+        except ValueError as exc:
+            return Response({"detail": str(exc)}, status=status.HTTP_400_BAD_REQUEST)
+        return Response(MissionSerializer(mission).data)
+
+
+class MissionReplyRejectView(APIView):
+    permission_classes = [permissions.IsAuthenticated]
+
+    def post(self, request, reply_id):
+        if not request.user.has_perm("mission.reject_mission_reply"):
+            return Response({"detail": "缺少撤销任务回应权限"}, status=status.HTTP_403_FORBIDDEN)
+        employee = _get_employee(request)
+        reply = get_object_or_404(
+            mission_models.MissionReply.objects.select_related("mission", "responder", "rejected_by").filter(
+                merchant=employee.merchant,
+            ),
+            id=reply_id,
+        )
+        try:
+            reply = mission_services.reject_reply(reply=reply, rejected_by=employee)
+        except ValueError as exc:
+            return Response({"detail": str(exc)}, status=status.HTTP_400_BAD_REQUEST)
+        return Response(MissionReplySerializer(reply).data)
+
+
+class MissionSetUrgentView(APIView):
+    permission_classes = [permissions.IsAuthenticated]
+
+    def post(self, request, mission_id):
+        employee = _get_employee(request)
+        mission = get_object_or_404(_mission_queryset_for_employee(employee), id=mission_id)
+        serializer = MissionUrgentSerializer(data=request.data)
+        serializer.is_valid(raise_exception=True)
+        try:
+            mission = mission_services.set_mission_urgent(
+                mission=mission,
+                updated_by=employee,
+                is_urgent=serializer.validated_data["is_urgent"],
+            )
+        except ValueError as exc:
+            return Response({"detail": str(exc)}, status=status.HTTP_400_BAD_REQUEST)
+        return Response(MissionSerializer(mission).data)
+
+
+class ContentTypeListView(APIView):
+    permission_classes = [permissions.IsAuthenticated]
+
+    # Models the frontend is allowed to use as mission content_type targets.
+    _ALLOWED_CONTENT_TYPES = [
+        ("printing", "plateorder"),
+        ("printing", "printingorder"),
+        ("printing", "printingjob"),
+        ("business", "purchaseorder"),
+        ("business", "presalesorder"),
+        ("business", "salesorder"),
+        ("business", "prepurchaseorder"),
+        ("business", "purchasereturnorder"),
+        ("business", "salesreturnorder"),
+        ("business", "paymentorder"),
+        ("business", "receiptorder"),
+        ("stock", "transferorder"),
+        ("shipment", "shipment"),
+        ("stateflow", "process"),
+    ]
+
+    # Human-readable labels keyed by (app_label, model).
+    _LABELS = {
+        ("printing", "plateorder"): "开版单",
+        ("printing", "printingorder"): "印刷单",
+        ("printing", "printingjob"): "印刷任务",
+        ("business", "purchaseorder"): "采购单",
+        ("business", "presalesorder"): "预销售单",
+        ("business", "salesorder"): "销售单",
+        ("business", "prepurchaseorder"): "预采购单",
+        ("business", "purchasereturnorder"): "采购退货单",
+        ("business", "salesreturnorder"): "销售退货单",
+        ("business", "paymentorder"): "付款单",
+        ("business", "receiptorder"): "收款单",
+        ("stock", "transferorder"): "调拨单",
+        ("shipment", "shipment"): "发货单",
+        ("stateflow", "process"): "工艺流程实例",
+    }
+
+    def get(self, request):
+        filters = Q()
+        for app_label, model in self._ALLOWED_CONTENT_TYPES:
+            filters |= Q(app_label=app_label, model=model)
+        cts = ContentType.objects.filter(filters).order_by("app_label", "model")
+        data = [
+            {
+                "id": ct.id,
+                "app_label": ct.app_label,
+                "model": ct.model,
+                "label": f"{ct.app_label}.{ct.model}",
+                "name": self._LABELS.get((ct.app_label, ct.model), ct.model),
+            }
+            for ct in cts
+        ]
+        return Response(data)