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