✨ Add Twitch (#118)

2025-06-17 08:25:25 +08:00 · 2021-05-02 20:03:53 +08:00 · 2021-05-02 20:03:53 +08:00 · 6a008bf0e9
commit 6a008bf0e9
parent 6aaabad43a
1 changed files with 93 additions and 0 deletions
--- a/twitch.py
+++ b/twitch.py
@ -0,0 +1,93 @@
+# -*- coding: utf-8 -*-
+# @Time: 2021/5/2 16:20
+# @Project: real-url
+# @Author: wbt5
+# @Blog: https://wbt5.com
+
+import json
+import re
+from urllib.parse import urlencode
+
+# twitch 直播需要科学上网
+import requests
+
+
+class Twitch:
+
+    def __init__(self, rid):
+        # rid = channel_name
+        self.rid = rid
+        with requests.Session() as self.s:
+            pass
+
+    def get_client_id(self):
+        try:
+            res = self.s.get(f'https://www.twitch.tv/{self.rid}').text
+            client_id = re.search(r'"Client-ID":"(.*?)"', res).group(1)
+            return client_id
+        except requests.exceptions.ConnectionError:
+            raise Exception('ConnectionError')
+
+    def get_sig_token(self):
+        data = {
+            "operationName": "PlaybackAccessToken_Template",
+            "query": "query PlaybackAccessToken_Template($login: String!, $isLive: Boolean!, $vodID: ID!, "
+                     "$isVod: Boolean!, $playerType: String!) {  streamPlaybackAccessToken(channelName: $login, "
+                     "params: {platform: \"web\", playerBackend: \"mediaplayer\", playerType: $playerType}) @include("
+                     "if: $isLive) {    value    signature    __typename  }  videoPlaybackAccessToken(id: $vodID, "
+                     "params: {platform: \"web\", playerBackend: \"mediaplayer\", playerType: $playerType}) @include("
+                     "if: $isVod) {    value    signature    __typename  }}",
+            "variables": {
+                "isLive": True,
+                "login": self.rid,
+                "isVod": False,
+                "vodID": "",
+                "playerType": "site"
+            }
+        }
+
+        headers = {
+            'Client-ID': self.get_client_id(),
+            'Referer': 'https://www.twitch.tv/',
+            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) '
+                          'Chrome/90.0.4430.93 Safari/537.36',
+        }
+        res = self.s.post('https://gql.twitch.tv/gql', headers=headers, data=json.dumps(data)).json()
+        try:
+            token, signature, _ = res['data']['streamPlaybackAccessToken'].values()
+        except AttributeError:
+            raise Exception("Channel does not exist")
+
+        return signature, token
+
+    def get_real_url(self):
+        signature, token = self.get_sig_token()
+        params = {
+            'allow_source': 'true',
+            'dt': 2,
+            'fast_bread': 'true',
+            'player_backend': 'mediaplayer',
+            'playlist_include_framerate': 'true',
+            'reassignments_supported': 'true',
+            'sig': signature,
+            'supported_codecs': 'vp09,avc1',
+            'token': token,
+            'cdm': 'wv',
+            'player_version': '1.4.0',
+        }
+        url = f'https://usher.ttvnw.net/api/channel/hls/{self.rid}.m3u8?{urlencode(params)}'
+        return url
+
+
+def get_real_url(rid):
+    try:
+        tw = Twitch(rid)
+        return tw.get_real_url()
+    except Exception as e:
+        print('Exception：', e)
+        return False
+
+
+if __name__ == '__main__':
+    r = input('请输入 twitch 房间名：\n')
+    print(get_real_url(r))