Добавлена ручка для выгрузки данных проектов в программах

(cherry picked from commit 2b83e93)

Author: Toksi86

partner_programs/serializers.py

@@ -80,9 +80,7 @@ class PartnerProgramForMemberSerializer(PartnerProgramBaseSerializerMixin):
 
     views_count = serializers.SerializerMethodField(method_name="count_views")
     links = serializers.SerializerMethodField(method_name="get_links")
-    is_user_manager = serializers.SerializerMethodField(
-        method_name="get_is_user_manager"
-    )
+    is_user_manager = serializers.SerializerMethodField(method_name="get_is_user_manager")
 
     def count_views(self, program):
         return get_views_count(program)

partner_programs/services.py

@@ -1,4 +1,5 @@
 import logging
+from collections import OrderedDict
 
 from partner_programs.models import PartnerProgramUserProfile
 from project_rates.models import Criteria, ProjectScore
@@ -120,3 +121,88 @@ def get_project_scores_info(self) -> dict[str, str]:
                     partner_program__id=self._program_id
                 )
             }
+
+
+BASE_COLUMNS = [
+    ("row_number", "№ п/п"),
+    ("project_name", "Название проекта"),
+    ("project_description", "Описание проекта"),
+    ("project_region", "Регион проекта"),
+    ("project_presentation", "Ссылка на презентацию"),
+    ("team_size", "Количество человек в команде"),
+    ("leader_full_name", "Имя фамилия лидера"),
+]
+
+
+def _leader_full_name(user):
+    if not user:
+        return ""
+    if hasattr(user, "get_full_name") and callable(user.get_full_name):
+        full = user.get_full_name()
+        if full:
+            return full
+    first = getattr(user, "first_name", "") or ""
+    last = getattr(user, "last_name", "") or ""
+    return (first + " " + last).strip() or getattr(user, "username", "") or str(user.pk)
+
+
+def _calc_team_size(project):
+    try:
+        if hasattr(project, "get_collaborators_user_list"):
+            return 1 + len(project.get_collaborators_user_list())
+        if hasattr(project, "collaborator_set"):
+            return 1 + project.collaborator_set.count()
+    except Exception:
+        pass
+    return 1
+
+
+def build_program_field_columns(program) -> list[tuple[str, str]]:
+    program_fields = program.fields.all().order_by("pk")
+    return [
+        (f"name:{program_field.name}", program_field.label)
+        for program_field in program_fields
+    ]
+
+
+def row_dict_for_link(
+    program_project_link,
+    extra_field_keys_order: list[str],
+    row_number: int,
+) -> OrderedDict:
+    """
+    program_project_link: PartnerProgramProject
+    extra_field_keys_order: список псевдоключей "name:<field.name>" в нужном порядке
+    row_number: порядковый номер строки в Excel (начиная с 1)
+    """
+    project = program_project_link.project
+    row = OrderedDict()
+
+    row["row_number"] = row_number
+
+    row["project_name"] = project.name or ""
+    row["project_description"] = project.description or ""
+    row["project_region"] = project.region or ""
+    row["project_presentation"] = project.presentation_address or ""
+    row["team_size"] = _calc_team_size(project)
+    row["leader_full_name"] = _leader_full_name(getattr(project, "leader", None))
+
+    values_map: dict[str, str] = {}
+    prefetched_values = getattr(program_project_link, "_prefetched_field_values", None)
+    field_values_iterable = (
+        prefetched_values
+        if prefetched_values is not None
+        else program_project_link.field_values.all()
+    )
+
+    for field_value in field_values_iterable:
+        if (
+            field_value.field.partner_program_id
+            == program_project_link.partner_program_id
+        ):
+            values_map[f"name:{field_value.field.name}"] = field_value.get_value()
+
+    for field_key in extra_field_keys_order:
+        row[field_key] = values_map.get(field_key, "")
+
+    return row

partner_programs/urls.py

@@ -5,6 +5,7 @@
     PartnerProgramCreateUserAndRegister,
     PartnerProgramDataSchema,
     PartnerProgramDetail,
+    PartnerProgramExportProjectsAPIView,
     PartnerProgramList,
     PartnerProgramProjectsAPIView,
     PartnerProgramProjectSubmitView,
@@ -50,4 +51,9 @@
         PartnerProgramProjectsAPIView.as_view(),
         name="partner-program-projects",
     ),
+    path(
+        "<int:pk>/export-projects/",
+        PartnerProgramExportProjectsAPIView.as_view(),
+        name="partner-program-export-projects",
+    ),
 ]

partner_programs/views.py

@@ -1,10 +1,17 @@
+import io
+import unicodedata
+from datetime import date
+
 from django.contrib.auth import get_user_model
 from django.db import IntegrityError, transaction
+from django.db.models import Prefetch
+from django.http import FileResponse
 from django.shortcuts import get_object_or_404
 from django.utils import timezone
 from django.utils.timezone import now
 from drf_yasg import openapi
 from drf_yasg.utils import swagger_auto_schema
