Уточнён контракт Feed и обработка новостей программ

Toksi86 · Toksi86 · commit 69e4e8001f20 · 2026-05-26T12:53:40.000+05:00
diff --git a/docs/modules/feed.md b/docs/modules/feed.md
@@ -26,10 +26,9 @@ handlers для служебных записей.
 
 ## Архитектура
 
-- `feed/views.py` - endpoint `/feed/`, фильтрация по типам и финальная сборка
-  response payload.
-- `feed/serializers.py` - `FeedNewsResponseSerializer`, который превращает
-  `news.News` в элемент ленты.
+- `feed/views.py` - endpoint `/feed/` и фильтрация по типам.
+- `feed/serializers.py` - `FeedItemResponseSerializer`, который превращает
+  `news.News` в элемент ленты формата `{type_model, content}`.
 - `feed/services.py` - helpers для лайков и служебных feed-записей.
 - `feed/mapping.py` - соответствие content object типам и serializers.
 - `feed/constants.py` - типы моделей, для которых signals создают feed-записи.
@@ -45,14 +44,28 @@ Frontend вызывает `/feed/?type=...`.
 View выбирает подходящие `news.News`, сериализует их и возвращает элементы
 ленты с полями `type_model` и `content`.
 
+Ответ использует стандартную пагинацию DRF: `count`, `next`, `previous`,
+`results`. Каждый элемент `results` содержит только:
+
+- `type_model` - тип элемента ленты;
+- `content` - сериализованный объект, соответствующий этому типу.
+
 ### 2. В ленту попадает обычная новость
 
-Если `news.News` содержит текст и относится к пользователю или проекту, лента
-возвращает ее как новость.
+Если `news.News` содержит текст и относится к пользователю, проекту или
+партнерской программе, лента возвращает ее как новость.
 
 Проектная новость с текстом возвращается как `type_model = "news"`, даже если
 ее `content_object` - проект.
 
+Новость партнерской программы с текстом тоже возвращается как
+`type_model = "news"`. Отдельный `type_model = "partner_program"` пока не
+вводится.
+
+Служебные feed-записи партнерских программ сейчас не создаются. Если такой
+сценарий понадобится, для него нужно отдельно согласовать `type_model` и
+frontend-контракт.
+
 ### 3. В ленту попадает служебная запись
 
 Служебные feed-записи создаются через `feed.services.create_news_for_model()`.
@@ -78,14 +91,17 @@ View выбирает подходящие `news.News`, сериализует 
 - `GET /feed/?type=news` - новости пользователей.
 - `GET /feed/?type=project` - проектные новости и проектные feed-записи.
 - `GET /feed/?type=vacancy` - служебные feed-записи вакансий.
-- `GET /feed/?type=project|news|vacancy` - комбинированная выдача по нескольким
-  типам.
+- `GET /feed/?type=partnerprogram` - новости партнерских программ.
+- `GET /feed/?type=project|vacancy|news|partnerprogram` - комбинированная
+  выдача по нескольким типам.
 
 ## Ограничения и правила
 
 - Feed читает данные из `news.News`, но не отвечает за создание обычных
   project/user/program news.
 - Служебная feed-запись определяется через пустой `text`.
+- Новости партнерских программ с текстом отображаются как обычные новости;
+  отдельные служебные карточки программ в ленте пока не поддерживаются.
 - Signals `feed` создают или удаляют служебные feed-записи для проектов и
   вакансий. Более широкие сценарии публикации проекта остаются в модуле
   `projects`.
@@ -96,6 +112,8 @@ View выбирает подходящие `news.News`, сериализует 
 
 - `/feed/?type=news` возвращает пользовательские новости;
 - `/feed/?type=project` возвращает проектные новости в frontend-формате;
+- `/feed/?type=partnerprogram` возвращает новости программ как
+  `type_model = "news"`;
 - `/feed/?type=project` возвращает служебную feed-запись проекта как
   `type_model = "project"`;
 - `/feed/?type=vacancy` возвращает служебную feed-запись вакансии как
diff --git a/feed/mapping.py b/feed/mapping.py
@@ -6,10 +6,9 @@
 from vacancy.models import Vacancy
 from vacancy.serializers import VacancyDetailSerializer
 
