forked from erp-dev/erp
514 lines
20 KiB
Python
514 lines
20 KiB
Python
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)
|