start commit

2025-11-13 23:24:20 +03:00
parent 0d3d8d9326
commit a0cb66f80e
2 changed files with 188 additions and 1 deletions
--- a/poterya_kodcsv_scheta.py
+++ b/poterya_kodcsv_scheta.py
@@ -0,0 +1,157 @@
+from datetime import datetime
+from xml.dom import minidom
+from dataclasses import dataclass
+import os
+import csv
+import logging
+from typing import List
+
+logger = logging.getLogger('application')
+logger.setLevel(logging.INFO)
+fh = logging.FileHandler('app.log', encoding='utf-8')
+logger.addHandler(fh)
+
+
+class DNaPlanGenerator:
+
+    @dataclass
+    class DataNode:
+        fam: str = ''
+        im: str = ''
+        ot: str = ''
+        dr: str = ''
+        w: str = ''
+        npolis: str = ''
+        mes: str = ''
+        nzap: str = ''
+        mcode: str = ''
+
+        def to_dict(self) -> dict:
+            return {
+                'N_ZAP': self.nzap,
+                'FAM': self.fam,
+                'IM': self.im,
+                'OT': self.ot,
+                'W': self.w,
+                'DR': self.dr,
+                'NPOLIS': self.npolis,
+                'MDP': '0',
+                'SMOCOD': '35003',
+                'DISP_TYP': '6',
+                'MES': self.mes,
+            }
+
+    def __init__(self, mo_code: str, year: int, month: int):
+        self.mo_code = mo_code
+        self.year = 2026
+        self.month = month
+        self.doc = minidom.Document()
+        self.root = self.doc.createElement('ZL_LIST')
+        self.xml_name = f'D-M{self.mo_code}-F35-2026-{month}'
+        self.root.appendChild(self.header())
+        self.doc.appendChild(self.root)
+
+    def header(self):
+        zglv_tag = self.doc.createElement('ZGLV')
+
+        fname_tag = self.doc.createElement('FILENAME')
+        fname_tag.appendChild(self.doc.createTextNode(self.xml_name))
+
+        data_tag = self.doc.createElement('DATA')
+        data_tag.appendChild(self.doc.createTextNode(
+            datetime.now().strftime('%Y-%m-%d')))
+
+        mo_tag = self.doc.createElement('CODE_MO')
+        mo_tag.appendChild(self.doc.createTextNode(self.mo_code))
+
+        year_tag = self.doc.createElement('YEAR')
+        year_tag.appendChild(self.doc.createTextNode(str(self.year)))
+
+        r_tag = self.doc.createElement('R')
+        r_tag.appendChild(self.doc.createTextNode(str(self.month)))
+
+        for item in (fname_tag, data_tag, mo_tag, year_tag, r_tag):
+            zglv_tag.appendChild(item)
+
+        return zglv_tag
+
+    def fill(self, data: List[dict]):
+        for node_data in data:
+            node = self.DataNode(**node_data)
+            zap_tag = self.doc.createElement('ZAP')
+            for tag_name, tag_value in node.to_dict().items():
+                t = self.doc.createElement(tag_name)
+                t.appendChild(self.doc.createTextNode(str(tag_value)))
+                zap_tag.appendChild(t)
+            self.root.appendChild(zap_tag)
+
+    def save(self):
+        xml_path = os.path.join(os.getcwd(), f'{self.xml_name}.xml')
+        with open(xml_path, 'w', encoding='cp1251') as f:
+            f.write(self.doc.toprettyxml(indent='\t',
+                    encoding='windows-1251').decode('cp1251'))
+        logger.info(f'Файл успешно создан: {xml_path}')
+
+
+if __name__ == '__main__':
+    mo_code = input('Введите код МО (6 цифр, например, 352530): ').strip()
+
+    if not (mo_code.isdigit() and len(mo_code) == 6):
+        print('Некорректный код МО. Нужно ввести ровно 6 цифр, например 352530.')
+        exit(1)
+
+    number_package = input('Введите порядковый номер пакета: ')
+    year_now = datetime.now().year
+
+    csv_filename = f"{mo_code}.csv"
+    csv_path = os.path.join(os.getcwd(), csv_filename)
+
+    if not os.path.exists(csv_path):
+        print('Файл не найден!')
+        exit(1)
+
+    data_list = []
+    seen_policies = set()
+    mes_counter = 1
+
+    with open(csv_path, encoding='utf-8') as f:
+        reader = csv.reader(f, delimiter=';')
+        for i, row in enumerate(reader, start=1):
+            try:
+                fam, im, ot, dr, npolis, w = row[:6]
+
+                if npolis in seen_policies:
+                    continue
+                seen_policies.add(npolis)
+
+                try:
+                    dr_obj = datetime.strptime(dr, '%d.%m.%Y')
+                    dr_iso = dr_obj.strftime('%Y-%m-%d')
+                except ValueError:
+                    logger.warning(f'Ошибка формата даты в строке {i}: {dr}')
+                    dr_iso = dr
+
+                data_list.append({
+                    'fam': fam,
+                    'im': im,
+                    'ot': ot,
+                    'dr': dr_iso,
+                    'w': w,
+                    'npolis': npolis,
+                    'mes': mes_counter,
+                    'nzap': str(i),
+                    'mcode': mo_code
+                })
+
+                mes_counter += 1
+                if mes_counter > 12:
+                    mes_counter = 1
+
+            except Exception as e:
+                logger.warning(f'Ошибка в строке {i}: {e}')
+
+    generator = DNaPlanGenerator(mo_code, year_now, int(number_package))
+    generator.fill(data_list)
+    generator.save()
+
+    print('✅ XML-файл успешно создан!')