fix bug
This commit is contained in:
@@ -37,39 +37,52 @@ class FileSeeker:
|
||||
return xml_files
|
||||
|
||||
def process_files(self) -> List[CadastralRecord]:
|
||||
"""Обрабатывает все найденные XML файлы и извлекает кадастровые номера"""
|
||||
"""Обрабатывает все найденные XML файлы и создает записи с основными и всеми номерами"""
|
||||
all_records = []
|
||||
|
||||
for xml_file in self.xml_files:
|
||||
cadastral_numbers = xml_file.extract_cadastral_numbers()
|
||||
main_number, all_numbers = xml_file.extract_cadastral_data()
|
||||
|
||||
for number in cadastral_numbers:
|
||||
record = CadastralRecord(number, str(xml_file.file_path))
|
||||
if main_number and all_numbers:
|
||||
record = CadastralRecord(main_number, str(xml_file.file_path), all_numbers)
|
||||
all_records.append(record)
|
||||
|
||||
self.cadastral_records = all_records
|
||||
print(f"Всего найдено кадастровых номеров: {len(all_records)}")
|
||||
print(f"Создано записей объектов: {len(all_records)}")
|
||||
|
||||
# Выводим статистику
|
||||
total_numbers = sum(len(record.all_numbers) for record in all_records)
|
||||
print(f"Всего номеров во всех объектах: {total_numbers}")
|
||||
|
||||
return all_records
|
||||
|
||||
def get_unique_cadastral_numbers(self) -> Set[str]:
|
||||
"""Возвращает уникальные кадастровые номера"""
|
||||
return {record.number for record in self.cadastral_records}
|
||||
"""Возвращает уникальные основные кадастровые номера"""
|
||||
return {record.main_number for record in self.cadastral_records}
|
||||
|
||||
def get_all_cadastral_numbers(self) -> Set[str]:
|
||||
"""Возвращает все уникальные кадастровые номера из всех объектов"""
|
||||
all_numbers = set()
|
||||
for record in self.cadastral_records:
|
||||
all_numbers.update(record.all_numbers)
|
||||
return all_numbers
|
||||
|
||||
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:
|
||||
# Проверяем, есть ли какой-либо из номеров объекта в списке для перемещения
|
||||
if any(num in cadastral_numbers_to_move for num in record.all_numbers):
|
||||
source_path = Path(record.file_path)
|
||||
if source_path.exists():
|
||||
# Создаем уникальное имя для файла во временной папке
|
||||
dest_name = f"{record.number.replace(':', '_')}_{source_path.name}"
|
||||
dest_name = f"{record.main_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)
|
||||
moved_files[record.main_number] = str(dest_path)
|
||||
print(f"Файл скопирован: {source_path.name} -> {dest_path}")
|
||||
except Exception as e:
|
||||
print(f"Ошибка при копировании файла {source_path}: {e}")
|
||||
@@ -78,14 +91,20 @@ class FileSeeker:
|
||||
|
||||
def get_statistics(self) -> Dict[str, int]:
|
||||
"""Возвращает статистику обработки"""
|
||||
total_all_numbers = sum(len(record.all_numbers) for record in self.cadastral_records)
|
||||
|
||||
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())
|
||||
'files_with_numbers': len([f for f in self.xml_files if f.main_cadastral_number]),
|
||||
'total_cadastral_objects': len(self.cadastral_records),
|
||||
'unique_main_numbers': len(self.get_unique_cadastral_numbers()),
|
||||
'total_all_numbers': total_all_numbers,
|
||||
'unique_all_numbers': len(self.get_all_cadastral_numbers())
|
||||
}
|
||||
|
||||
def __str__(self):
|
||||
stats = self.get_statistics()
|
||||
return (f"FileSeeker: {stats['total_xml_files']} XML файлов, "
|
||||
f"{stats['unique_cadastral_numbers']} уникальных номеров")
|
||||
f"{stats['total_cadastral_objects']} объектов, "
|
||||
f"{stats['unique_main_numbers']} уникальных основных номеров, "
|
||||
f"{stats['unique_all_numbers']} всего уникальных номеров")
|
||||
|
||||
Reference in New Issue
Block a user