mirror of
https://github.com/nirvana-7777/script.service.ultimate.git
synced 2026-10-27 17:22:56 +01:00
569 lines
20 KiB
Python
569 lines
20 KiB
Python
# streaming_providers/providers/rtlplus/vod_manager.py
|
|
"""
|
|
RTL+ VOD Manager
|
|
|
|
Handles Video on Demand content using the Bedrock layout API.
|
|
Follows the same patterns as channel_manager and event_manager.
|
|
|
|
Manifest and DRM resolution is always delegated to the central
|
|
provider.get_manifest() / provider.get_drm() methods, which own
|
|
the upfront-token flow and all format/quality selection logic.
|
|
"""
|
|
|
|
from typing import Dict, Any, List, Optional, Union
|
|
|
|
from ...base.models.vod import VodCategory, VodItem
|
|
from ...base.models import DRMConfig
|
|
from ...base.utils.logger import logger
|
|
from .constants import RTLPlusDefaults
|
|
|
|
|
|
class RTLPlusVodManager:
|
|
"""
|
|
Manages VOD content for RTL+.
|
|
|
|
Navigation hierarchy:
|
|
root
|
|
└── folder:<id> (Bedrock folder, e.g. folder:3 = Serien)
|
|
└── program:<id> (program layout → seasons)
|
|
└── season:<block_id> (block layout → episodes)
|
|
└── clip_id (video layout, playable)
|
|
|
|
Manifest and DRM for playable items are always resolved through
|
|
the provider's central get_manifest() / get_drm() methods.
|
|
"""
|
|
|
|
def __init__(self, provider):
|
|
self._provider = provider
|
|
|
|
# ------------------------------------------------------------------
|
|
# Convenience accessors
|
|
# ------------------------------------------------------------------
|
|
|
|
@property
|
|
def cfg(self):
|
|
return self._provider.rtl_config
|
|
|
|
@property
|
|
def http(self):
|
|
return self._provider.http_manager
|
|
|
|
@property
|
|
def auth(self):
|
|
return self._provider.authenticator
|
|
|
|
# ------------------------------------------------------------------
|
|
# Public API (VodOperations interface)
|
|
# ------------------------------------------------------------------
|
|
|
|
def get_vod_category(
|
|
self,
|
|
content_id: str = "",
|
|
cursor: Optional[str] = None,
|
|
page_size: int = 24,
|
|
**kwargs,
|
|
) -> Dict[str, Any]:
|
|
"""
|
|
Return the children of a VOD node.
|
|
|
|
content_id conventions
|
|
----------------------
|
|
"" → root (fetched from home layout)
|
|
"folder:<id>" → Bedrock folder (e.g. "folder:3")
|
|
"program:<id>" → program/series (e.g. "program:10438")
|
|
"season:<id>" → season block (e.g. "season:abc-123")
|
|
|
|
Returns
|
|
-------
|
|
{
|
|
"entries": List[VodCategory | VodItem],
|
|
"next_cursor": Optional[str],
|
|
"total": Optional[int],
|
|
}
|
|
"""
|
|
if not content_id:
|
|
return self._get_root_category()
|
|
|
|
if content_id.startswith("folder:"):
|
|
folder_id = content_id[len("folder:"):]
|
|
return self._get_folder_contents(folder_id, cursor, page_size)
|
|
|
|
if content_id.startswith("program:"):
|
|
program_id = content_id[len("program:"):]
|
|
return self._get_program_seasons(program_id, cursor, page_size)
|
|
|
|
if content_id.startswith("season:"):
|
|
season_id = content_id[len("season:"):]
|
|
return self._get_season_episodes(season_id, cursor, page_size)
|
|
|
|
# Bare numeric ID → treat as program
|
|
if content_id.isdigit():
|
|
return self._get_program_seasons(content_id, cursor, page_size)
|
|
|
|
logger.warning(f"Unrecognised VOD content_id format: {content_id!r}")
|
|
return {"entries": [], "next_cursor": None, "total": 0}
|
|
|
|
def get_vod_item_info(self, clip_id: str) -> Optional[VodItem]:
|
|
"""
|
|
Return metadata for a single playable clip.
|
|
Does NOT fetch manifest/DRM — callers use get_manifest/get_drm for that.
|
|
"""
|
|
layout = self._provider.fetch_layout(
|
|
layout_type="video",
|
|
content_id=clip_id,
|
|
location=f"{self.cfg.beta_website}{clip_id}",
|
|
)
|
|
if not layout:
|
|
return None
|
|
return self._extract_vod_item_from_layout(layout, clip_id)
|
|
|
|
def get_manifest_for_video(self, clip_id: str) -> Optional[str]:
|
|
"""
|
|
Resolve the playback manifest URL for a VOD clip.
|
|
|
|
Delegates entirely to provider.get_manifest(), which owns
|
|
format/quality selection and redirect resolution.
|
|
"""
|
|
return self._provider.get_manifest(clip_id)
|
|
|
|
def get_drm_for_video(self, clip_id: str) -> List[DRMConfig]:
|
|
"""
|
|
Resolve DRM configuration (Widevine + PlayReady) for a VOD clip.
|
|
|
|
Delegates entirely to provider.get_drm(), which owns the
|
|
upfront-token fetch, asset selection, and DRMConfig construction.
|
|
"""
|
|
return self._provider.get_drm(clip_id)
|
|
|
|
# ------------------------------------------------------------------
|
|
# Root category — parsed from home layout
|
|
# ------------------------------------------------------------------
|
|
|
|
def _get_root_category(self) -> Dict[str, Any]:
|
|
"""
|
|
Return root VOD entries by parsing the Bedrock home layout.
|
|
"""
|
|
entries: List[Union[VodCategory, VodItem]] = []
|
|
|
|
layout = self._provider.fetch_layout(
|
|
layout_type="alias",
|
|
content_id="home",
|
|
location=f"{self.cfg.beta_website}",
|
|
)
|
|
|
|
if not layout:
|
|
logger.warning("Failed to fetch home layout for VOD root; returning empty")
|
|
return {"entries": [], "next_cursor": None, "total": 0}
|
|
|
|
blocks = layout.get("blocks", [])
|
|
if not blocks:
|
|
logger.warning("No blocks found in home layout")
|
|
return {"entries": [], "next_cursor": None, "total": 0}
|
|
|
|
for block in blocks:
|
|
if block.get("type") != "bffPaginated":
|
|
continue
|
|
|
|
block_title = (
|
|
block.get("analytics", {})
|
|
.get("tealium", {})
|
|
.get("block_title", "")
|
|
)
|
|
|
|
# Get content with safe defaults
|
|
content = block.get("content", {})
|
|
items = content.get("items")
|
|
|
|
# Skip if items is None or empty
|
|
if not items:
|
|
logger.debug(f"No items in block: {block_title}")
|
|
continue
|
|
|
|
# Continue-watching items - look for items with video assets
|
|
if "Weiterschauen" in block_title or "Continue" in block_title:
|
|
for item in items:
|
|
# Check if this item has video assets (playable content)
|
|
if self._item_has_video_assets(item):
|
|
vod_item = self._extract_vod_item_from_block_item(item)
|
|
if vod_item:
|
|
entries.append(vod_item)
|
|
continue
|
|
|
|
# Folder navigation block - look for folder/program navigation items
|
|
for item in items:
|
|
cat = self._extract_vod_category_from_block_item(item)
|
|
if cat:
|
|
entries.append(cat)
|
|
|
|
logger.info(f"Found {len(entries)} entries in root VOD category")
|
|
|
|
return {
|
|
"entries": entries,
|
|
"next_cursor": None,
|
|
"total": len(entries),
|
|
}
|
|
|
|
# ------------------------------------------------------------------
|
|
# Folder → program listing
|
|
# ------------------------------------------------------------------
|
|
|
|
def _get_folder_contents(
|
|
self,
|
|
folder_id: str,
|
|
cursor: Optional[str] = None,
|
|
page_size: int = 24,
|
|
) -> Dict[str, Any]:
|
|
"""
|
|
Return the programs/sub-folders inside a Bedrock folder.
|
|
Items that carry a video asset are surfaced as VodItem;
|
|
everything else becomes a VodCategory for further navigation.
|
|
"""
|
|
layout = self._provider.fetch_layout(
|
|
layout_type="folder",
|
|
content_id=folder_id,
|
|
)
|
|
if not layout:
|
|
return {"entries": [], "next_cursor": None, "total": 0}
|
|
|
|
entries: List[Union[VodCategory, VodItem]] = []
|
|
for block in layout.get("blocks", []):
|
|
if block.get("type") != "bffPaginated":
|
|
continue
|
|
for item in block.get("content", {}).get("items", []):
|
|
if self._item_has_video_assets(item):
|
|
vod_item = self._extract_vod_item_from_block_item(item)
|
|
if vod_item:
|
|
entries.append(vod_item)
|
|
else:
|
|
cat = self._extract_vod_category_from_block_item(item)
|
|
if cat:
|
|
entries.append(cat)
|
|
|
|
return {"entries": entries, "next_cursor": None, "total": len(entries)}
|
|
|
|
# ------------------------------------------------------------------
|
|
# Program → seasons
|
|
# ------------------------------------------------------------------
|
|
|
|
def _get_program_seasons(
|
|
self,
|
|
program_id: str,
|
|
cursor: Optional[str] = None,
|
|
page_size: int = 24,
|
|
) -> Dict[str, Any]:
|
|
"""
|
|
Return a VodCategory per season for the given program.
|
|
Supports simple offset-based cursor pagination over the collected list.
|
|
"""
|
|
layout = self._provider.fetch_layout(
|
|
layout_type="program",
|
|
content_id=program_id,
|
|
)
|
|
if not layout:
|
|
return {"entries": [], "next_cursor": None, "total": 0}
|
|
|
|
seasons: List[VodCategory] = []
|
|
for block in layout.get("blocks", []):
|
|
if block.get("type") != "bffPaginated":
|
|
continue
|
|
|
|
# Concurrent blocks = individual season selectors
|
|
for cb in (
|
|
block.get("alternativeContent", {}).get("concurrentBlocks", [])
|
|
):
|
|
seasons.append(VodCategory(
|
|
name=cb.get("title", "Unbekannte Staffel"),
|
|
content_id=f"season:{cb.get('id')}",
|
|
provider=self._provider.provider_name,
|
|
description=None,
|
|
child_count=(
|
|
cb.get("content", {})
|
|
.get("pagination", {})
|
|
.get("totalItems", 0)
|
|
),
|
|
))
|
|
|
|
# Block itself is a season selector
|
|
tealium = block.get("analytics", {}).get("tealium", {})
|
|
if tealium.get("template_name") == "SelectorCardListM":
|
|
seasons.append(VodCategory(
|
|
name=(
|
|
block.get("content", {})
|
|
.get("title", {})
|
|
.get("short", "Alle Staffeln")
|
|
),
|
|
content_id=f"season:{block.get('id')}",
|
|
provider=self._provider.provider_name,
|
|
child_count=(
|
|
block.get("content", {})
|
|
.get("pagination", {})
|
|
.get("totalItems", 0)
|
|
),
|
|
))
|
|
|
|
start = 0
|
|
if cursor:
|
|
try:
|
|
start = int(cursor)
|
|
except ValueError:
|
|
start = 0
|
|
|
|
page = seasons[start : start + page_size]
|
|
next_cursor = str(start + page_size) if (start + page_size) < len(seasons) else None
|
|
|
|
return {"entries": page, "next_cursor": next_cursor, "total": len(seasons)}
|
|
|
|
# ------------------------------------------------------------------
|
|
# Season block → episodes
|
|
# ------------------------------------------------------------------
|
|
|
|
def _get_season_episodes(
|
|
self,
|
|
season_block_id: str,
|
|
cursor: Optional[str] = None,
|
|
page_size: int = 24,
|
|
) -> Dict[str, Any]:
|
|
"""
|
|
Return VodItem per episode in a season block.
|
|
cursor is the Bedrock page number (1-based).
|
|
"""
|
|
page = 1
|
|
if cursor:
|
|
try:
|
|
page = int(cursor)
|
|
except ValueError:
|
|
page = 1
|
|
|
|
layout = self._provider.fetch_layout(
|
|
layout_type="block",
|
|
content_id=season_block_id,
|
|
block_page=page,
|
|
nb_pages=RTLPlusDefaults.DEFAULT_BLOCK_NB_PAGES,
|
|
)
|
|
if not layout:
|
|
return {"entries": [], "next_cursor": None, "total": 0}
|
|
|
|
episodes: List[VodItem] = []
|
|
for item in layout.get("content", {}).get("items", []):
|
|
vod_item = self._extract_vod_item_from_block_item(item)
|
|
if vod_item:
|
|
episodes.append(vod_item)
|
|
|
|
pagination = layout.get("content", {}).get("pagination", {})
|
|
total = pagination.get("totalItems", len(episodes))
|
|
next_page = pagination.get("nextPage")
|
|
next_cursor = str(next_page) if next_page else None
|
|
|
|
return {"entries": episodes, "next_cursor": next_cursor, "total": total}
|
|
|
|
# ------------------------------------------------------------------
|
|
# Extraction helpers
|
|
# ------------------------------------------------------------------
|
|
|
|
def _extract_vod_item_from_layout(
|
|
self, layout: Dict, clip_id: str
|
|
) -> Optional[VodItem]:
|
|
"""Build a VodItem from a full video layout (detail view)."""
|
|
assets = self._provider.extract_video_assets(layout)
|
|
if not assets:
|
|
return None
|
|
|
|
entity = layout.get("entity", {})
|
|
metadata = entity.get("metadata", {})
|
|
parent = layout.get("parent", {})
|
|
video_meta = layout.get("seo", {}).get("video", {})
|
|
|
|
vod_item = VodItem.create_episode(
|
|
name=metadata.get("title", clip_id),
|
|
content_id=clip_id,
|
|
provider=self._provider.provider_name,
|
|
season_number=video_meta.get("season"),
|
|
episode_number=video_meta.get("episode"),
|
|
)
|
|
vod_item.description = metadata.get("description")
|
|
vod_item.logo_url = self._extract_thumbnail_from_layout(layout)
|
|
vod_item.duration_seconds = video_meta.get("duration")
|
|
vod_item.genre = parent.get("seo", "")
|
|
vod_item.series_title = parent.get("name")
|
|
|
|
if any(a.get("drm") for a in assets):
|
|
vod_item.use_cdm = True
|
|
vod_item.cdm_type = "widevine"
|
|
|
|
return vod_item
|
|
|
|
def _extract_vod_item_from_block_item(self, item: Dict) -> Optional[VodItem]:
|
|
"""Build a VodItem from a list/block item (episode row)."""
|
|
if not item or item.get("itemType") != "classic":
|
|
return None
|
|
|
|
item_content = item.get("itemContent")
|
|
if not item_content:
|
|
return None
|
|
|
|
target = item_content.get("action", {}).get("target", {})
|
|
value_layout = target.get("value_layout", {})
|
|
|
|
if value_layout.get("type") != "video":
|
|
return None
|
|
|
|
clip_id = value_layout.get("id")
|
|
if not clip_id:
|
|
return None
|
|
|
|
title = item_content.get("title", "")
|
|
extra_title = item_content.get("extraTitle", "")
|
|
full_title = f"{title} - {extra_title}" if extra_title else title
|
|
|
|
highlight = item_content.get("highlight", "")
|
|
season_number: Optional[int] = None
|
|
episode_number: Optional[int] = None
|
|
|
|
if highlight:
|
|
import re
|
|
m = re.search(r"Staffel\s*(\d+)", highlight, re.IGNORECASE)
|
|
if m:
|
|
season_number = int(m.group(1))
|
|
m = re.search(r"Folge\s*(\d+)", highlight, re.IGNORECASE)
|
|
if m:
|
|
episode_number = int(m.group(1))
|
|
|
|
vod_item = VodItem.create_episode(
|
|
name=full_title or title or "Unbekannte Episode",
|
|
content_id=clip_id,
|
|
provider=self._provider.provider_name,
|
|
season_number=season_number,
|
|
episode_number=episode_number,
|
|
)
|
|
vod_item.description = item_content.get("description")
|
|
vod_item.logo_url = self._extract_thumbnail(item_content)
|
|
vod_item.duration_seconds = self._extract_duration(item_content)
|
|
vod_item.progress = item_content.get("progress", 0)
|
|
|
|
return vod_item
|
|
|
|
def _extract_vod_category_from_block_item(self, item: Dict) -> Optional[VodCategory]:
|
|
"""Build a VodCategory from a list/block item (folder or program row)."""
|
|
if not item or item.get("itemType") != "classic":
|
|
return None
|
|
|
|
item_content = item.get("itemContent")
|
|
if not item_content:
|
|
return None
|
|
|
|
action = item_content.get("action", {})
|
|
target = action.get("target", {})
|
|
value_layout = target.get("value_layout", {})
|
|
|
|
layout_type = value_layout.get("type")
|
|
if layout_type not in ("folder", "program"):
|
|
return None
|
|
|
|
content_id = value_layout.get("id")
|
|
if not content_id:
|
|
return None
|
|
|
|
return VodCategory(
|
|
name=item_content.get("title", "Unbekannt"),
|
|
content_id=f"{layout_type}:{content_id}",
|
|
provider=self._provider.provider_name,
|
|
logo_url=self._extract_thumbnail(item_content),
|
|
description=item_content.get("description"),
|
|
)
|
|
|
|
@staticmethod
|
|
def _item_has_video_assets(item: Dict) -> bool:
|
|
"""True when a block item directly carries playable video assets."""
|
|
return bool(
|
|
item.get("itemContent", {})
|
|
.get("video", {})
|
|
.get("assets")
|
|
)
|
|
|
|
# ------------------------------------------------------------------
|
|
# Thumbnail / duration helpers (pure, no I/O)
|
|
# ------------------------------------------------------------------
|
|
|
|
@staticmethod
|
|
def _extract_thumbnail(item_content: Dict) -> Optional[str]:
|
|
image = item_content.get("image", {})
|
|
if not image:
|
|
return None
|
|
for ratio in ("16:9", "3:1", "1:1", "2:3"):
|
|
image_id = image.get("idsByRatio", {}).get(ratio)
|
|
if image_id:
|
|
return f"https://images.rtl.de/{image_id}?format=webp&width=400"
|
|
image_id = image.get("id")
|
|
if image_id:
|
|
return f"https://images.rtl.de/{image_id}?format=webp&width=400"
|
|
return None
|
|
|
|
@staticmethod
|
|
def _extract_thumbnail_from_layout(layout: Dict) -> Optional[str]:
|
|
image_id = layout.get("seo", {}).get("image", {}).get("id")
|
|
if image_id:
|
|
return f"https://images.rtl.de/{image_id}?format=webp&width=400"
|
|
return None
|
|
|
|
@staticmethod
|
|
def _extract_duration(item_content: Dict) -> Optional[int]:
|
|
"""Parse HH:MM:SS from progressBar.endTitle → seconds."""
|
|
end_title = item_content.get("progressBar", {}).get("endTitle", "")
|
|
if end_title:
|
|
parts = end_title.split(":")
|
|
if len(parts) == 3:
|
|
try:
|
|
return int(parts[0]) * 3600 + int(parts[1]) * 60 + int(parts[2])
|
|
except ValueError:
|
|
pass
|
|
return None
|
|
|
|
# ------------------------------------------------------------------
|
|
# Cache management
|
|
# ------------------------------------------------------------------
|
|
|
|
def invalidate_cache(self, content_id: Optional[str] = None) -> None:
|
|
"""
|
|
Invalidate cached layouts for the given node, or everything.
|
|
|
|
content_id uses the same prefix conventions as get_vod_category.
|
|
"""
|
|
if not content_id:
|
|
self._provider.invalidate_layout_cache()
|
|
return
|
|
|
|
if content_id.startswith("program:"):
|
|
program_id = content_id[len("program:"):]
|
|
# Program layouts are cached under "program:<id>:<page>:<nb>"
|
|
cache_key = (
|
|
f"program:{program_id}"
|
|
f":{RTLPlusDefaults.DEFAULT_BLOCK_PAGE}"
|
|
f":{RTLPlusDefaults.DEFAULT_NB_PAGES}"
|
|
)
|
|
self._provider.invalidate_layout_cache(cache_key)
|
|
|
|
elif content_id.startswith("season:"):
|
|
season_id = content_id[len("season:"):].split("?")[0]
|
|
cache_key = (
|
|
f"block:{season_id}"
|
|
f":{RTLPlusDefaults.DEFAULT_BLOCK_PAGE}"
|
|
f":{RTLPlusDefaults.DEFAULT_BLOCK_NB_PAGES}"
|
|
)
|
|
self._provider.invalidate_layout_cache(cache_key)
|
|
|
|
elif content_id.startswith("folder:"):
|
|
folder_id = content_id[len("folder:"):]
|
|
cache_key = (
|
|
f"folder:{folder_id}"
|
|
f":{RTLPlusDefaults.DEFAULT_BLOCK_PAGE}"
|
|
f":{RTLPlusDefaults.DEFAULT_NB_PAGES}"
|
|
)
|
|
self._provider.invalidate_layout_cache(cache_key)
|
|
|
|
else:
|
|
# Unknown prefix — wipe everything rather than silently doing nothing
|
|
logger.debug(
|
|
f"invalidate_cache: unknown content_id format {content_id!r}, "
|
|
"clearing full layout cache"
|
|
)
|
|
self._provider.invalidate_layout_cache() |