fix(core): add fallback to absolute imports in cleanup.py

This commit is contained in:
pacnpal
2024-11-17 22:10:44 +00:00
parent c0c70f346e
commit 26f173ce72

View File

@@ -8,15 +8,20 @@ from enum import Enum, auto
from pathlib import Path from pathlib import Path
from typing import TYPE_CHECKING, Dict, Any, Optional, TypedDict, ClassVar from typing import TYPE_CHECKING, Dict, Any, Optional, TypedDict, ClassVar
from ..utils.file_ops import cleanup_downloads try:
from ..utils.exceptions import ( # Try relative imports first
CleanupError, from ..utils.file_ops import cleanup_downloads
ErrorContext, from ..utils.exceptions import CleanupError, ErrorContext, ErrorSeverity
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: if TYPE_CHECKING:
from .base import VideoArchiver try:
from .base import VideoArchiver
except ImportError:
from videoarchiver.core.base import VideoArchiver
logger = logging.getLogger("VideoArchiver") logger = logging.getLogger("VideoArchiver")