From 56953183b8d6592e7af8ef9e434b5704acab079e Mon Sep 17 00:00:00 2001 From: mattdamon108 Date: Tue, 30 Oct 2018 17:02:40 +0900 Subject: [PATCH] set symmentrical=False in User model, add/remove codes for followers in views.py --- nomadgram/users/models.py | 4 ++-- nomadgram/users/views.py | 2 ++ 2 files changed, 4 insertions(+), 2 deletions(-) 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()