diff --git a/nomadgram/users/models.py b/nomadgram/users/models.py index c79dc32..e61608d 100644 --- a/nomadgram/users/models.py +++ b/nomadgram/users/models.py @@ -24,8 +24,8 @@ class User(AbstractUser): bio = models.TextField(null=True) phone = models.CharField(max_length=140, null=True) gender = models.CharField(max_length=80, choices=GENDER_CHOICES, null=True) - followers = models.ManyToManyField("self", blank=True) - following = models.ManyToManyField("self", blank=True) + followers = models.ManyToManyField("self", blank=True, symmetrical=False, related_name='nomadgram_followers') + following = models.ManyToManyField("self", blank=True, symmetrical=False, related_name='nomadgram_following') push_token = models.TextField(default='') def __str__(self): diff --git a/nomadgram/users/views.py b/nomadgram/users/views.py index d4a4e67..0fb8d48 100644 --- a/nomadgram/users/views.py +++ b/nomadgram/users/views.py @@ -31,6 +31,7 @@ def post(self, request, user_id, format=None): return Response(status=status.HTTP_404_NOT_FOUND) user.following.add(user_to_follow) + user_to_follow.followers.add(user) user.save() @@ -51,6 +52,7 @@ def post(self, request, user_id, format=None): return Response(status=status.HTTP_404_NOT_FOUND) user.following.remove(user_to_follow) + user_to_follow.followers.remove(user) user.save()