Merge pull request #274 from PROCOLLAB-github/dev

Dev

Author: yakser

core/migrations/0011_auto_20240128_2214.py

@@ -0,0 +1,27 @@
+# Generated by Django 4.2.3 on 2024-01-28 19:33
+
+from django.db import migrations
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("core", "0011_auto_20240128_2214"),
+    ]
+
+    operations = [
+        migrations.AlterModelOptions(
+            name="specialization",
+            options={
+                "verbose_name": "Специализация",
+                "verbose_name_plural": "Специализации",
+            },
+        ),
+        migrations.AlterModelOptions(
+            name="specializationcategory",
+            options={
+                "verbose_name": "Категория специализации",
+                "verbose_name_plural": "Категории специализаций",
+            },
+        ),
+    ]

core/migrations/0012_alter_specialization_options_and_more.py

@@ -155,13 +155,21 @@ class SkillToObject(models.Model):
 class SpecializationCategory(models.Model):
     name = models.TextField()
 
+    class Meta:
+        verbose_name = "Категория специализации"
+        verbose_name_plural = "Категории специализаций"
+
 
 class Specialization(models.Model):
     name = models.TextField()
     category = models.ForeignKey(
         SpecializationCategory, related_name="specializations", on_delete=models.CASCADE
     )
 
+    class Meta:
+        verbose_name = "Специализация"
+        verbose_name_plural = "Специализации"
+
 
 class SpecializationToObject(models.Model):
     specialization = models.ForeignKey(

core/models.py

@@ -0,0 +1,6 @@
+from django.apps import AppConfig
+
+
+class FeedConfig(AppConfig):
+    default_auto_field = "django.db.models.BigAutoField"
+    name = "feed"

feed/__init__.py

@@ -0,0 +1,20 @@
+import enum
+
+from rest_framework import serializers
+
+from news.serializers import NewsListSerializer
+from projects.serializers import ProjectListSerializer
+from vacancy.serializers import VacancyDetailSerializer
+
+
+class FeedItemType(enum.Enum):
+    PROJECT = "project"
+    NEWS = "news"
+    VACANCY = "vacancy"
+
+
+FEED_SERIALIZER_MAPPING: dict[FeedItemType, serializers.Serializer] = {
+    FeedItemType.PROJECT.value: ProjectListSerializer,
+    FeedItemType.NEWS.value: NewsListSerializer,
+    FeedItemType.VACANCY.value: VacancyDetailSerializer,
+}

feed/apps.py

@@ -0,0 +1,69 @@
+import random
+import typing
+
+from feed import constants
+from feed.serializers import FeedItemSerializer
+from news.models import News
+from projects.models import Project
+from vacancy.models import Vacancy
+
+
+def collect_feed() -> list:
+    # да, это ужасно
+    n_random_projects = get_n_random_projects(3)
+    n_latest_created_projects = get_n_latest_created_projects(3)
+    n_latest_created_news = get_n_latest_created_news(3)
+    n_latest_created_vacancies = get_n_latest_created_vacancies(3)
+
+    feed = (
+        to_feed_items(
+            constants.FeedItemType.PROJECT.value,
+            set(n_random_projects + n_latest_created_projects),
+        )
+        + to_feed_items(constants.FeedItemType.NEWS.value, n_latest_created_news)
+        + to_feed_items(constants.FeedItemType.VACANCY.value, n_latest_created_vacancies)
+    )
+
+    random.shuffle(feed)
+    return feed
+
+
+def to_feed_items(type_: constants.FeedItemType, items: typing.Iterable) -> list[dict]:
+    feed_items = []
+    for item in items:
+        serializer = to_feed_item(type_, item)
+        serializer.is_valid()
+        feed_items += [serializer.data]
+    return feed_items
+
+
+def get_n_random_projects(num: int) -> list[Project]:
+    tries = 3
+    projects = set()
+
+    while len(projects) < num and tries > 0:
+        project = Project.objects.filter(draft=False).order_by("?").first()
+
+        if project not in projects:
+            projects.add(project)
+        else:
+            tries -= 1
+    return list(projects)
+
+
+def get_n_latest_created_projects(num: int) -> list[Project]:
+    return list(Project.objects.filter(draft=False).order_by("-datetime_created")[:num])
+
+
+def get_n_latest_created_news(num: int) -> list[Project]:
+    return list(News.objects.order_by("-datetime_created")[:num])
+
+
+def get_n_latest_created_vacancies(num: int) -> list[Project]:
+    return list(Vacancy.objects.order_by("-datetime_created")[:num])
+
+
+def to_feed_item(type_: constants.FeedItemType, data):
+    serializer = constants.FEED_SERIALIZER_MAPPING[type_](data)
+
+    return FeedItemSerializer(data={"type": type_, "content": serializer.data})

feed/constants.py

@@ -0,0 +1,7 @@
+from rest_framework import serializers
+from feed import constants
+
+
+class FeedItemSerializer(serializers.Serializer):
+    type = serializers.ChoiceField(choices=constants.FeedItemType, required=True)
+    content = serializers.JSONField(required=True)

feed/helpers.py

@@ -0,0 +1,9 @@
+from django.urls import path
+
+from feed.views import FeedList
+
+app_name = "feed"
+
+urlpatterns = [
+    path("", FeedList.as_view()),
+]