User initial filtering

This commit is contained in:
Fergal Moran
2016-06-30 21:54:08 +01:00
parent 970c251af1
commit ad2a3163ae
3 changed files with 8 additions and 5 deletions

View File

@@ -72,6 +72,7 @@ class UserProfileViewSet(viewsets.ModelViewSet):
lookup_field = 'slug'
filter_fields = (
'slug',
'user__first_name',
)
def get_queryset(self):
@@ -81,6 +82,9 @@ class UserProfileViewSet(viewsets.ModelViewSet):
ret = UserProfile.objects.filter(followers__slug__in=[self.request.query_params['followers']])
elif 'messaged_with' in self.request.query_params:
ret = UserProfile.objects.filter(messages__slug__in=[self.request.query_params['followers']])
elif 'initial' in self.request.query_params:
ret = UserProfile.objects.filter(user__first_name__startswith=self.request.query_params['initial']) \
.annotate(mix_count=Count('mixes')).order_by('-mix_count')
else:
ret = super(UserProfileViewSet, self).get_queryset()