如何與其他模型值過濾結果 - Django的restframework

Question

我有4個表：

Book: 
- id 
- name 
- user_id #(ForeignKey) 
- visitor_id #(ForeignKey) 

Car: 
- id 
- name 
- user_id #(ForeignKey) 
- visitor_id #(ForeignKey) 

Pen: 
- id 
- name 
- user_id #(ForeignKey) 
- visitor_id #(ForeignKey) 

Role: 
- id 
- obj_type #(ChoicesField. equals `1`,`2` or `3`) 
- obj_id 
- obj_role #(Boolean) 
- visitor_id #(ForeignKey) 

現在我有這樣的串行：

class BookSerializer(serializers.ModelSerializer): 
    queryset = Book.objects.filter(?!?!?) # Here I need help 
    serializer_class = BookSerializer 

我要檢查Role表，並根據過濾Book到Role的結果。我如何做到這一點？什麼是最佳解決方案？

我想是這樣的：

queryset = Book.objects.filter(
    ID__in(Role.objects.filter(user=request.user, obj_id=book_id, role__gt=0)) 
) 

Answer 1

您可以定義get_queryset方法：

class BookSerializer(serializers.ModelSerializer): 
    queryset = Book.objects.all() 

    def get_queryset(self): 
     # current user is self.request.user 
     return self.queryset.filter(....) 

例如，如果您只想讓書的主人看/編輯書

：

class BookSerializer(serializers.ModelSerializer): 
    queryset = Book.objects.all() 
    # ... 

    def get_queryset(self): 
     return self.queryset.filter(user_id=self.request.user.pk) 

更新：你需要這樣的：

items = Role.objects.filter(user=request.user, role__gt=0).values_list("pk", flat=True) 
return self.queryset.filter(pk__in=items)