-CONTENT_OBJECT_MAPPING: dict[str, str | None] = {
+CONTENT_OBJECT_MAPPING: dict[str, str] = {
     Project.__name__.lower(): "project",
     CustomUser.__name__.lower(): "news",
-    "partnerprogram": None,
     Vacancy.__name__.lower(): "vacancy",
 }
 
diff --git a/feed/serializers.py b/feed/serializers.py
@@ -6,11 +6,12 @@
 from news.mapping import NewsMapping
 from news.models import News
 from news.services import is_content_news
+from partner_programs.models import PartnerProgram
 from projects.models import Project
 from users.models import CustomUser
 
 
-class FeedNewsResponseSerializer(serializers.ModelSerializer):
+class FeedNewsContentSerializer(serializers.ModelSerializer):
     name = serializers.SerializerMethodField()
     image_address = serializers.SerializerMethodField()
     is_user_liked = serializers.SerializerMethodField()
@@ -20,11 +21,18 @@ class FeedNewsResponseSerializer(serializers.ModelSerializer):
     content_object = serializers.SerializerMethodField()
     type_model = serializers.SerializerMethodField()
 
-    def get_type_model(self, obj) -> str:
-        model_type = CONTENT_OBJECT_MAPPING[obj.content_type.model]
-        if is_content_news(obj) and model_type == "project":
+    def get_type_model(self, obj) -> str | None:
+        content_model = obj.content_type.model
+
+        if content_model == PartnerProgram.__name__.lower():
+            # Новости программ сейчас отображаются как обычные новости.
+            # Отдельная служебная карточка программы в ленте пока не согласована.
+            return "news" if is_content_news(obj) else None
+
+        if is_content_news(obj) and content_model == Project.__name__.lower():
             return "news"
-        return model_type
+
+        return CONTENT_OBJECT_MAPPING[content_model]
 
     def get_content_object(self, obj) -> dict:
         type_model = obj.content_type.model
@@ -65,3 +73,20 @@ class Meta:
             "type_model",
         ]
         read_only_fields = ["views_count", "likes_count", "type_model"]
+
+
+class FeedItemResponseSerializer(serializers.Serializer):
+    def to_representation(self, instance):
+        data = FeedNewsContentSerializer(instance, context=self.context).data
+        type_model = data["type_model"]
+
+        if type_model == "news":
+            content = dict(data)
+            del content["type_model"]
+        else:
+            content = data["content_object"]
+
+        return {
+            "type_model": type_model,
+            "content": content,
+        }
diff --git a/feed/tests/test_feed_api.py b/feed/tests/test_feed_api.py
@@ -4,7 +4,12 @@
 from core.services import set_like
 from feed.services import create_news_for_model
 from feed.tests.helpers import create_vacancy
-from news.tests.helpers import create_news_for, create_project, create_user
+from news.tests.helpers import (
+    create_news_for,
+    create_partner_program,
+    create_project,
+    create_user,
+)
 
 
 class FeedAPITests(TestCase):
@@ -20,6 +25,7 @@ def test_feed_returns_user_news_when_news_filter_requested(self):
 
         self.assertEqual(response.status_code, 200)
         item = response.data["results"][0]
+        self.assertEqual(set(item.keys()), {"type_model", "content"})
         self.assertEqual(item["type_model"], "news")
         self.assertEqual(item["content"]["id"], news.id)
         self.assertEqual(item["content"]["text"], "User feed news")
@@ -32,10 +38,24 @@ def test_feed_returns_project_news_as_news_content(self):
 
         self.assertEqual(response.status_code, 200)
         item = response.data["results"][0]
+        self.assertEqual(set(item.keys()), {"type_model", "content"})
         self.assertEqual(item["type_model"], "news")
         self.assertEqual(item["content"]["id"], news.id)
         self.assertEqual(item["content"]["text"], "Project feed news")
 
+    def test_feed_returns_program_news_as_news_content(self):
+        program = create_partner_program(name="Feed program")
+        news = create_news_for(program, text="Program feed news")
+
+        response = self.client.get("/feed/?type=partnerprogram")
+
+        self.assertEqual(response.status_code, 200)
+        item = response.data["results"][0]
+        self.assertEqual(set(item.keys()), {"type_model", "content"})
+        self.assertEqual(item["type_model"], "news")
+        self.assertEqual(item["content"]["id"], news.id)
+        self.assertEqual(item["content"]["text"], "Program feed news")
+
     def test_feed_returns_project_feed_record_as_project_content(self):
         project = create_project(name="Feed record project")
         create_news_for_model(project)
