From e3ff1709b271d6f3a3ab7a9ef1174fe44a1760f1 Mon Sep 17 00:00:00 2001 From: Nirvana Date: Thu, 25 Dec 2025 21:32:10 +0100 Subject: [PATCH] Restructure for enabling/disabling --- lib/streaming_providers/base/provider.py | 2 +- lib/streaming_providers/providers/hrti/provider.py | 1 + lib/streaming_providers/providers/magentaeu/__init__.py | 4 ++-- 3 files changed, 4 insertions(+), 3 deletions(-) diff --git a/lib/streaming_providers/base/provider.py b/lib/streaming_providers/base/provider.py index 124d0c9..bcb146e 100644 --- a/lib/streaming_providers/base/provider.py +++ b/lib/streaming_providers/base/provider.py @@ -931,7 +931,7 @@ class StreamingProvider(ABC): Returns: True if provider supports country-specific instances """ - return len(cls.SUPPORTED_COUNTRIES) > 0 + return len(cls.SUPPORTED_COUNTRIES) > 1 @classmethod def validate_country(cls, country: str) -> bool: diff --git a/lib/streaming_providers/providers/hrti/provider.py b/lib/streaming_providers/providers/hrti/provider.py index 93786df..cfc5d73 100644 --- a/lib/streaming_providers/providers/hrti/provider.py +++ b/lib/streaming_providers/providers/hrti/provider.py @@ -24,6 +24,7 @@ class HRTiProvider(StreamingProvider): PROVIDER_LABEL: ClassVar[str] = "HRTi" SUPPORTED_AUTH_TYPES: ClassVar[List[str]] = ['client_credentials', 'user_credentials'] PROVIDER_LOGO: ClassVar[str] = HRTiDefaults.PROVIDER_LOGO + SUPPORTED_COUNTRIES: ClassVar[List[str]] = ['HR'] def __init__(self, country: str = 'HR', config: Optional[Dict] = None, proxy_config: Optional[ProxyConfig] = None): super().__init__(country) diff --git a/lib/streaming_providers/providers/magentaeu/__init__.py b/lib/streaming_providers/providers/magentaeu/__init__.py index 8a2a07a..0f8e4a8 100644 --- a/lib/streaming_providers/providers/magentaeu/__init__.py +++ b/lib/streaming_providers/providers/magentaeu/__init__.py @@ -1,5 +1,5 @@ # streaming_providers/providers/magenta_eu/__init__.py -from .provider import MagentaProvider +from .provider import MagentaEUProvider from .auth import MagentaAuthenticator, MagentaAuthToken from .constants import ( SUPPORTED_COUNTRIES, @@ -9,7 +9,7 @@ from .constants import ( ) __all__ = [ - 'MagentaProvider', + 'MagentaEUProvider', 'MagentaAuthenticator', 'MagentaAuthToken', 'SUPPORTED_COUNTRIES',