diff --git a/videoarchiver/core/cleanup.py b/videoarchiver/core/cleanup.py index a286d1b..80df32a 100644 --- a/videoarchiver/core/cleanup.py +++ b/videoarchiver/core/cleanup.py @@ -8,15 +8,20 @@ from enum import Enum, auto from pathlib import Path from typing import TYPE_CHECKING, Dict, Any, Optional, TypedDict, ClassVar -from ..utils.file_ops import cleanup_downloads -from ..utils.exceptions import ( - CleanupError, - ErrorContext, - ErrorSeverity -) +try: + # Try relative imports first + from ..utils.file_ops import cleanup_downloads + from ..utils.exceptions import CleanupError, ErrorContext, ErrorSeverity +except ImportError: + # Fall back to absolute imports if relative imports fail + from videoarchiver.utils.file_ops import cleanup_downloads + from videoarchiver.utils.exceptions import CleanupError, ErrorContext, ErrorSeverity if TYPE_CHECKING: - from .base import VideoArchiver + try: + from .base import VideoArchiver + except ImportError: + from videoarchiver.core.base import VideoArchiver logger = logging.getLogger("VideoArchiver")