Merge pull request #265 from PROCOLLAB-github/yakser-pro-110

Лента с рандомными новостями

Author: yakser

feed/constants.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/helpers.py

@@ -1,4 +1,40 @@
+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]:
@@ -17,3 +53,17 @@ def get_n_random_projects(num: int) -> list[Project]:
 
 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/serializers.py

@@ -1,6 +1,7 @@
 from rest_framework import serializers
+from feed import constants
 
 
-class FeedListSerializer(serializers.Serializer):
-    # todo: add type for feed news item
-    pass
+class FeedItemSerializer(serializers.Serializer):
+    type = serializers.ChoiceField(choices=constants.FeedItemType, required=True)
+    content = serializers.JSONField(required=True)

feed/views.py

@@ -5,8 +5,7 @@
 from rest_framework.response import Response
 from rest_framework.views import APIView
 
-from feed.helpers import get_n_random_projects, get_n_latest_created_projects
-from projects.serializers import ProjectListSerializer
+from feed.helpers import collect_feed
 
 
 class FeedList(APIView):
@@ -19,21 +18,14 @@ class FeedList(APIView):
                     items=openapi.Schema(
                         type=openapi.TYPE_OBJECT,
                         description="Feed item",
+                        properties={
+                            "type": openapi.TYPE_STRING,
+                            "content": openapi.TYPE_OBJECT,
+                        },
                     ),
                 ),
             )
         }
     )
     def get(self, request: Request, *args, **kwargs) -> Response:
         return Response(status=status.HTTP_200_OK, data=collect_feed())
-
-
-def collect_feed() -> list:
-    n_random_projects = get_n_random_projects(3)
-    n_latest_created_projects = get_n_latest_created_projects(3)
-    serializer = ProjectListSerializer(
-        data=set(n_random_projects + n_latest_created_projects), many=True
-    )
-
-    serializer.is_valid()
-    return serializer.data