Django REST Framework : filtering with another table

Question

I'm using Django REST Framework to implement an associative schedule system, where a user can subscribe to an association and receive its events. Each association can have many groups (you choose yours when you subscribe to the association), so that the events are often the same for each group but with different places and dates.

The model looks like this :

class Association(models.Model):
    name = models.CharField(max_length=40, blank=False, unique=True)
    description = models.CharField(max_length=500, blank=False)

class AssocMember(models.Model):
    class Meta:
        unique_together = (("user", "assoc"),)

    user = models.ForeignKey('user.User')
    assoc = models.ForeignKey('associations.Association')
    is_admin = models.BooleanField(default=False)
    group = models.SmallIntegerField(default=-1)

class Event(models.Model):
    name = models.CharField(max_length=40, blank=False)
    date = models.DateTimeField(blank=False)
    assoc = models.ForeignKey('associations.Association')
    group = models.IntegerField(blank=True, default=-1)

I would need to implement a get_queryset() in my API that returns all the events of the connected user corresponding to its group. But for that, I have to access AssocMember for each event during the filtering, and create a custom condition (like normal .filter() in Python)

What is the best way to do so with Django REST filters ?

Thanks in advance.

YPCrumble · Accepted Answer · 2016-10-18 15:32:43Z

0

Check out a ModelViewSet. You can override the get_queryset method to filter based on request.user:

class EventViewSet(viewsets.ModelViewSet):
    def get_queryset(self):
        group = self.request.user.group  # Your models might need adjusting to do this
        return Event.objects.filter(group__in=group)

answered Oct 18, 2016 at 15:32

YPCrumble

29.1k26 gold badges115 silver badges177 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

Arthur Pesah Over a year ago

The problem is that the variable "group" depends on each event, because an event is related to an association, and the group of a user depends on this association). I found kind of a solution by turning the queryset into a list and using python filter function, but it seems ugly (I can't convert the list back to a queryset after)

Collectives™ on Stack Overflow

Django REST Framework : filtering with another table

1 Answer 1

1 Comment

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Linked

Related