refactor: separate models to separate files

Author: danihodovic

django_wtf/core/models.py

@@ -0,0 +1,8 @@
+from .category_model import Category
+from .contributor_model import Contributor
+from .email_subscriber_model import EmailSubscriber
+from .profile_model import Profile, ProfileFollowers, UserType
+from .pypi_project import PypiProject
+from .pypi_release import PypiRelease
+from .repository_model import Repository, RepositoryStars, RepositoryType
+from .social_media_news_model import SocialNews, SocialNewsType

django_wtf/core/models/__init__.py

@@ -0,0 +1,14 @@
+from django.db import models
+from django.urls import reverse
+from model_utils.models import TimeStampedModel
+
+
+class Category(TimeStampedModel):
+    name = models.CharField(max_length=50, unique=True)
+    emoji = models.CharField(max_length=5)
+
+    def __str__(self):
+        return f"<Category: {self.name}>"
+
+    def get_absolute_url(self):
+        return reverse("core:category-detail", args=[self.name])

django_wtf/core/models/category_model.py

@@ -0,0 +1,19 @@
+from django.db import models
+from model_utils.models import TimeStampedModel
+
+
+class Contributor(TimeStampedModel):
+    profile = models.ForeignKey(
+        "core.Profile",
+        on_delete=models.CASCADE,
+    )
+    repository = models.ForeignKey(
+        "core.Repository",
+        on_delete=models.CASCADE,
+        related_name="contributors",
+        related_query_name="contributor",
+    )
+    contributions = models.PositiveIntegerField()
+
+    def __str__(self):
+        return f"<Contributor {self.profile.login} to {self.repository.full_name}>"

django_wtf/core/models/contributor_model.py

@@ -0,0 +1,11 @@
+from django.contrib.auth import get_user_model
+from django.db import models
+from model_utils.models import TimeStampedModel
+
+User = get_user_model()
+
+
+class EmailSubscriber(TimeStampedModel):
+    user = models.OneToOneField(
+        User, on_delete=models.CASCADE, related_name="subscriber"
+    )