From f91625cd122d913eddcb8ad7860644bc17717ce5 Mon Sep 17 00:00:00 2001
From: Alexander Nartov <alexander.nartov@gmail.com>
Date: Mon, 21 Sep 2020 03:17:02 +1000
Subject: [PATCH] [rutube] Better login and error handling, code style fixes

---
 youtube_dl/extractor/rutube.py | 76 +++++++++++++++++++++++-----------
 1 file changed, 51 insertions(+), 25 deletions(-)

diff --git a/youtube_dl/extractor/rutube.py b/youtube_dl/extractor/rutube.py
index f82448f9b..fdda8c4c9 100644
--- a/youtube_dl/extractor/rutube.py
+++ b/youtube_dl/extractor/rutube.py
@@ -18,7 +18,7 @@ from ..utils import (
     try_get,
     unified_timestamp,
     url_or_none,
-)
+    ExtractorError)
 
 
 class RutubeBaseIE(InfoExtractor):
@@ -61,6 +61,39 @@ class RutubeBaseIE(InfoExtractor):
             'is_club': bool_or_none(video.get('is_club')),
         }
 
+    def _real_initialize(self):
+        self._login()
+
+    def _login(self):
+        username, password = self._get_login_info()
+        if username is None:
+            return
+
+        login = self._download_json(
+            'https://pass.rutube.ru/api/accounts/phone/login/', None,
+            'Logging in', 'Unable to log in',
+            data=json.dumps({
+                'phone': username,
+                'password': password,
+            }).encode(),
+            headers={
+                'Content-Type': 'application/json',
+                'Accept': 'application/json'
+            },
+            expected_status=400
+        )
+        if not login.get('success'):
+            msg = login.get('message')
+            raise ExtractorError(
+                'Unable to log in. %s said: %s' % (self.IE_NAME, msg),
+                expected=True
+            )
+
+        self._download_webpage(
+            'https://rutube.ru/social/auth/rupass/?callback_path=/social/login/rupass/',
+            None, False
+        )
+
     def _download_and_extract_info(self, video_id, query=None):
         return self._extract_info(
             self._download_api_info(video_id, query=query), video_id)
@@ -148,33 +181,26 @@ class RutubeIE(RutubeBaseIE):
         query = {}
 
         if info['is_club']:
-            username, password = self._get_login_info()
-            if username is None or password is None:
+            visitor_json, urlh = self._download_webpage_handle(
+                'https://rutube.ru/api/accounts/visitor/', video_id,
+                'Downloading visitor JSON', 'Unable to download visitor JSON'
+            )
+            if not visitor_json:
                 self.raise_login_required()
 
-            login = self._download_json(
-                'https://pass.rutube.ru/api/accounts/phone/login/', video_id, data=json.dumps({
-                    'phone': username,
-                    'password': password,
-                }).encode(),
-                headers={'Content-Type': 'application/json', 'Accept': 'application/json'}
-            )
-            if not login['success']:
-                self.raise_login_required('Invalid login or password')
-
-            self._download_webpage(
-                'https://rutube.ru/social/auth/rupass/?callback_path=/social/login/rupass/', video_id)
-
-            visitor = self._download_json(
-                'https://rutube.ru/api/accounts/visitor/', video_id, 'Downloading visitor JSON',
-                'Unable to download visitor JSON')
-            clubParams = visitor['club_params_encrypted']
+            visitor = self._parse_json(visitor_json, video_id)
+            club_params = visitor['club_params_encrypted']
             ad = self._download_json(
-                'https://mtr.rutube.ru/api/v3/interactive?' + clubParams + '&video_id=' + video_id, video_id,
-                'Downloading AD JSON', 'Unable to download AD JSON')
-            clubToken = ad['award']
-            query['club_token'] = clubToken
-            query['no_404'] = 'true'
+                'https://mtr.rutube.ru/api/v3/interactive?%s&video_id=%s' %
+                (club_params, video_id),
+                video_id, 'Downloading AD JSON', 'Unable to download AD JSON'
+            )
+            club_token = ad['award']
+
+            query.update({
+                'club_token': club_token,
+                'no_404': 'true',
+            })
 
         info['formats'] = self._download_and_extract_formats(video_id, query)
         return info