fix: полная переработка циклов через python-docx API вместо XML

2026-07-30 17:26:52 +04:00
commit 3b9816c788
15 changed files with 3270 additions and 0 deletions
@@ -0,0 +1,148 @@
+# -*- coding: utf-8 -*-
+"""DokoGen — Модели данных (только 152-ФЗ, только нужные поля)"""
+
+from dataclasses import dataclass, field, asdict
+from typing import List, Dict, Any
+
+
+@dataclass
+class CommissionMember:
+    """Член комиссии по ПДн"""
+    role: str = "член комиссии"
+    position: str = ""
+    fio: str = ""
+
+
+@dataclass
+class InformationSystem:
+    """Информационная система (обработка ПДн)"""
+    name: str = ""
+    description: str = ""
+    is_local_network: bool = False
+    is_internet: bool = False
+    defence_level: str = ""
+    threat_type: str = "3"
+    category: str = ""
+    pd_list: str = ""
+    pd_count: str = "менее 100 000"
+    users: str = ""
+    defense_tools_list: List[str] = field(default_factory=list)
+    processing_modes: str = ""
+    subject_type: str = "работники организации"
+    pd_actions_list: List[str] = field(default_factory=list)
+    pd_subjects_list: List[str] = field(default_factory=list)
+    personal_data_list: List[str] = field(default_factory=list)
+    users_list: List[str] = field(default_factory=list)
+    personal_data_category: List[str] = field(default_factory=list)
+    purpose: str = ""
+    room: str = ""
+
+
+@dataclass
+class PaperDocument:
+    """Бумажный документ с ПДн"""
+    name: str = ""
+    storage: str = ""
+
+
+@dataclass
+class Company:
+    """Организация — основной объект (152-ФЗ)"""
+    # Полное название
+    full_name: str = ""
+    company_name_1: str = ""  # именительный
+    company_name_2: str = ""  # родительный
+    company_name_3: str = ""  # дательный
+    company_name_4: str = ""  # винительный
+    company_name_5: str = ""  # творительный
+    company_name_6: str = ""  # предложный
+
+    # Краткое название
+    short_name: str = ""
+    short_name_1: str = ""
+    short_name_2: str = ""
+    short_name_3: str = ""
+    short_name_4: str = ""
+    short_name_5: str = ""
+    short_name_6: str = ""
+
+    # Адрес и реквизиты
+    address: str = ""
+    city_name: str = ""
+    inn: str = ""
+    ogrn: str = ""
+    ogrn_date: str = ""
+    kpp: str = ""
+
+    # Должностные лица
+    chief_position: str = ""
+    chief_fio: str = ""
+    responsible_position: str = ""
+    responsible_fio: str = ""
+    admin_position: str = ""
+    admin_fio: str = ""
+
+    # Документ
+    contract_number: str = ""
+    contract_date: str = ""
+
+    # Списки
+    commission: List[CommissionMember] = field(default_factory=list)
+    information_systems: List[InformationSystem] = field(default_factory=list)
+    paper_documents_list: List[PaperDocument] = field(default_factory=list)
+    employees_access: List[Dict[str, str]] = field(default_factory=list)
+
+    def update_declensions(self, inflect_func):
+        """Обновить склонения полного названия"""
+        if self.full_name:
+            inflected = inflect_func(self.full_name)
+            self.company_name_1 = self.full_name
+            self.company_name_2 = inflected.get("gent", "")
+            self.company_name_3 = inflected.get("datv", "")
+            self.company_name_4 = inflected.get("accs", "")
+            self.company_name_5 = inflected.get("ablt", "")
+            self.company_name_6 = inflected.get("loct", "")
+
+    def update_short_declensions(self, inflect_func):
+        """Обновить склонения сокращённого названия"""
+        if self.short_name:
+            inflected = inflect_func(self.short_name)
+            self.short_name_1 = self.short_name
+            self.short_name_2 = inflected.get("gent", "")
+            self.short_name_3 = inflected.get("datv", "")
+            self.short_name_4 = inflected.get("accs", "")
+            self.short_name_5 = inflected.get("ablt", "")
+            self.short_name_6 = inflected.get("loct", "")
+
+    def to_dict(self) -> Dict[str, Any]:
+        return asdict(self)
+
+    @classmethod
+    def from_dict(cls, data: Dict[str, Any]) -> "Company":
+        d = dict(data)
+        if "commission" in d and isinstance(d["commission"], list):
+            d["commission"] = [
+                CommissionMember(**c) if isinstance(c, dict) else c
+                for c in d["commission"]
+            ]
+        if "information_systems" in d and isinstance(d["information_systems"], list):
+            d["information_systems"] = [
+                InformationSystem(**isys) if isinstance(isys, dict) else isys
+                for isys in d["information_systems"]
+            ]
+        if "paper_documents_list" in d and isinstance(d["paper_documents_list"], list):
+            d["paper_documents_list"] = [
+                PaperDocument(**p) if isinstance(p, dict) else p
+                for p in d["paper_documents_list"]
+            ]
+        if "employees_access" in d and not isinstance(d["employees_access"], list):
+            d["employees_access"] = []
+        valid = {f.name for f in cls.__dataclass_fields__.values()}
+        return cls(**{k: v for k, v in d.items() if k in valid})
+
+    def get_folder_name(self) -> str:
+        if self.short_name:
+            return self.short_name
+        if self.full_name:
+            return self.full_name[:30]
+        return "Новая_организация"