Ghi ngoại lệ KhôngNotExist trong trình quản lý tùy chỉnh ở Django

Question

Tôi có trình quản lý tùy chỉnh cho mô hình Django. Tôi không có vẻ để có thể bắt ngoại lệ DoesNotExist ở đây. Tôi biết cách làm điều đó bên trong mô hình nhưng nó không hoạt động ở đây:

class TaskManager(models.Manager): 
    def task_depend_tree(self, *args, **kwargs): 
     if "id" in kwargs: 
      try: 
       task = self.get(id=kwargs["id"]) 
      except DoesNotExist: 
       raise Http404 

Get_object_or_404 cũng không hoạt động. Có gì sai ở đây?

Answer 1

Hãy thử sử dụng ObjectDoesNotExist thay vì DoesNotExist hoặc có thể self.DoesNotExist. Nếu mọi thứ khác thất bại, chỉ cần thử và bắt một vanilla Exception và đánh giá nó để xem loại đó().

from django.core.exceptions import ObjectDoesNotExist

Answer 2

bạn có thể sử dụng DoesNotExist từ dụ Manager.model (self.model), khi bạn nói objects = MyManager() bạn đang gán self.model bên trong lớp MyManager.

 try: 
      task = self.get(id=kwargs["id"]) 
      return task 
     except self.DoesNotExist: 
      return None 

Answer 3

Như đã đề nghị panchicore, self.model là cách để thực hiện.

class TaskManager(models.Manager): 
    def task_depend_tree(self, *args, **kwargs): 
     if "id" in kwargs: 
      try: 
       task = self.get(id=kwargs["id"]) 
      except self.model.DoesNotExist: 
       raise Http404 

Answer 4

Nếu bạn cần phải thực hiện điều này trên một phương pháp danh sách (DRF) sử dụng GenericViewSet, và cần một danh sách trống để được trả lại, sử dụng này:

def list(self, request, *args, **kwargs): 
    self.get_object() # I use this to trigger the object_permission 
    try: 
     queryset = self.queryset.filter(user=(YourModel.objects.get(user=request.user).user)) 
    except YourModel.DoesNotExist: 
     return Response(YourModel.objects.none()) 

    serializer = YSourModelSerializer(queryset, many=True) 
    return Response(serializer.data)