Django REST Framework：使用组合过滤器从另一个查询集中排除一个查询集

Question

I have the following method which return objects for is_active=True and for particular user.

def filter_active_user(self, queryset, name, value):
    return queryset.filter(active_states__is_active=True,
                           active_states__user_id=value)

I would like to build a method which will return a queryset with all the objects which are not included in the above method. So I created the following method using exclude :

def filter_inactive_user(self, queryset, name, value):
    return queryset.exclude(active_states__is_active=True,
                            active_states__user_id=value)

I also tried the following implementation using ~Q :

def filter_inactive_user(self, queryset, name, value):
    return queryset.filter(~Q(active_states__is_active=True,
                              active_states__user_id=value))

But, I don't get the correct results. For example in a test case, I have created some data, consisted of 3 objects. Using the method filter_active_user I'm getting 1 object and I expect to get the other 2 using the filter_inactive_user method, but I'm getting also 1. To be more specific below presented my test data:

record_1.set_active_states_for_users([self.user1.uuid], True)
record_2.set_active_states_for_users([self.user1.uuid], False)
record_2.set_active_states_for_users([self.user2.uuid], True)
record_3.set_active_states_for_users([self.user2.uuid], False)

When I'm using the the filter_active_user for user=user2 I'm getting the record_2 . When I'm using the filter_inactive_user for user=user2 I'm getting only the record_1 and not the record_3 . Can you help me to build the properly way to combine two filters in a queryset?

Answer 1

Solution 1:

def filter_inactive_user(self, queryset, name, value):
    qs_active = queryset.filter(active_states__is_active=True, active_states__user_id=value)
    qs_inactive = queryset.exclude(pk__in=qs_active.values_list('pk', flat=True))

Solution 2:

def filter_inactive_user(self, queryset, name, value):
    qs_inactive = queryset.exclude(active_states__is_active=True).exclude(active_states__user_id=value)