Astro-Session-Watcher/services/file_service.py

"""
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