diff options
| author | Sergey M․ <dstftw@gmail.com> | 2021-01-06 02:10:44 +0700 | 
|---|---|---|
| committer | Sergey M․ <dstftw@gmail.com> | 2021-01-06 02:11:49 +0700 | 
| commit | ccc71122915e630d99e8266c73a2eba26707f199 (patch) | |
| tree | 6c7f476c8587e3b75cf93dabc0e181cac13c9057 /youtube_dl/extractor/twitch.py | |
| parent | 5b24f8f505582f353c3c2f7b79b5f67612ba9c87 (diff) | |
[twitch] Improve access token extraction and remove unused code (closes #27646)
Diffstat (limited to 'youtube_dl/extractor/twitch.py')
| -rw-r--r-- | youtube_dl/extractor/twitch.py | 33 | 
1 files changed, 15 insertions, 18 deletions
| diff --git a/youtube_dl/extractor/twitch.py b/youtube_dl/extractor/twitch.py index 50dcb93ef..a939ea24e 100644 --- a/youtube_dl/extractor/twitch.py +++ b/youtube_dl/extractor/twitch.py @@ -192,29 +192,27 @@ class TwitchGraphQLBaseIE(TwitchBaseIE):              }          return self._download_base_gql(video_id, ops, note) -    def _download_access_token_gql(self, video_id, item_type=None): -        if item_type == 'vod': -            method = 'videoPlaybackAccessToken' -            param_name = 'id' -        else: -            method = 'streamPlaybackAccessToken' -            param_name = 'channelName' +    def _download_access_token_gql(self, video_id, token_kind, param_name): +        method = '%sPlaybackAccessToken' % token_kind          ops = {              'query': '''{                %s(                  %s: "%s", -                  params: { -                    platform: "web", -                    playerBackend: "mediaplayer", -                    playerType: "site" -                  }) { +                params: { +                  platform: "web", +                  playerBackend: "mediaplayer", +                  playerType: "site" +                } +              ) +              {                  value                  signature                }              }''' % (method, param_name, video_id),          } -        note = 'Downloading access token GraphQL' -        return self._download_base_gql(video_id, ops, note)['data'][method] +        return self._download_base_gql( +            video_id, ops, +            'Downloading %s access token GraphQL' % token_kind)['data'][method]  class TwitchVodIE(TwitchGraphQLBaseIE): @@ -227,8 +225,6 @@ class TwitchVodIE(TwitchGraphQLBaseIE):                          )                          (?P<id>\d+)                      ''' -    _ITEM_TYPE = 'vod' -    _ITEM_SHORTCUT = 'v'      _TESTS = [{          'url': 'http://www.twitch.tv/riotgames/v/6528877?t=5m10s', @@ -333,7 +329,7 @@ class TwitchVodIE(TwitchGraphQLBaseIE):          vod_id = self._match_id(url)          info = self._download_info(vod_id) -        access_token = self._download_access_token_gql(vod_id, self._ITEM_TYPE) +        access_token = self._download_access_token_gql(vod_id, 'video', 'id')          formats = self._extract_m3u8_formats(              '%s/vod/%s.m3u8?%s' % ( @@ -839,7 +835,8 @@ class TwitchStreamIE(TwitchGraphQLBaseIE):          if not stream:              raise ExtractorError('%s is offline' % channel_name, expected=True) -        access_token = self._download_access_token_gql(channel_name) +        access_token = self._download_access_token_gql( +            channel_name, 'stream', 'channelName')          token = access_token['value']          stream_id = stream.get('id') or channel_name | 
