11from django .db import models
22from django .conf import settings
3- from django .utils import timezone
3+ # from django.utils import timezone
44from django .db .models .signals import post_save
55from django .dispatch import receiver
66from django .contrib .auth import get_user_model
@@ -13,18 +13,17 @@ class FriendList(models.Model):
1313
1414 user = models .OneToOneField (settings .AUTH_USER_MODEL , on_delete = models .CASCADE , related_name = "friend_list" )
1515 friends = models .ManyToManyField (settings .AUTH_USER_MODEL , blank = True , related_name = "friends_of" )
16-
17-
1816 def __str__ (self ):
1917 return self .user .username
18+
2019 def add_friend (self , account ):
2120 """
2221 Add a new friend
2322 """
24- if not account in self .friends .all ():
23+ if account not in self .friends .all ():
2524 self .friends .add (account )
2625 self .save ()
27-
26+
2827 def remove_friend (self , account ):
2928 """
3029 Remove a friend
@@ -37,7 +36,7 @@ def unfriend(self, removee):
3736 """
3837 Initiate action of unfriending someone
3938 """
40- remover_friends_list = self # person terminating friendship
39+ remover_friends_list = self # person terminating friendship
4140
4241 # Remove friend from remover friend list
4342 remover_friends_list .remove_friend (removee )
@@ -55,9 +54,10 @@ def is_mutual_friend(self, friend):
5554 return True
5655 return False
5756
57+
5858class FriendRequest (models .Model ):
5959 """
60- Friend consists of two parts,
60+ Friend consists of two parts,
6161 1. Sender -- Person sending friend request
6262 2. Reciever -- Person recieving friend request
6363 """
@@ -69,7 +69,7 @@ class FriendRequest(models.Model):
6969 timestamp = models .DateTimeField (auto_now_add = True )
7070 def __str__ (self ):
7171 return self .sender .username
72-
72+
7373 def accept (self ):
7474 """
7575 Accept a friend request
0 commit comments