Merge branch 'dev'

Conflicts:
	setup.py
	vk_api/__init__.py

Author: python273

examples/capthca_handle.py

@@ -25,13 +25,13 @@ def main():
     """ Пример: обработка капчи """
 
     login, password = 'python@vk.com', 'mypassword'
-    vk = vk_api.VkApi(
+    vk_session = vk_api.VkApi(
         login, password,
         captcha_handler=captcha_handler  # функция для обработки капчи
     )
 
     try:
-        vk.authorization()
+        vk_session.authorization()
     except vk_api.AuthorizationError as error_msg:
         print(error_msg)
         return

examples/get_full_wall.py

@@ -12,15 +12,15 @@ def main():
     """ Пример: получение всех постов со стены """
 
     login, password = 'python@vk.com', 'mypassword'
-    vk = vk_api.VkApi(login, password)
+    vk_session = vk_api.VkApi(login, password)
 
     try:
-        vk.authorization()
+        vk_session.authorization()
     except vk_api.AuthorizationError as error_msg:
         print(error_msg)
         return
 
-    tools = vk_api.VkTools(vk)
+    tools = vk_api.VkTools(vk_session)
     """
         VkTools.get_all позволяет получить все итемы, например со стены или
         получить все диалоги, или сообщения. При использовании get_all

examples/requests_pool.py

@@ -0,0 +1,54 @@
+# -*- coding: utf-8 -*-
+
+"""
+@author: Kirill Python
+@contact: https://vk.com/python273
+"""
+
+import vk_api
+
+
+def main():
+    """ Пример работы с VkRequestsPool """
+
+    login, password = 'python@vk.com', 'mypassword'
+    vk_session = vk_api.VkApi(login, password)
+
+    try:
+        vk_session.authorization()
+    except vk_api.AuthorizationError as error_msg:
+        print(error_msg)
+        return
+
+    """
+        Позволяет выполнять до 25 запросов к API за один запрос.
+        Работает через метод execute.
+    """
+
+    friends = {}
+
+    with vk_api.VkRequestsPool(vk_session) as pool:
+        for user_id in [1, 183433824]:
+            friends[user_id] = pool.method('friends.get', {'user_id': user_id})
+
+    print(friends)
+
+    # Same result
+    with vk_api.VkRequestsPool(vk_session) as pool:
+        friends = pool.method_one_param(
+            'friends.get', key='user_id', values=[1, 183433824])
+
+    print(friends)
+
+    with vk_api.VkRequestsPool(vk_session) as pool:
+        friends = pool.method('friends.get')
+        status = pool.method('status.get')
+
+    # Обратите внимание, что запрос выполняется после выхода из with
+    # и к результатам обратиться можно только после выхода из него
+
+    print(friends)
+    print(status)
+
+if __name__ == '__main__':
+    main()

examples/simple_example.py

@@ -12,21 +12,23 @@ def main():
     """ Пример получения последнего сообщения со стены """
 
     login, password = 'python@vk.com', 'mypassword'
-    vk = vk_api.VkApi(login, password)
+    vk_session = vk_api.VkApi(login, password)
 
     try:
-        vk.authorization()
+        vk_session.authorization()
     except vk_api.AuthorizationError as error_msg:
         print(error_msg)
         return
 
+    vk = vk_session.get_api()
+
     """
         VkApi.method позволяет выполнять запросы к API. В этом примере
         используется метод wall.get (https://vk.com/dev/wall.get) с параметром
         count = 1, т.е. мы получаем один последний пост со стены текущего
         пользователя.
     """
-    response = vk.method('wall.get', {'count': 1})  # Используем метод wall.get
+    response = vk.wall.get(count=1)  # Используем метод wall.get
 
     if response['items']:
         print(response['items'][0])

examples/upload_photo.py

@@ -12,10 +12,10 @@ def main():
     """ Пример: загрузка фото """
 
     login, password = 'python@vk.com', 'mypassword'
-    vk = vk_api.VkApi(login, password)
+    vk_session = vk_api.VkApi(login, password)
 
     try:
-        vk.authorization()
+        vk_session.authorization()
     except vk_api.AuthorizationError as error_msg:
         print(error_msg)
         return
@@ -24,7 +24,7 @@ def main():
         В VkUpload реализованы методы загрузки файлов в ВК.
         (Не все, если что-то понадобится - могу дописать)
     """
-    upload = vk_api.VkUpload(vk)
+    upload = vk_api.VkUpload(vk_session)
 
     photo = upload.photo(  # Подставьте свои данные
         'D:/downloads/tube.jpg',

setup.py

@@ -14,7 +14,7 @@
 
 setup(
     name='vk_api',
-    version='6.7',
+    version='7.0',
     author='Kirill Python',
     author_email='python273@ya.ru',
     url='https://github.com/python273/vk_api',

vk_api/__init__.py

@@ -9,7 +9,7 @@
 """
 
 __author__ = 'Kirill Python'
-__version__ = '6.7'
+__version__ = '7.0'
 __email__ = 'python273@ya.ru'
 __contact__ = 'https://vk.com/python273'
 

vk_api/vk_api.py

@@ -20,10 +20,12 @@
 CAPTCHA_ERROR_CODE = 14
 NEED_VALIDATION_CODE = 17
 HTTP_ERROR_CODE = -1
+TWOFACTOR_CODE = -2
 
 RE_LOGIN_HASH = re.compile(r'name="lg_h" value="([a-z0-9]+)"')
 RE_CAPTCHAID = re.compile(r'sid=(\d+)')
 RE_NUMBER_HASH = re.compile(r"al_page: '3', hash: '([a-z0-9]+)'")
+RE_AUTH_HASH = re.compile(r"hash: '([a-z_0-9]+)'")
 RE_TOKEN_URL = re.compile(r'location\.href = "(.*?)"\+addr;')
 
 RE_PHONE_PREFIX = re.compile(r'phone_number">(.*?)<')
@@ -33,9 +35,11 @@
 
 class VkApi(object):
     def __init__(self, login=None, password=None, number=None, sec_number=None,
-                 token=None, proxies=None, captcha_handler=None,
+                 token=None,
+                 proxies=None,
+                 auth_handler=None, captcha_handler=None,
                  config_filename='vk_config.json',
-                 api_version='5.35', app_id=2895443, scope=33554431,
+                 api_version='5.44', app_id=2895443, scope=33554431,
                  client_secret=None):
         """
         :param login: Логин ВКонтакте
@@ -51,6 +55,9 @@ def __init__(self, login=None, password=None, number=None, sec_number=None,
         :param proxies: proxy server
                         {'http': 'http://127.0.0.1:8888/',
                         'https': 'https://127.0.0.1:8888/'}
+        :param auth_handler: Функция для обработки двухфакторной аутентификации,
+                              обязана возвращать строку с кодом для
+                              прохождения аутентификации
         :param captcha_handler: Функция для обработки капчи
         :param config_filename: Расположение config файла
 
@@ -88,7 +95,8 @@ def __init__(self, login=None, password=None, number=None, sec_number=None,
         self.error_handlers = {
             NEED_VALIDATION_CODE: self.need_validation_handler,
             CAPTCHA_ERROR_CODE: captcha_handler or self.captcha_handler,
-            TOO_MANY_RPS_CODE: self.too_many_rps_handler
+            TOO_MANY_RPS_CODE: self.too_many_rps_handler,
+            TWOFACTOR_CODE: auth_handler or self.auth_handler
         }
 
     def authorization(self, reauth=False):
@@ -139,6 +147,10 @@ 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)
+
         if 'remixsid' in self.http.cookies:
             remixsid = self.http.cookies['remixsid']
         elif 'remixsid6' in self.http.cookies:  # ipv6?
@@ -162,7 +174,7 @@ def vk_login(self, captcha_sid=None, captcha_key=None):
             captcha = Captcha(self, captcha_sid, self.vk_login)
 
             if self.error_handlers[CAPTCHA_ERROR_CODE]:
-                self.error_handlers[CAPTCHA_ERROR_CODE](captcha)
+                return self.error_handlers[CAPTCHA_ERROR_CODE](captcha)
             else:
                 raise AuthorizationError('Authorization error (capcha)')
         elif 'm=1' in response.url:
@@ -176,6 +188,28 @@ 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):
+        """ Двухфакторная аутентификация
+        :param reponse: запрос, содержащий страницу с приглашением к аутентификации
+        :param code: код, который необходимо ввести для успешной аутентификации
+        """
+        assert code != None, "Empty code doesn't acceptable"
+        assert len(code) == 6, "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)
+                    'hash': auth_hash,
+                    }
+            response = self.http.post(url, values, cookies=response.cookies)
+            if url not in response.url:
+                return response
+        raise TwoFactorError('Incorrect code: %s' % code)
+
     def security_check(self, url=None, response=None):
         if url:
             response = self.http.get(url)
@@ -315,6 +349,12 @@ def too_many_rps_handler(self, error):
         time.sleep(0.5)
         return error.try_method()
 
+    def auth_handler(self):
+        raise AuthorizationError("No handler for two-factor authorization.")
+
+    def get_api(self):
+        return VkApiMethod(self)
+
     def method(self, method, values=None, captcha_sid=None, captcha_key=None):
         """ Использование методов API
 
@@ -387,6 +427,25 @@ def method(self, method, values=None, captcha_sid=None, captcha_key=None):
         return response['response']
 
 
+class VkApiMethod:
+    def __init__(self, vk, method=None):
+        self._vk = vk
+        self._method = method
+
+    def __getattr__(self, method):
+        if self._method:
+            self._method += '.' + method
+            return self
+
+        return VkApiMethod(self._vk, method)
+
+    def __call__(self, *args, **kwargs):
+        return self._vk.method(self._method, kwargs)
+
+    def get_doc(self):
+        doc(self._method)
+
+
 def doc(method=None):
     """ Открывает документацию на метод или список всех методов
 
@@ -444,6 +503,10 @@ class AccountBlocked(AuthorizationError):
     pass
 
 
+class TwoFactorError(AuthorizationError):
+    pass
+
+
 class SecurityCheck(AuthorizationError):
     def __init__(self, phone_prefix=None, phone_postfix=None, response=None):
         self.phone_prefix = phone_prefix