diff --git a/lib/streaming_providers/base/provider.py b/lib/streaming_providers/base/provider.py index bcb146e..8922a3c 100644 --- a/lib/streaming_providers/base/provider.py +++ b/lib/streaming_providers/base/provider.py @@ -105,10 +105,13 @@ class StreamingProvider(ABC): return cls.SUPPORTED_AUTH_TYPES.copy() @classmethod - def get_static_logo(cls) -> str: + def get_static_logo(cls, country: str = None) -> str: """ Get provider logo URL without instantiation. + Args: + country: Optional country code for country-specific logos + Returns: Logo URL string """ diff --git a/lib/streaming_providers/base/provider_registry.py b/lib/streaming_providers/base/provider_registry.py index 56f2712..2aa284c 100644 --- a/lib/streaming_providers/base/provider_registry.py +++ b/lib/streaming_providers/base/provider_registry.py @@ -46,7 +46,7 @@ class ProviderMetadata: # Rest of the method remains the same... self.label = self.plugin_class.get_static_label(self.country) self.supported_auth_types = self.plugin_class.get_static_auth_types() - self.logo = self.plugin_class.get_static_logo() + self.logo = self.plugin_class.get_static_logo(self.country) self.supported_countries = self.plugin_class.get_static_supported_countries() self.requires_credentials = any( auth_type in ['user_credentials', 'client_credentials']