fix(processor): add fallback to absolute imports in __init__.py

This commit is contained in:
pacnpal
2024-11-17 22:52:16 +00:00
parent c1bfc5e90b
commit 643763db0a

View File

@@ -3,6 +3,8 @@
from typing import Dict, Any, Optional, Union, List, Tuple
import discord # type: ignore
try:
# Try relative imports first
from .core import VideoProcessor
from .constants import (
REACTIONS,
@@ -41,6 +43,46 @@ from .reactions import (
update_download_progress_reaction,
)
from ..utils.progress_tracker import progress_tracker
except ImportError:
# Fall back to absolute imports if relative imports fail
from videoarchiver.processor.core import VideoProcessor
from videoarchiver.processor.constants import (
REACTIONS,
ReactionType,
ReactionEmojis,
ProgressEmojis,
get_reaction,
get_progress_emoji,
)
from videoarchiver.processor.url_extractor import (
URLExtractor,
URLMetadata,
URLPattern,
URLType,
URLPatternManager,
URLValidator,
URLMetadataExtractor,
)
from videoarchiver.processor.message_validator import (
MessageValidator,
ValidationContext,
ValidationRule,
ValidationResult,
ValidationRuleManager,
ValidationCache,
ValidationStats,
ValidationCacheEntry,
ValidationError,
)
from videoarchiver.processor.message_handler import MessageHandler
from videoarchiver.processor.queue_handler import QueueHandler
from videoarchiver.processor.reactions import (
handle_archived_reaction,
update_queue_position_reaction,
update_progress_reaction,
update_download_progress_reaction,
)
from videoarchiver.utils.progress_tracker import progress_tracker
# Export public classes and constants
__all__ = [