我正在嘗試在 Django 中創(chuàng)建一個(gè)追隨者系統(tǒng)。follows = models.ManyToManyField('self', related_name='follower', symmetrical=False, null=True, blank=True)由于系統(tǒng)不應(yīng)該是對(duì)稱的，我如何檢查兩個(gè)用戶是否互相關(guān)注？如果可能的話，我想在模型中有一個(gè)函數(shù)。我可能會(huì)在limit_choices_to另一個(gè)領(lǐng)域和其他一些地方使用這個(gè)功能。我最初的想法是添加一個(gè)“朋友”字段并使其對(duì)稱，但這可能會(huì)在以后帶來其他問題。
1 回答
慕尼黑8549860
TA貢獻(xiàn)1818條經(jīng)驗(yàn) 獲得超11個(gè)贊
我沒有找到一個(gè)簡單的解決方案，不需要額外的模型。這是我所做的：
Person(models.Model)如果您有類似的CustomUser(AbstractUser)實(shí)現(xiàn)，則不需要?jiǎng)?chuàng)建。否則Person，類在許多情況下都很有用。
count_函數(shù)非常有用，但對(duì)于大多數(shù)用例來說不是必需的。
在里面accounts/models.py我補(bǔ)充說：
RELATIONSHIP_FOLLOWING = 1
RELATIONSHIP_BLOCKED = 2
RELATIONSHIP_STATUSES = (
(RELATIONSHIP_FOLLOWING, 'Following'),
(RELATIONSHIP_BLOCKED, 'Blocked'),
)
class Person(models.Model):
relationships = models.ManyToManyField('self', through='Relationship',
symmetrical=False,
related_name='related_to')
def get_relationships(self, status):
return self.relationships.filter(
to_people__status=status,
to_people__from_person=self)
def get_related_to(self, status):
return self.related_to.filter(
from_people__status=status,
from_people__to_person=self)
def get_following(self):
return self.get_relationships(RELATIONSHIP_FOLLOWING)
def get_followers(self):
return self.get_related_to(RELATIONSHIP_FOLLOWING)
def count_following(self):
return len(self.get_relationships(RELATIONSHIP_FOLLOWING))
def count_followers(self):
return len(self.get_related_to(RELATIONSHIP_FOLLOWING))
def get_friends(self):
return self.relationships.filter(
to_people__status=RELATIONSHIP_FOLLOWING,
to_people__from_person=self,
from_people__status=RELATIONSHIP_FOLLOWING,
from_people__to_person=self)
def count_friends(self):
return len(self.get_friends())
class Relationship(models.Model):
from_person = models.ForeignKey(Person, related_name='from_people', on_delete=models.CASCADE)
to_person = models.ForeignKey(Person, related_name='to_people', on_delete=models.CASCADE)
status = models.IntegerField(choices=RELATIONSHIP_STATUSES)
def add_relationship(self, person, status):
relationship, created = Relationship.objects.get_or_create(
from_person=self,
to_person=person,
status=status)
return relationship
def remove_relationship(self, person, status):
Relationship.objects.filter(
from_person=self,
to_person=person,
status=status).delete()
return
添加回答
舉報(bào)
0/150
提交
取消