Рефакторинг проекта: разделение на модули, улучшенный интерфейс, обработка Excel

services/file_seeker.py

@@ -0,0 +1,91 @@
+"""
+Сервис для поиска и обработки XML файлов.
+"""
+import os
+import shutil
+import tempfile
+from pathlib import Path
+from typing import List, Set, Dict
+from models.xml_file_info import XMLFileInfo
+from models.cadastral_record import CadastralRecord
+
+class FileSeeker:
+    def __init__(self, base_directory: str, temp_directory: str = None):
+        self.base_directory = Path(base_directory)
+        self.temp_directory = Path(temp_directory) if temp_directory else Path(tempfile.mkdtemp(prefix="kadastr14_"))
+        self.xml_files: List[XMLFileInfo] = []
+        self.cadastral_records: List[CadastralRecord] = []
+        
+        # Создаем временную директорию если её нет
+        self.temp_directory.mkdir(parents=True, exist_ok=True)
+    
+    def find_xml_files(self) -> List[XMLFileInfo]:
+        """Находит все XML файлы в базовой директории"""
+        xml_files = []
+        
+        if not self.base_directory.exists():
+            print(f"Директория {self.base_directory} не существует")
+            return xml_files
+        
+        for file_path in self.base_directory.rglob("*.xml"):
+            if file_path.is_file():
+                xml_info = XMLFileInfo(str(file_path))
+                xml_files.append(xml_info)
+        
+        self.xml_files = xml_files
+        print(f"Найдено XML файлов: {len(xml_files)}")
+        return xml_files
+    
+    def process_files(self) -> List[CadastralRecord]:
+        """Обрабатывает все найденные XML файлы и извлекает кадастровые номера"""
+        all_records = []
+        
+        for xml_file in self.xml_files:
+            cadastral_numbers = xml_file.extract_cadastral_numbers()
+            
+            for number in cadastral_numbers:
+                record = CadastralRecord(number, str(xml_file.file_path))
+                all_records.append(record)
+        
+        self.cadastral_records = all_records
+        print(f"Всего найдено кадастровых номеров: {len(all_records)}")
+        return all_records
+    
+    def get_unique_cadastral_numbers(self) -> Set[str]:
+        """Возвращает уникальные кадастровые номера"""
+        return {record.number for record in self.cadastral_records}
+    
+    def move_files_to_temp(self, cadastral_numbers_to_move: Set[str]) -> Dict[str, str]:
+        """Перемещает файлы с указанными кадастровыми номерами во временную папку"""
+        moved_files = {}
+        
+        for record in self.cadastral_records:
+            if record.number in cadastral_numbers_to_move:
+                source_path = Path(record.file_path)
+                if source_path.exists():
+                    # Создаем уникальное имя для файла во временной папке
+                    dest_name = f"{record.number.replace(':', '_')}_{source_path.name}"
+                    dest_path = self.temp_directory / dest_name
+                    
+                    try:
+                        shutil.copy2(source_path, dest_path)
+                        moved_files[record.number] = str(dest_path)
+                        print(f"Файл скопирован: {source_path.name} -> {dest_path}")
+                    except Exception as e:
+                        print(f"Ошибка при копировании файла {source_path}: {e}")
+        
+        return moved_files
+    
+    def get_statistics(self) -> Dict[str, int]:
+        """Возвращает статистику обработки"""
+        return {
+            'total_xml_files': len(self.xml_files),
+            'files_with_numbers': len([f for f in self.xml_files if len(f.cadastral_numbers) > 0]),
+            'total_cadastral_numbers': len(self.cadastral_records),
+            'unique_cadastral_numbers': len(self.get_unique_cadastral_numbers())
+        }
+    
+    def __str__(self):
+        stats = self.get_statistics()
+        return (f"FileSeeker: {stats['total_xml_files']} XML файлов, "
+                f"{stats['unique_cadastral_numbers']} уникальных номеров")