+from openpyxl import Workbook
 from rest_framework import generics, permissions, status
 from rest_framework.exceptions import NotFound, PermissionDenied, ValidationError
 from rest_framework.generics import GenericAPIView
@@ -34,16 +41,18 @@
     PartnerProgramUserSerializer,
     ProgramProjectFilterRequestSerializer,
 )
+from partner_programs.services import (
+    BASE_COLUMNS,
+    build_program_field_columns,
+    row_dict_for_link,
+)
 from partner_programs.utils import filter_program_projects_by_field_name
 from projects.models import Project
 from projects.serializers import (
     PartnerProgramFieldValueUpdateSerializer,
     ProjectListSerializer,
 )
-from vacancy.mapping import (
-    MessageTypeEnum,
-    UserProgramRegisterParams,
-)
+from vacancy.mapping import MessageTypeEnum, UserProgramRegisterParams
 from vacancy.tasks import send_email
 
 User = get_user_model()
@@ -255,9 +264,7 @@ def get_project(self, project_id):
         except Project.DoesNotExist:
             raise NotFound("Проект не найден")
 
-    @swagger_auto_schema(
-        request_body=PartnerProgramFieldValueUpdateSerializer(many=True)
-    )
+    @swagger_auto_schema(request_body=PartnerProgramFieldValueUpdateSerializer(many=True))
     def put(self, request, project_id, *args, **kwargs):
         project = self.get_project(project_id)
 
@@ -439,3 +446,92 @@ class PartnerProgramProjectsAPIView(generics.ListAPIView):
     def get_queryset(self):
         program = get_object_or_404(PartnerProgram, pk=self.kwargs["pk"])
         return Project.objects.filter(program_links__partner_program=program).distinct()
+
+
+def _slugify_filename(filename: str) -> str:
+    """
+    Преобразует произвольную строку в безопасное имя файла:
+    - нормализует Unicode;
+    - оставляет только буквы, цифры, дефисы, подчёркивания и пробелы;
+    - заменяет группы пробелов на один дефис.
+    """
+    normalized_name = unicodedata.normalize("NFKD", filename)
+    safe_chars = [
+        char for char in normalized_name if char.isalnum() or char in ("-", "_", " ")
+    ]
+    cleaned_name = "".join(safe_chars)
+    return "-".join(cleaned_name.split())
+
+
+class PartnerProgramExportProjectsAPIView(APIView):
+    """Возвращает Excel-файл со всеми проектами программы."""
+
+    permission_classes = [permissions.IsAdminUser]
+
+    def get(self, request, pk: int):
+        try:
+            program = PartnerProgram.objects.get(pk=pk)
+        except PartnerProgram.DoesNotExist:
+            return Response(
+                {"detail": "Программа не найдена."}, status=status.HTTP_404_NOT_FOUND
+            )
+
+        user = request.user
+        if not (
+            getattr(user, "is_staff", False)
+            or getattr(user, "is_superuser", False)
+            or program.is_manager(user)
+        ):
+            return Response(
+                {"detail": "Недостаточно прав."}, status=status.HTTP_403_FORBIDDEN
+            )
+
+        only_submitted = request.query_params.get("only_submitted") in (
+            "1",
+            "true",
+            "True",
+        )
+
+        extra_cols = build_program_field_columns(program)
+        header_pairs = BASE_COLUMNS + extra_cols
+
+        fv_qs = PartnerProgramFieldValue.objects.select_related("field").filter(
+            field__partner_program_id=program.id
+        )
+        links_qs = program.program_projects.select_related(
+            "project", "project__leader"
+        ).prefetch_related(
+            Prefetch("field_values", queryset=fv_qs, to_attr="_prefetched_field_values")
+        )
+        if only_submitted:
+            links_qs = links_qs.filter(submitted=True)
+
+        wb = Workbook(write_only=True)
+        ws = wb.create_sheet(title="Проекты")
+        ws.append([title for _, title in header_pairs])
+
+        extra_keys_order = [key for key, _ in extra_cols]
+
+        for row_number, program_project_link in enumerate(links_qs, start=1):
+            row_dict = row_dict_for_link(
+                program_project_link=program_project_link,
+                extra_field_keys_order=extra_keys_order,
+                row_number=row_number,
+            )
+            ws.append([row_dict.get(key, "") for key, _ in header_pairs])
+
+        bio = io.BytesIO()
+        wb.save(bio)
+        bio.seek(0)
+
+        fname_base = _slugify_filename(
+            f"{program.name or 'program'}-{program.pk}-projects-{date.today():%Y-%m-%d}"
+        )
+        filename = f"{fname_base}.xlsx"
+
+        return FileResponse(
+            bio,
+            as_attachment=True,
+            filename=filename,
+            content_type="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
+        )