ticket.py 1.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950
  1. from django.shortcuts import get_object_or_404
  2. from rest_framework import serializers
  3. from tickets.models import Ticket
  4. class BaseTicketSerializer(serializers.ModelSerializer):
  5. status = serializers.IntegerField(read_only=True)
  6. def validate(self, data):
  7. user = self.context['request'].user
  8. ticket_list = data.get("ticket_list")
  9. if ticket_list:
  10. if not (user.is_superuser or ticket_list.group in user.groups.all()):
  11. raise serializers.ValidationError("You don't have access to this list.")
  12. return data
  13. class TicketSerializer(BaseTicketSerializer):
  14. note = serializers.CharField(write_only=True, required=False)
  15. class Meta():
  16. model = Ticket
  17. exclude = ["updated_at"]
  18. class TicketDetailSerializer(BaseTicketSerializer):
  19. available_statuses = serializers.SerializerMethodField()
  20. def get_available_statuses(self, obj):
  21. return obj.get_available_statuses()
  22. class Meta():
  23. model = Ticket
  24. fields = "__all__"
  25. class TicketStatusSerializer(BaseTicketSerializer):
  26. status = serializers.IntegerField(read_only=False, required=True)
  27. available_statuses = serializers.SerializerMethodField()
  28. def get_available_statuses(self, obj):
  29. return obj.get_available_statuses()
  30. class Meta():
  31. model = Ticket
  32. fields = ["id", "available_statuses", "status"]