fix bug
This commit is contained in:
@@ -90,46 +90,57 @@ class ExcelProcessor:
|
||||
pattern = r'^\d{1,2}:\d{1,2}:\d{6,8}:\d{1,5}$'
|
||||
return bool(re.match(pattern, text))
|
||||
|
||||
def update_excel_with_file_info(self, cadastral_to_file_map: Dict[str, str],
|
||||
output_file_path: str = None) -> bool:
|
||||
"""Обновляет Excel файл информацией о найденных файлах и смежных номерах"""
|
||||
def update_excel_with_cadastral_records(self, cadastral_records: List,
|
||||
output_file_path: str = None) -> bool:
|
||||
"""Обновляет Excel файл информацией о найденных кадастровых объектах"""
|
||||
try:
|
||||
from models.cadastral_record import CadastralRecord
|
||||
|
||||
if self.df is None or self.cadastral_column is None:
|
||||
print("Excel файл не загружен или не найдена колонка с номерами")
|
||||
return False
|
||||
|
||||
# Создаем карту файл -> множество кадастровых номеров для поиска смежных
|
||||
file_to_numbers_map = {}
|
||||
for number, file_path in cadastral_to_file_map.items():
|
||||
if file_path not in file_to_numbers_map:
|
||||
file_to_numbers_map[file_path] = set()
|
||||
file_to_numbers_map[file_path].add(number)
|
||||
# Создаем карты для быстрого поиска
|
||||
# номер -> запись объекта
|
||||
number_to_record = {}
|
||||
for record in cadastral_records:
|
||||
# Основной номер
|
||||
number_to_record[record.main_number] = record
|
||||
# Все остальные номера тоже ведут к тому же объекту
|
||||
for number in record.other_numbers:
|
||||
number_to_record[number] = record
|
||||
|
||||
# Добавляем новые колонки
|
||||
self.df['Найденные файлы'] = ''
|
||||
self.df['Смежные кадастровые номера'] = ''
|
||||
self.df['Найден в файле'] = ''
|
||||
self.df['Основной номер объекта'] = ''
|
||||
self.df['Дополнительные номера'] = ''
|
||||
self.df['Кадастровый блок'] = ''
|
||||
|
||||
# Заполняем информацию для каждой строки
|
||||
matched_count = 0
|
||||
for index, row in self.df.iterrows():
|
||||
cadastral_number = str(row[self.cadastral_column]).strip()
|
||||
|
||||
if cadastral_number in cadastral_to_file_map:
|
||||
# 1. Получаем файл(ы) где найден номер
|
||||
main_file_path = cadastral_to_file_map[cadastral_number]
|
||||
file_name = Path(main_file_path).name
|
||||
self.df.at[index, 'Найденные файлы'] = file_name
|
||||
if cadastral_number in number_to_record:
|
||||
record = number_to_record[cadastral_number]
|
||||
matched_count += 1
|
||||
|
||||
# 2. Находим все смежные кадастровые номера из того же файла
|
||||
if main_file_path in file_to_numbers_map:
|
||||
all_numbers_in_file = file_to_numbers_map[main_file_path]
|
||||
# Исключаем текущий номер из списка смежных
|
||||
adjacent_numbers = all_numbers_in_file - {cadastral_number}
|
||||
if adjacent_numbers:
|
||||
self.df.at[index, 'Смежные кадастровые номера'] = ', '.join(sorted(adjacent_numbers))
|
||||
# 1. Имя файла
|
||||
file_name = Path(record.file_path).name
|
||||
self.df.at[index, 'Найден в файле'] = file_name
|
||||
|
||||
# 3. Определяем кадастровый блок (первые три части номера)
|
||||
cadastral_block = self._extract_cadastral_block(cadastral_number)
|
||||
# 2. Основной номер объекта
|
||||
self.df.at[index, 'Основной номер объекта'] = record.main_number
|
||||
|
||||
# 3. Дополнительные номера в объекте (без основного)
|
||||
other_numbers_sorted = sorted(record.other_numbers)
|
||||
if other_numbers_sorted:
|
||||
self.df.at[index, 'Дополнительные номера'] = ', '.join(other_numbers_sorted)
|
||||
else:
|
||||
self.df.at[index, 'Дополнительные номера'] = ''
|
||||
|
||||
# 4. Кадастровый блок (от основного номера)
|
||||
cadastral_block = self._extract_cadastral_block(record.main_number)
|
||||
if cadastral_block:
|
||||
self.df.at[index, 'Кадастровый блок'] = cadastral_block
|
||||
|
||||
@@ -138,6 +149,7 @@ class ExcelProcessor:
|
||||
self.df.to_excel(output_path, index=False, engine='openpyxl')
|
||||
|
||||
print(f"Обновленный Excel файл сохранен: {output_path}")
|
||||
print(f"Найдено совпадений: {matched_count} из {len(self.cadastral_numbers)}")
|
||||
return True
|
||||
|
||||
except Exception as e:
|
||||
|
||||
@@ -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