clean: project is healthy now :)

profiles/models/BlockModel.py

@@ -0,0 +1,7 @@
+from django.db import models
+from django.contrib.auth.models import User
+
+
+class BlockModel(models.Model):
+    blocker = models.ForeignKey(User, on_delete=models.CASCADE, related_name='blocker')
+    blocked = models.ForeignKey(User, on_delete=models.CASCADE, related_name='blocked')

profiles/models/FriendModel.py

@@ -0,0 +1,7 @@
+from django.contrib.auth.models import User
+from django.db import models
+
+
+class FriendModel(models.Model):
+    friend1 = models.ForeignKey(User, on_delete=models.CASCADE, related_name='friend1')
+    friend2 = models.ForeignKey(User, on_delete=models.CASCADE, related_name='friend2')

profiles/models/ProfileModel.py

@@ -0,0 +1,53 @@
+from os.path import splitext
+
+from django.contrib.auth.models import User
+from django.db import models
+from django.db.models.signals import post_save, pre_delete
+from django.dispatch import receiver
+
+from games.consumers import game_manager
+from ..models.FriendModel import FriendModel
+
+
+def upload_to(instance, filename: str):
+    return f"./profiles/static/avatars/{instance.pk}{splitext(filename)[1]}"
+
+
+class ProfileModel(models.Model):
+    user = models.ForeignKey(User, on_delete=models.CASCADE)
+    avatar = models.ImageField(upload_to=upload_to, default="./profiles/static/avatars/default.avif")
+
+    def get_game(self) -> int:
+        for game in game_manager._game_list:
+            for player in game.get_players_connected():
+                if (player.user_id == self.user.pk):
+                    return game.game_id
+        return None
+
+    def getFriends(self):
+        friends = []
+
+        for friendship in FriendModel.objects.filter(models.Q(friend1=self.user) | models.Q(friend2=self.user)):
+            friends.append(friendship.friend1 if friendship.friend1 != self.user else friendship.friend2)
+
+        return friends
+
+    def isFriend(self, user):
+        return user in self.getFriends()
+
+    def deleteFriend(self, user):
+        for friendship in FriendModel.objects.filter(models.Q(friend1=self.user) | models.Q(friend2=self.user)):
+            friendship.delete()
+
+
+@receiver(pre_delete, sender=ProfileModel)
+def delete_profile_picture(sender, instance, **kwargs):
+    if instance.avatar.name != './profiles/static/avatars/default.avif':
+        instance.avatar.storage.delete(instance.avatar.name)
+
+
+@receiver(post_save, sender=User)
+def on_user_created(sender, instance, created, **kwargs):
+    if created:
+        profile: ProfileModel = ProfileModel.objects.create(pk=instance.pk, user=instance)
+        profile.save()