Merge pull request #24 from AlexTalker/fix-asserts-two-factor

Fixes and improvements in two-factor authentication

Author: python273

vk_api/vk_api.py

@@ -56,8 +56,9 @@ def __init__(self, login=None, password=None, number=None, sec_number=None,
                         {'http': 'http://127.0.0.1:8888/',
                         'https': 'https://127.0.0.1:8888/'}
         :param auth_handler: Функция для обработки двухфакторной аутентификации,
-                              обязана возвращать строку с кодом для
-                              прохождения аутентификации
+                              обязана возвращать строку с кодом и булевое значение,
+                              означающее, стоит ли вк запомнить это устройство, для
+                              прохождения аутентификации.
         :param captcha_handler: Функция для обработки капчи
         :param config_filename: Расположение config файла
 
@@ -148,8 +149,8 @@ def vk_login(self, captcha_sid=None, captcha_key=None):
         remixsid = None
 
         if 'act=authcheck' in response.url:
-            code = self.error_handlers[TWOFACTOR_CODE]()
-            response = self.twofactor(response, code)
+            code, remember_device = self.error_handlers[TWOFACTOR_CODE]()
+            response = self.twofactor(response, code, remember_device)
 
         if 'remixsid' in self.http.cookies:
             remixsid = self.http.cookies['remixsid']
@@ -188,21 +189,27 @@ def vk_login(self, captcha_sid=None, captcha_key=None):
         if 'act=blocked' in response.url:
             raise AccountBlocked('Account is blocked')
 
-    def twofactor(self, response, code):
+    def twofactor(self, response, code, remember_device=False):
         """ Двухфакторная аутентификация
         :param reponse: запрос, содержащий страницу с приглашением к аутентификации
         :param code: код, который необходимо ввести для успешной аутентификации
+        :param remember_device: параметр, означающий,
+               стоит ли запоминать это устройство в целях
+               избежания повторного ввода кода(default: False)
         """
-        assert code != None, "Empty code doesn't acceptable"
-        assert len(code) == 6, "Length of code cannot be other than 6."
+
+        if code == None:
+            raise TwoFactorError("Empty code doesn't acceptable")
+        if len(code) != 6:
+            raise TwoFactorError("Length of code cannot be other than 6.")
 
         auth_hash = search_re(RE_AUTH_HASH, response.text)
         url = 'https://vk.com/al_login.php'
         if auth_hash:
             values = {
                     'act': 'a_authcheck_code',
                     'code': code,
-                    'remember': 0, # TODO: Fix me(device remembering)
+                    'remember': int(remember_device),
                     'hash': auth_hash,
                     }
             response = self.http.post(url, values, cookies=response.cookies)