From aa062713c1980f3bd199fbe58f73a236e07ac40b Mon Sep 17 00:00:00 2001
From: Ashish Gupta <39122144+Ashish0804@users.noreply.github.com>
Date: Fri, 14 Jan 2022 18:12:58 +0530
Subject: [PATCH] [PokerGo] Add extractors (#2331)

Authored by: Ashish0804
Closes: #2316
---
 yt_dlp/extractor/extractors.py |   4 ++
 yt_dlp/extractor/pokergo.py    | 111 +++++++++++++++++++++++++++++++++
 2 files changed, 115 insertions(+)
 create mode 100644 yt_dlp/extractor/pokergo.py

diff --git a/yt_dlp/extractor/extractors.py b/yt_dlp/extractor/extractors.py
index 869aae7d9..3525bb840 100644
--- a/yt_dlp/extractor/extractors.py
+++ b/yt_dlp/extractor/extractors.py
@@ -1164,6 +1164,10 @@ from .pokemon import (
     PokemonIE,
     PokemonWatchIE,
 )
+from .pokergo import (
+    PokerGoIE,
+    PokerGoCollectionIE,
+)
 from .polsatgo import PolsatGoIE
 from .polskieradio import (
     PolskieRadioIE,
diff --git a/yt_dlp/extractor/pokergo.py b/yt_dlp/extractor/pokergo.py
new file mode 100644
index 000000000..d27031c91
--- /dev/null
+++ b/yt_dlp/extractor/pokergo.py
@@ -0,0 +1,111 @@
+# coding: utf-8
+from __future__ import unicode_literals
+
+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 _login(self):
+        username, password = self._get_login_info()
+        if not username:
+            self.raise_login_required(method='password')
+
+        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._login()
+
+
+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)