fix catchup

lib/streaming_providers/base/catchup_operations.py

@@ -25,6 +25,7 @@ class CatchupOperations:
         end_time: int,
         epg_id: Optional[str] = None,
         country: Optional[str] = None,
+        drm_variant: Optional[str] = "auto",
     ) -> Optional[str]:
         """Get catchup manifest URL."""
         provider = self.registry.get_provider(provider_name)
@@ -47,6 +48,7 @@ class CatchupOperations:
                 end_time=end_time,
                 epg_id=epg_id,
                 country=country,
+                drm_variant=drm_variant,
             )
         except NotImplementedError:
             logger.error(f"Provider '{provider_name}' hasn't implemented catchup")

lib/streaming_providers/base/manager.py

@@ -194,9 +194,10 @@ class ProviderManager:
             end_time: int,
             epg_id: Optional[str] = None,
             country: Optional[str] = None,
+            drm_variant = "auto",
     ) -> Optional[str]:
         return self.catchup_ops.get_catchup_manifest(
-            provider_name, channel_id, start_time, end_time, epg_id, country
+            provider_name, channel_id, start_time, end_time, epg_id, country, drm_variant
         )
 
     def get_catchup_drm_configs(

lib/streaming_providers/base/provider.py

@@ -1189,6 +1189,7 @@ class StreamingProvider(ABC):
         start_time: int,
         end_time: int,
         epg_id: Optional[str] = None,
+        drm_variant: Optional[str] = "auto",
         **kwargs,
     ) -> Optional[str]:
         """

lib/streaming_providers/providers/magenta2/playback_manager.py

@@ -179,7 +179,7 @@ class PlaybackManager:
         return self._smil_manager.get_drm(content_id, content_type, **kwargs)
 
     def get_catchup_manifest(
-        self, content_id: str, start_time: int, end_time: int, **kwargs
+        self, content_id: str, start_time: int, end_time: int, drm_variant: str, **kwargs
     ) -> Optional[str]:
         """
         Return a catchup manifest URL for *channel_id*.

lib/streaming_providers/providers/magenta2/provider.py

@@ -664,10 +664,10 @@ class Magenta2Provider(StreamingProvider):
         return self._playback_manager.get_drm(content_id, content_type, **kwargs)
 
     def get_catchup_manifest(
-        self, content_id: str, start_time: int, end_time: int, **kwargs: Any
+        self, content_id: str, start_time: int, end_time: int, drm_variant: str, **kwargs: Any
     ) -> Optional[str]:
         return self._playback_manager.get_catchup_manifest(
-            content_id, start_time, end_time, **kwargs
+            content_id, start_time, end_time, drm_variant, **kwargs
         )
 
     def get_vod_category(

lib/streaming_providers/providers/magentaeu/provider.py

@@ -365,7 +365,7 @@ class MagentaEUProvider(StreamingProvider):
         return None
 
     def get_catchup_manifest(
-            self, content_id: str, start_time: int, end_time: int, **kwargs
+            self, content_id: str, start_time: int, end_time: int, drm_variant: str, **kwargs
     ) -> Optional[str]:
         """
         Get catchup manifest URL for Magenta TV.