initial

web/models/__init__.py

@@ -0,0 +1,3 @@
+from .custom_user import CustomUser
+from .project import Project
+from .membership import Membership

web/models/custom_user.py

@@ -0,0 +1,19 @@
+from django.contrib.auth.models import AbstractUser
+from django.db import models
+
+
+class CustomUser(AbstractUser):
+    vk_id = models.IntegerField(null=True, blank=True)
+    yandex_id = models.BigIntegerField(null=True, blank=True)
+    only_one_project = models.BooleanField(default=True)
+    selected_project = models.ForeignKey('Project', on_delete=models.CASCADE, null=True)
+
+    @property
+    def profile_photo_url(self):
+        return f"/profile_photo?id={self.id}"
+
+    @property
+    def projects(self):
+        from web.models import Membership
+        for membership in Membership.objects.filter(user=self):
+            yield membership.project

web/models/membership.py

@@ -0,0 +1,9 @@
+from django.db import models
+
+from web.models.custom_user import CustomUser
+from web.models.project import Project
+
+
+class Membership(models.Model):
+    user = models.ForeignKey(CustomUser, on_delete=models.CASCADE)
+    project = models.ForeignKey(Project, on_delete=models.CASCADE)

web/models/project.py

@@ -0,0 +1,5 @@
+from django.db import models
+
+
+class Project(models.Model):
+    name = models.TextField()