Login Captcha with redirect_uri

Author: python273

vk_api/exceptions.py

@@ -108,7 +108,7 @@ def __str__(self):
 
 class Captcha(VkApiError):
 
-    def __init__(self, vk, captcha_sid, func, args=None, kwargs=None, url=None):
+    def __init__(self, vk, captcha_sid, func, args=None, kwargs=None, url=None, redirect_uri=None):
         super(Captcha, self).__init__()
 
         self.vk = vk
@@ -121,6 +121,7 @@ def __init__(self, vk, captcha_sid, func, args=None, kwargs=None, url=None):
 
         self.key = None
         self.url = url
+        self.redirect_uri = redirect_uri
         self.image = None
 
     def get_url(self):
@@ -139,10 +140,11 @@ def get_image(self):
 
         return self.image
 
-    def try_again(self, key=None):
+    def try_again(self, key=None, success_token=None):
         """ Отправить запрос заново с ответом капчи
 
         :param key: ответ капчи
+        :param success_token: токен капчи с redirect_uri
         """
 
         if key:
@@ -152,6 +154,10 @@ def try_again(self, key=None):
                 'captcha_sid': self.sid,
                 'captcha_key': self.key
             })
+        if success_token:
+            self.kwargs.update({
+                'success_token': success_token,
+            })
 
         return self.func(*self.args, **self.kwargs)
 

vk_api/vk_api.py

@@ -942,6 +942,7 @@ def vk_login_method(
         headers: t.Optional[t.Dict[str, t.Any]] = None,
         captcha_sid: t.Optional[str] = None,
         captcha_key: t.Optional[str] = None,
+        success_token: t.Optional[str] = None,
     ) -> t.Dict[str, t.Any]:
         """
         Вызов действия для https://login.vk.ru с обработкой капчи.
@@ -965,6 +966,9 @@ def vk_login_method(
             self.logger.info(f'Using captcha code: {captcha_sid}: {captcha_key}')
             values['captcha_sid'] = captcha_sid
             values['captcha_key'] = captcha_key
+        if success_token:
+            self.logger.info(f'Using captcha success_token: {success_token}')
+            values['success_token'] = success_token
 
         response = self.http.post(
             url=f'https://login.vk.ru/?act={action}',
@@ -981,6 +985,7 @@ def vk_login_method(
                 vk=self,
                 captcha_sid=response_dict['captcha_sid'],
                 url=response_dict['captcha_img'],
+                redirect_uri=response_dict.get('redirect_uri'),
                 func=self.vk_login_method,
                 kwargs={
                     'action': action,