Add initial implementations for forms and organization apps with serializers, factories, and admin configurations

src/apps/form_6/services.py

@@ -0,0 +1,112 @@
+"""
+Сервисы для работы с формой Ф-6.
+
+Содержит:
+- FormF6Service - CRUD операции
+- FormF6Parser - парсинг Excel
+"""
+
+import logging
+from typing import Any
+
+from apps.core.excel import (
+    BaseExcelParser,
+    ColumnMapping,
+    ParseResult,
+    RowData,
+)
+from apps.core.services import BaseService, BulkOperationsMixin
+from apps.form_6.models import FormF6Record
+from apps.organization.services import OrganizationService
+from django.db import transaction
+from django.db.models import Count, Max
+
+logger = logging.getLogger(__name__)
+
+
+class FormF6Service(BulkOperationsMixin, BaseService[FormF6Record]):
+    """Сервис для работы с записями формы Ф-6."""
+
+    model = FormF6Record
+
+    @classmethod
+    def get_by_batch(cls, batch_id: int):
+        return cls.get_queryset().filter(load_batch=batch_id)
+
+    @classmethod
+    def get_next_batch_id(cls) -> int:
+        max_batch = cls.model.objects.aggregate(max_batch=Max("load_batch"))
+        return (max_batch["max_batch"] or 0) + 1
+
+    @classmethod
+    def get_batches(cls) -> list[dict[str, Any]]:
+        return list(
+            cls.model.objects.values("load_batch")
+            .annotate(count=Count("id"), created_at=Max("created_at"))
+            .order_by("-load_batch")
+        )
+
+
+class FormF6Parser(BaseExcelParser[FormF6Record]):
+    """Парсер Excel файла формы Ф-6 (Возрастная структура оборудования)."""
+
+    ORG_NAME_COLUMN = 0
+    OKPO_COLUMN = 1
+    OGRN_COLUMN = 2
+    INN_COLUMN = 3
+
+    def get_column_mappings(self) -> list[ColumnMapping]:
+        return [
+            # Категоризация
+            ColumnMapping(4, "Код строки", "row_code", field_type="str"),
+            ColumnMapping(5, "Категория оборудования", "category", field_type="str"),
+            # Общие данные
+            ColumnMapping(6, "Всего оборудования", "total_equipment", field_type="int"),
+            ColumnMapping(7, "Отечественное оборудование", "domestic_equipment", field_type="int"),
+            ColumnMapping(8, "Импортное оборудование", "imported_equipment", field_type="int"),
+            # Возрастная структура
+            ColumnMapping(9, "До 5 лет", "age_under_5", field_type="int"),
+            ColumnMapping(10, "5-10 лет", "age_5_10", field_type="int"),
+            ColumnMapping(11, "10-15 лет", "age_10_15", field_type="int"),
+            ColumnMapping(12, "15-20 лет", "age_15_20", field_type="int"),
+            ColumnMapping(13, "Свыше 20 лет", "age_over_20", field_type="int"),
+            # С ЧПУ
+            ColumnMapping(14, "С ЧПУ всего", "cnc_total", field_type="int"),
+            ColumnMapping(15, "С ЧПУ до 5 лет", "cnc_under_5", field_type="int"),
+            ColumnMapping(16, "С ЧПУ 5-10 лет", "cnc_5_10", field_type="int"),
+            ColumnMapping(17, "С ЧПУ 10-15 лет", "cnc_10_15", field_type="int"),
+            ColumnMapping(18, "С ЧПУ 15-20 лет", "cnc_15_20", field_type="int"),
+            ColumnMapping(19, "С ЧПУ свыше 20 лет", "cnc_over_20", field_type="int"),
+            # Показатели
+            ColumnMapping(20, "Средняя сменность работы", "avg_shift_work", field_type="decimal"),
+            ColumnMapping(21, "Коэффициент загрузки", "utilization_rate", field_type="decimal"),
+            ColumnMapping(22, "Физический износ, %", "physical_wear_percent", field_type="decimal"),
+            # Потребности
+            ColumnMapping(23, "Рабочие места без оборудования", "workplaces_without_equipment", field_type="int"),
+            ColumnMapping(24, "Оборудование к замене", "equipment_to_replace", field_type="int"),
+        ]
+
+    def get_next_batch_id(self) -> int:
+        return FormF6Service.get_next_batch_id()
+
+    @transaction.atomic
+    def create_record(self, row_data: RowData, batch_id: int) -> FormF6Record:
+        org, _ = OrganizationService.get_or_create_by_inn(
+            inn=row_data.inn,
+            defaults={
+                "name": row_data.organization_name,
+                "ogrn": row_data.ogrn or "",
+                "okpo": row_data.okpo or "",
+                "kpp": row_data.kpp or "",
+            },
+        )
+        return FormF6Record.objects.create(
+            organization=org,
+            load_batch=batch_id,
+            **row_data.fields,
+        )
+
+
+def parse_form_f6_file(file) -> ParseResult:
+    parser = FormF6Parser()
+    return parser.parse(file)