88 lines
No EOL
3.1 KiB
Python
88 lines
No EOL
3.1 KiB
Python
"""
|
||
FileService - сервис для работы с файлами
|
||
"""
|
||
import shutil
|
||
from datetime import datetime
|
||
from pathlib import Path
|
||
from typing import Optional
|
||
|
||
|
||
class FileService:
|
||
"""Сервис для перемещения файлов и ведения логов"""
|
||
|
||
SUPPORTED_EXTENSIONS = {'.cr2', '.dng', '.arw', '.jpg', '.jpeg', '.png', '.raw', '.tiff'}
|
||
|
||
@classmethod
|
||
def is_photo(cls, file_path: Path) -> bool:
|
||
"""Проверяет, является ли файл фотографией"""
|
||
return file_path.suffix.lower() in cls.SUPPORTED_EXTENSIONS
|
||
|
||
@classmethod
|
||
def resolve_conflict(cls, target_path: Path) -> Path:
|
||
"""Разрешает конфликт имён файлов"""
|
||
if not target_path.exists():
|
||
return target_path
|
||
|
||
counter = 1
|
||
stem = target_path.stem
|
||
suffix = target_path.suffix
|
||
parent = target_path.parent
|
||
|
||
while True:
|
||
new_name = f"{stem}_{counter}{suffix}"
|
||
new_path = parent / new_name
|
||
if not new_path.exists():
|
||
return new_path
|
||
counter += 1
|
||
|
||
@classmethod
|
||
def write_object_log(cls, folder: Path, filename: str, camera: str, optics: str,
|
||
timestamp: Optional[datetime] = None) -> None:
|
||
"""Записывает запись в лог объекта"""
|
||
if timestamp is None:
|
||
timestamp = datetime.now()
|
||
|
||
log_file = folder / "ObjectLog.txt"
|
||
line = f"{timestamp} {camera if camera else 'Unknown'} {optics if optics else 'Unknown'} - {filename}\n"
|
||
|
||
try:
|
||
with open(log_file, 'a', encoding='utf-8') as f:
|
||
f.write(line)
|
||
except Exception as e:
|
||
print(f"Ошибка записи лога: {e}")
|
||
|
||
@classmethod
|
||
def move_file(cls, source: Path, target_folder: Path, camera: str, optics: str) -> bool:
|
||
"""Перемещает файл в целевую папку"""
|
||
if not source.exists():
|
||
return False
|
||
|
||
if not cls.is_photo(source):
|
||
return False
|
||
|
||
try:
|
||
target_folder.mkdir(parents=True, exist_ok=True)
|
||
creation_time = datetime.fromtimestamp(source.stat().st_ctime)
|
||
cls.write_object_log(target_folder, source.name, camera, optics, creation_time)
|
||
target_path = cls.resolve_conflict(target_folder / source.name)
|
||
shutil.move(str(source), str(target_path))
|
||
return True
|
||
except Exception as e:
|
||
print(f"Ошибка перемещения {source.name}: {e}")
|
||
return False
|
||
|
||
@classmethod
|
||
def clear_watch_folder(cls, folder: Path) -> int:
|
||
"""Очищает папку наблюдения"""
|
||
if not folder.exists():
|
||
return 0
|
||
|
||
count = 0
|
||
for file_path in folder.iterdir():
|
||
if file_path.is_file() and cls.is_photo(file_path):
|
||
try:
|
||
file_path.unlink()
|
||
count += 1
|
||
except Exception as e:
|
||
print(f"Ошибка удаления {file_path.name}: {e}")
|
||
return count |