Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Django check if object in ManyToMany field

I have quite a simple problem to solve. I have Partner model which has >= 0 Users associated with it:

class Partner(models.Model):     name = models.CharField(db_index=True, max_length=255)     slug = models.SlugField(db_index=True)     user = models.ManyToManyField(User) 

Now, if I have a User object and I have a Partner object, what is the most Pythonic way of checking if the User is associated with a Partner? I basically want a statement which returns True if the User is associated to the Partner.

I have tried:

users = Partner.objects.values_list('user', flat=True).filter(slug=requested_slug) if request.user.pk in users:     # do some private stuff 

This works but I have a feeling there is a better way. Additionally, would this be easy to roll into a decorator, baring in mind I need both a named parameter (slug) and a request object (user).

like image 893
Darwin Tech Avatar asked May 23 '13 19:05

Darwin Tech


2 Answers

if user.partner_set.filter(slug=requested_slug).exists():      # do some private stuff 
like image 59
Peter DeGlopper Avatar answered Oct 01 '22 09:10

Peter DeGlopper


If we just need to know whether a user object is associated to a partner object, we could just do the following (as in this answer):

if user in partner.user.all():     #do something 
like image 39
Anupam Avatar answered Oct 01 '22 10:10

Anupam