python - 如果用户 "owns"是模型实例，如何进行身份验证

Question

我有以下模型:

class UserProfile(models.Model):
    user = models.OneToOneField(User)

class Site(models.Model):
    user = models.ForeignKey(User)
    site_name = models.CharField(max_length=128, blank=False, null=False)

class Team(models.Model):
    site = models.ForeignKey(Site)
    team_member_name = models.CharField(default='name', max_length=128, blank=False, null=False)

我有一个通过 URL 传递 team_member id 的 View :

网址.py:

url(r'^team/(?P<team_member_id>\d+)/$', 'team.views.home', name='team_view_team_member')

Views.py:

@login_required
def home(request, team_member_id=None):
    team_member = Team.objects.get(id=team_member_id)

请注意，有许多 Team 实例(即很多团队成员)具有相同的 Site_id。

我如何测试 request.user 是否具有与返回的任何 team_member 相同的 site_id？如果我想在多个 View 中使用它，是否有一种简单的方法可以重复此操作？

Answer 1

试试这个:

team_member = Team.objects.get(id=team_member_id)
if team_member.site.id == request.user.site_set.all().first().id:
   print "same site"
else:
   print "different site"