70

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).

4

2 回答 2

101
if user.partner_set.filter(slug=requested_slug).exists():
     # do some private stuff
于 2013-05-23T20:12:39.933 回答
3

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
于 2021-11-06T08:32:49.167 回答