Skip to content

Commit 3d3e4b6

Browse files
authored
Update models.py to Flake8 conventions
1 parent 0cfe5fc commit 3d3e4b6

1 file changed

Lines changed: 8 additions & 8 deletions

File tree

server/friend/models.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
from django.db import models
22
from django.conf import settings
3-
from django.utils import timezone
3+
# from django.utils import timezone
44
from django.db.models.signals import post_save
55
from django.dispatch import receiver
66
from 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+
5858
class 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

Comments
 (0)