summaryrefslogtreecommitdiffstats
path: root/yt_dlp/extractor/pokergo.py
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-15 16:49:24 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-15 16:49:24 +0000
commit2415e66f889f38503b73e8ebc5f43ca342390e5c (patch)
treeac48ab69d1d96bae3d83756134921e0d90593aa5 /yt_dlp/extractor/pokergo.py
parentInitial commit. (diff)
downloadyt-dlp-2415e66f889f38503b73e8ebc5f43ca342390e5c.tar.xz
yt-dlp-2415e66f889f38503b73e8ebc5f43ca342390e5c.zip
Adding upstream version 2024.03.10.upstream/2024.03.10
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'yt_dlp/extractor/pokergo.py')
-rw-r--r--yt_dlp/extractor/pokergo.py106
1 files changed, 106 insertions, 0 deletions
diff --git a/yt_dlp/extractor/pokergo.py b/yt_dlp/extractor/pokergo.py
new file mode 100644
index 0000000..5c7baad
--- /dev/null
+++ b/yt_dlp/extractor/pokergo.py
@@ -0,0 +1,106 @@
+import base64
+
+from .common import InfoExtractor
+from ..utils import (
+ ExtractorError,
+ try_get,
+)
+
+
+class PokerGoBaseIE(InfoExtractor):
+ _NETRC_MACHINE = 'pokergo'
+ _AUTH_TOKEN = None
+ _PROPERTY_ID = '1dfb3940-7d53-4980-b0b0-f28b369a000d'
+
+ def _perform_login(self, username, password):
+ if self._AUTH_TOKEN:
+ return
+ self.report_login()
+ PokerGoBaseIE._AUTH_TOKEN = self._download_json(
+ f'https://subscription.pokergo.com/properties/{self._PROPERTY_ID}/sign-in', None,
+ headers={'authorization': f'Basic {base64.b64encode(f"{username}:{password}".encode()).decode()}'},
+ data=b'')['meta']['token']
+ if not self._AUTH_TOKEN:
+ raise ExtractorError('Unable to get Auth Token.', expected=True)
+
+ def _real_initialize(self):
+ if not self._AUTH_TOKEN:
+ self.raise_login_required(method='password')
+
+
+class PokerGoIE(PokerGoBaseIE):
+ _VALID_URL = r'https?://(?:www\.)?pokergo\.com/videos/(?P<id>[^&$#/?]+)'
+
+ _TESTS = [{
+ 'url': 'https://www.pokergo.com/videos/2a70ec4e-4a80-414b-97ec-725d9b72a7dc',
+ 'info_dict': {
+ 'id': 'aVLOxDzY',
+ 'ext': 'mp4',
+ 'title': 'Poker After Dark | Season 12 (2020) | Cry Me a River | Episode 2',
+ 'description': 'md5:c7a8c29556cbfb6eb3c0d5d622251b71',
+ 'thumbnail': 'https://cdn.jwplayer.com/v2/media/aVLOxDzY/poster.jpg?width=720',
+ 'timestamp': 1608085715,
+ 'duration': 2700.12,
+ 'season_number': 12,
+ 'episode_number': 2,
+ 'series': 'poker after dark',
+ 'upload_date': '20201216',
+ 'season': 'Season 12',
+ 'episode': 'Episode 2',
+ 'display_id': '2a70ec4e-4a80-414b-97ec-725d9b72a7dc',
+ },
+ 'params': {'skip_download': True}
+ }]
+
+ def _real_extract(self, url):
+ id = self._match_id(url)
+ data_json = self._download_json(f'https://api.pokergo.com/v2/properties/{self._PROPERTY_ID}/videos/{id}', id,
+ headers={'authorization': f'Bearer {self._AUTH_TOKEN}'})['data']
+ v_id = data_json['source']
+
+ thumbnails = [{
+ 'url': image['url'],
+ 'id': image.get('label'),
+ 'width': image.get('width'),
+ 'height': image.get('height')
+ } for image in data_json.get('images') or [] if image.get('url')]
+ series_json = next(dct for dct in data_json.get('show_tags') or [] if dct.get('video_id') == id) or {}
+
+ return {
+ '_type': 'url_transparent',
+ 'display_id': id,
+ 'title': data_json.get('title'),
+ 'description': data_json.get('description'),
+ 'duration': data_json.get('duration'),
+ 'thumbnails': thumbnails,
+ 'season_number': series_json.get('season'),
+ 'episode_number': series_json.get('episode_number'),
+ 'series': try_get(series_json, lambda x: x['tag']['name']),
+ 'url': f'https://cdn.jwplayer.com/v2/media/{v_id}'
+ }
+
+
+class PokerGoCollectionIE(PokerGoBaseIE):
+ _VALID_URL = r'https?://(?:www\.)?pokergo\.com/collections/(?P<id>[^&$#/?]+)'
+
+ _TESTS = [{
+ 'url': 'https://www.pokergo.com/collections/19ffe481-5dae-481a-8869-75cc0e3c4700',
+ 'playlist_mincount': 13,
+ 'info_dict': {
+ 'id': '19ffe481-5dae-481a-8869-75cc0e3c4700',
+ },
+ }]
+
+ def _entries(self, id):
+ data_json = self._download_json(f'https://api.pokergo.com/v2/properties/{self._PROPERTY_ID}/collections/{id}?include=entities',
+ id, headers={'authorization': f'Bearer {self._AUTH_TOKEN}'})['data']
+ for video in data_json.get('collection_video') or []:
+ video_id = video.get('id')
+ if video_id:
+ yield self.url_result(
+ f'https://www.pokergo.com/videos/{video_id}',
+ ie=PokerGoIE.ie_key(), video_id=video_id)
+
+ def _real_extract(self, url):
+ id = self._match_id(url)
+ return self.playlist_result(self._entries(id), playlist_id=id)