From ddaab828fac181c38f06dec371cf09aae72ddf94 Mon Sep 17 00:00:00 2001 From: pacnpal <183241239+pacnpal@users.noreply.github.com> Date: Sun, 17 Nov 2024 22:14:23 +0000 Subject: [PATCH] fix(core): add fallback to absolute imports in lifecycle.py --- videoarchiver/core/lifecycle.py | 28 ++++++++++++++++++++-------- 1 file changed, 20 insertions(+), 8 deletions(-) diff --git a/videoarchiver/core/lifecycle.py b/videoarchiver/core/lifecycle.py index 7d00433..d078849 100644 --- a/videoarchiver/core/lifecycle.py +++ b/videoarchiver/core/lifecycle.py @@ -7,14 +7,26 @@ from typing import Optional, Dict, Any, Set, List, Callable, TypedDict, ClassVar from enum import Enum, auto from datetime import datetime -from ..core.cleanup import cleanup_resources, force_cleanup_resources -from ..utils.exceptions import ( - VideoArchiverError, - ErrorContext, - ErrorSeverity, - ComponentError, - CleanupError, -) +try: + # Try relative imports first + from .cleanup import cleanup_resources, force_cleanup_resources + from ..utils.exceptions import ( + VideoArchiverError, + ErrorContext, + ErrorSeverity, + ComponentError, + CleanupError, + ) +except ImportError: + # Fall back to absolute imports if relative imports fail + from videoarchiver.core.cleanup import cleanup_resources, force_cleanup_resources + from videoarchiver.utils.exceptions import ( + VideoArchiverError, + ErrorContext, + ErrorSeverity, + ComponentError, + CleanupError, + ) logger = logging.getLogger("VideoArchiver")