@@ -44,6 +64,7 @@ def test_feed_returns_project_feed_record_as_project_content(self):
 
         self.assertEqual(response.status_code, 200)
         item = response.data["results"][0]
+        self.assertEqual(set(item.keys()), {"type_model", "content"})
         self.assertEqual(item["type_model"], "project")
         self.assertEqual(item["content"]["id"], project.id)
 
@@ -54,10 +75,59 @@ def test_feed_returns_vacancy_feed_record_as_vacancy_content(self):
 
         self.assertEqual(response.status_code, 200)
         item = response.data["results"][0]
+        self.assertEqual(set(item.keys()), {"type_model", "content"})
         self.assertEqual(item["type_model"], "vacancy")
         self.assertEqual(item["content"]["id"], vacancy.id)
         self.assertEqual(item["content"]["role"], "Backend developer")
 
+    def test_feed_combines_all_supported_filters(self):
+        project_news = create_news_for(
+            create_project(name="Combined project news"),
+            text="Combined project news",
+        )
+        program_news = create_news_for(
+            create_partner_program(name="Combined program"),
+            text="Combined program news",
+        )
+        user_news = create_news_for(self.user, text="Combined user news")
+        project = create_project(name="Combined project record")
+        vacancy = create_vacancy(role="Combined vacancy")
+        create_news_for_model(project)
+
+        response = self.client.get(
+            "/feed/?type=project|vacancy|news|partnerprogram"
+        )
+
+        self.assertEqual(response.status_code, 200)
+        items_by_text = {
+            item["content"].get("text"): item
+            for item in response.data["results"]
+            if item["type_model"] == "news"
+        }
+        content_ids_by_type = {
+            type_model: {
+                item["content"]["id"]
+                for item in response.data["results"]
+                if item["type_model"] == type_model
+            }
+            for type_model in ["project", "vacancy"]
+        }
+
+        self.assertEqual(
+            items_by_text[project_news.text]["content"]["id"],
+            project_news.id,
+        )
+        self.assertEqual(
+            items_by_text[program_news.text]["content"]["id"],
+            program_news.id,
+        )
+        self.assertEqual(
+            items_by_text[user_news.text]["content"]["id"],
+            user_news.id,
+        )
+        self.assertIn(project.id, content_ids_by_type["project"])
+        self.assertIn(vacancy.id, content_ids_by_type["vacancy"])
+
     def test_feed_excludes_feed_record_for_inactive_vacancy(self):
         vacancy = create_vacancy(role="Inactive vacancy", is_active=False)
         create_news_for_model(vacancy)
diff --git a/feed/views.py b/feed/views.py
@@ -7,11 +7,11 @@
 from projects.models import Project
 from vacancy.models import Vacancy
 
-from .serializers import FeedNewsResponseSerializer
+from .serializers import FeedItemResponseSerializer
 
 
 class NewSimpleFeed(APIView):
-    serializator_class = FeedNewsResponseSerializer
+    serializer_class = FeedItemResponseSerializer
     pagination_class = FeedPagination
 
     def _get_filter_data(self) -> list[str]:
@@ -54,26 +54,12 @@ def get_queryset(self) -> QuerySet[News]:
     def get(self, *args, **kwargs):
         paginator = self.pagination_class()
         paginated_data = paginator.paginate_queryset(self.get_queryset(), self.request)
-        serializer = FeedNewsResponseSerializer(
+        serializer = FeedItemResponseSerializer(
             paginated_data,
             context={
                 "user": self.request.user,
                 "liked_news": get_liked_news(self.request.user, paginated_data),
             },
             many=True,
         )
-
-        new_data = []
-        # временная подстройка данных под фронт
-        for data in serializer.data:
-            if data["type_model"] in ["project", "vacancy", None]:
-                formatted_data = {
-                    "type_model": data["type_model"],
-                    "content": data["content_object"],
-                }
-            elif data["type_model"] == "news":
-                del data["type_model"]
-                formatted_data = {"type_model": "news", "content": data}
-            new_data.append(formatted_data)
-
-        return paginator.get_paginated_response(new_data)
+        return paginator.get_paginated_response(serializer.data)
