From 15a0bfc7e82c03ca03e97a3ffe77c2d25b7993f2 Mon Sep 17 00:00:00 2001
From: Oliver Pfeiffer <olipfei@netscape.net>
Date: Sun, 1 Mar 2020 17:11:59 +0100
Subject: [PATCH] [kuaishou] Add new extractor

---
 youtube_dl/extractor/extractors.py |  4 ++
 youtube_dl/extractor/kuaishou.py   | 67 ++++++++++++++++++++++++++++++
 2 files changed, 71 insertions(+)
 create mode 100644 youtube_dl/extractor/kuaishou.py

diff --git a/youtube_dl/extractor/extractors.py b/youtube_dl/extractor/extractors.py
index 64d1fa251..f1fb9dee3 100644
--- a/youtube_dl/extractor/extractors.py
+++ b/youtube_dl/extractor/extractors.py
@@ -512,6 +512,10 @@ from .kinopoisk import KinoPoiskIE
 from .konserthusetplay import KonserthusetPlayIE
 from .krasview import KrasViewIE
 from .ku6 import Ku6IE
+from .kuaishou import (
+    KuaishouIE,
+    KuaishouProfileIE,
+)
 from .kusi import KUSIIE
 from .kuwo import (
     KuwoIE,
diff --git a/youtube_dl/extractor/kuaishou.py b/youtube_dl/extractor/kuaishou.py
new file mode 100644
index 000000000..7902baf4f
--- /dev/null
+++ b/youtube_dl/extractor/kuaishou.py
@@ -0,0 +1,67 @@
+# coding: utf-8
+from __future__ import unicode_literals
+
+from datetime import datetime
+import re
+
+from .common import InfoExtractor
+from ..utils import orderedSet
+
+
+class KuaishouIE(InfoExtractor):
+    _VALID_URL = r'https?://live\.kuaishou\.com/u/[a-zA-Z0-9]+/(?P<id>[a-z0-9]+)\?did=(?P<d_id>web_[a-zA-Z0-9]+)'
+    _TEST = {
+        'url': 'https://live.kuaishou.com/u/jjworld126/3xajmx3ayxnsut4?did=web_5eb8e90590774dc5b99d5af119be9911',
+        'md5': '73996963def08727f536a4a0cd030225',
+        'info_dict': {
+            'id': '3xajmx3ayxnsut4',
+            'url': 'http://jsmov2.a.yximgs.com/bs2/newWatermark/MjEwMjIwNTk0MTU_zh_4.mp4',
+            'title': '#快影 神农配合，总冠军牛牛牛。#斗地主 #电竞梦 #每个人都有自己的奋斗',
+            'thumbnail': 'http://js2.a.yximgs.com/upic/2019/12/28/19/BMjAxOTEyMjgxOTAwNTlfOTY3Mzk5MjI3XzIxMDIyMDU5NDE1XzFfMw==_Be9263b19ab4c41603ea7cbb5be834169.jpg',
+            'timestamp': 1577530862.302,
+            'upload_date': '20191228',
+            'ext': 'mp4',
+        }
+    }
+
+    def _real_extract(self, url):
+        video_id, d_id = re.match(self._VALID_URL, url).groups()
+        d_idv = (int(datetime.now().timestamp()) - 15 * 60) * 1000
+        headers = {'Cookie': 'did=%s; didv=%s' % (d_id, d_idv)}
+        webpage = self._download_webpage(url, video_id, headers=headers)
+        video_feed_json = self._search_regex(r'"VideoFeed:%s":(.*),"\$VideoFeed:%s.counts"' % (video_id, video_id), webpage, 'video_feed')
+        video_feed = self._parse_json(video_feed_json, video_id)
+
+        return {
+            'id': video_feed['id'],
+            'url': video_feed['playUrl'],
+            'title': video_feed['caption'],
+            'thumbnail': video_feed['thumbnailUrl'],
+            'timestamp': video_feed['timestamp'] / 1000,
+        }
+
+
+class KuaishouProfileIE(InfoExtractor):
+    _VALID_URL = r'https?://live\.kuaishou\.com/profile/(?P<id>[a-zA-Z0-9\-]+)\?did=(?P<d_id>web_[a-zA-Z0-9]+)'
+    _TEST = {
+        'url': 'https://live.kuaishou.com/profile/jjworld126?did=web_5da4e11838fc2152269bf983bd8180bc',
+        'info_dict': {
+            'id': 'jjworld126',
+        },
+        'playlist_mincount': 24,
+    }
+
+    def _real_extract(self, url):
+        profile_id, d_id = re.match(self._VALID_URL, url).groups()
+        d_idv = (int(datetime.now().timestamp()) - 15 * 60) * 1000
+        headers = {'Cookie': 'did=%s, didv=%s, clientid=%s, client_key=%s, kuaishou.live.bfb1s=%s' % (d_id, d_idv, 3, '65890b29', '3e261140b0cf7444a0ba411c6f227d88')}
+        webpage = self._download_webpage(url, profile_id, headers=headers)
+        profile_feeds_json = self._search_regex(r'({"pcursor":.*,"__typename":"PCProfileFeeds"})', webpage, 'profile_feeds')
+        profile_feeds = self._parse_json(profile_feeds_json, profile_id)
+
+        entries = [
+            self.url_result('https://live.kuaishou.com/u/%s/%s?did=%s' % (profile_id, video_id, d_id), KuaishouIE.ie_key(), video_id)
+            for video_id in [video_feed['id'].split(':')[1] for video_feed in profile_feeds['list']]
+        ]
+
+        return self.playlist_result(orderedSet(entries), profile_id)