More refactoring

Author: python273

setup.py

@@ -3,7 +3,7 @@
 from distutils.core import setup
 
 """
-@author: Kirill Python
+@author: python273
 @contact: https://vk.com/python273
 @license Apache License, Version 2.0, see LICENSE file
 

vk_api/__init__.py

@@ -6,7 +6,7 @@
 from .vk_api import VkApi
 
 """
-@author: Kirill Python
+@author: python273
 @contact: https://vk.com/python273
 @license Apache License, Version 2.0, see LICENSE file
 

vk_api/audio.py

@@ -10,11 +10,14 @@
 
 
 class VkAudio:
+
+    __slots__ = ('_vk',)
+
     def __init__(self, vk):
         self._vk = vk
 
     def get(self, owner_id, offset=0):
-        """ Получение списка аудиозаписей пользователя
+        """ Получить список аудиозаписей пользователя
 
         :param owner_id: ID владельца (отрицательные значения для групп)
         :param offset: смещение
@@ -38,7 +41,7 @@ def get(self, owner_id, offset=0):
         return scrap_data(response.text)
 
     def search_user(self, owner_id, q=''):
-        """ Поиск по аудиозаписям пользователя
+        """ Искать по аудиозаписям пользователя
 
         :param owner_id: ID владельца (отрицательные значения для групп)
         :param q: запрос
@@ -66,7 +69,7 @@ def search_user(self, owner_id, q=''):
         ]
 
     def search(self, q='', offset=0):
-        """ Поиск аудиозаписей
+        """ Искать аудиозаписи
 
         :param q: запрос
         :param offset: смещение
@@ -85,7 +88,7 @@ def search(self, q='', offset=0):
 
 
 def scrap_data(html):
-    """ Парсинг списка аудиозаписей """
+    """ Парсинг списка аудиозаписей из html странцы """
 
     soup = BeautifulSoup(html, 'html.parser')
     tracks = []

vk_api/exceptions.py

@@ -1,6 +1,6 @@
 # -*- coding: utf-8 -*-
 """
-@author: Kirill Python
+@author: python273
 @contact: https://vk.com/python273
 @license Apache License, Version 2.0, see LICENSE file
 
@@ -23,10 +23,6 @@ class AuthError(VkApiError):
     pass
 
 
-class AuthorizationError(AuthError):  # todo: delete
-    pass
-
-
 class LoginRequired(AuthError):
     pass
 
@@ -48,6 +44,7 @@ class TwoFactorError(AuthError):
 
 
 class SecurityCheck(AuthError):
+
     def __init__(self, phone_prefix=None, phone_postfix=None, response=None):
         self.phone_prefix = phone_prefix
         self.phone_postfix = phone_postfix
@@ -59,11 +56,12 @@ def __str__(self):
                 self.phone_prefix, self.phone_postfix
             )
         else:
-            return ('Security check. Phone prefix and postfix not detected. '
-                    'Please send bugreport. Response in self.response')
+            return ('Security check. Phone prefix and postfix are not detected.'
+                    ' Please send bugreport (response in self.response)')
 
 
 class ApiError(VkApiError):
+
     def __init__(self, vk, method, values, error):
         self.vk = vk
         self.method = method
@@ -72,9 +70,7 @@ def __init__(self, vk, method, values, error):
         self.error = error
 
     def try_method(self):
-        """ Пробует отправить запрос заново
-
-        """
+        """ Отправить запрос заново """
 
         return self.vk.method(self.method, self.values)
 
@@ -84,16 +80,15 @@ def __str__(self):
 
 
 class ApiHttpError(VkApiError):
+
     def __init__(self, vk, method, values, response):
         self.vk = vk
         self.method = method
         self.values = values
         self.response = response
 
     def try_method(self):
-        """ Пробует отправить запрос заново
-
-        """
+        """ Отправить запрос заново """
 
         return self.vk.method(self.method, self.values)
 
@@ -102,6 +97,7 @@ def __str__(self):
 
 
 class Captcha(VkApiError):
+
     def __init__(self, vk, captcha_sid, func, args=None, kwargs=None, url=None):
         self.vk = vk
         self.sid = captcha_sid
@@ -116,28 +112,25 @@ def __init__(self, vk, captcha_sid, func, args=None, kwargs=None, url=None):
         self.image = None
 
     def get_url(self):
-        """ Возвращает ссылку на изображение капчи
-
-        """
+        """ Получить ссылку на изображение капчи """
 
         if not self.url:
             self.url = 'https://api.vk.com/captcha.php?sid={}'.format(self.sid)
 
         return self.url
 
     def get_image(self):
-        """ Возвращает бинарное изображение капчи, получаемое по get_url()
-        """
+        """ Получить изображение капчи (jpg) """
 
         if not self.image:
             self.image = self.vk.http.get(self.get_url()).content
 
         return self.image
 
-    def try_again(self, key):
-        """ Отправляет запрос заново с ответом капчи
+    def try_again(self, key=None):
+        """ Отправить запрос заново с ответом капчи
 
-        :param key: текст капчи
+        :param key: ответ капчи
         """
 
         if key:

vk_api/execute.py

@@ -12,12 +12,16 @@
 
 
 class VkFunction(object):
+
+    __slots__ = ('code', '_minified_code', 'args', 'clean_args', 'return_raw')
+
     def __init__(self, code, args=None, clean_args=None, return_raw=False):
-        """ Wrapper around execute method
+        """ Обертка над методом execute
 
-        :param code: function code
-        :param args: tuple with args (will be converted via json.dumps)
-        :param clean_args: tuple with args (will be converted via str)
+        :param code: код функции (VKScript)
+        :param args: список аргументов (будут конвертированы в JSON)
+        :param clean_args: список raw аргументов (будут вставлены как строки)
+        :param return_raw: аргумент raw функции VkApi.method
         """
 
         self.code = code
@@ -42,7 +46,7 @@ def compile(self, args):
     def __call__(self, vk, *args, **kwargs):
 
         if not isinstance(vk, (VkApi, VkApiMethod)):
-            raise VkFunctionException(
+            raise TypeError(
                 'The first arg should be VkApi or VkApiMethod instance'
             )
 

vk_api/longpoll.py

@@ -1,6 +1,6 @@
 # -*- coding: utf-8 -*-
 """
-@author: Kirill Python
+@author: python273
 @contact: https://vk.com/python273
 @license Apache License, Version 2.0
 
@@ -123,6 +123,7 @@ def get_all_event_attrs():
 
 
 class VkLongPoll(object):
+
     __slots__ = (
         'vk', 'wait', 'use_ssl', 'mode',
         'url', 'session',
@@ -213,6 +214,7 @@ def listen(self):
 
 
 class Event(object):
+
     __slots__ = (
         'raw', 'type', 'message_flags', 'platform', 'offline_type',
         'user_id', 'group_id',

vk_api/requests_pool.py

@@ -16,37 +16,51 @@
 if sys.version_info.major == 2:
     range = xrange
 
+
+class VkRequestsPoolException(Exception):
+    pass
+
+
 PoolRequest = namedtuple('PoolRequest', ['method', 'values', 'result'])
 
 
 class RequestResult(object):
+
+    __slots__ = ('_result', 'ready', '_error')
+
     def __init__(self):
         self._result = None
         self.ready = False
-        self.error = False
+        self._error = False
 
-    def set_result(self, result):
-        self._result = result
-        self.ready = True
+    @property
+    def error(self):
+        return self._error
 
-    def set_error(self, error):
-        self.error = error
+    @error.setter
+    def error(self, value):
+        self._error = value
         self.ready = True
 
-    @property
-    def ok(self):
-        return not self.error
-
     @property
     def result(self):
         if not self.ready:
-            raise Exception('Result is not ready')
+            raise RuntimeError('Result is not available in `with` context')
 
-        if self.error:
-            raise Exception('Got error while executing request')
+        if self._error:
+            raise VkRequestsPoolException('Got error while executing request')
 
         return self._result
 
+    @result.setter
+    def result(self, result):
+        self._result = result
+        self.ready = True
+
+    @property
+    def ok(self):
+        return self.ready and not self._error
+
 
 class VkRequestsPool(object):
     """ Позволяет сделать несколько обращений к API за один запрос
@@ -86,7 +100,7 @@ def method(self, method, values=None):
         """
 
         if self.one_param:
-            raise Exception('One param mode is not working with self.method')
+            raise RuntimeError('One param mode does not work with self.method')
 
         if values is None:
             values = {}
@@ -115,7 +129,7 @@ def method_one_param(self, method, key, values, default_values=None):
         """
 
         if not self.one_param and self.pool:
-            raise Exception('One param mode is not working with self.method')
+            raise RuntimeError('One param mode does not work with self.method')
 
         if default_values is None:
             default_values = {}
@@ -150,11 +164,11 @@ def execute(self):
             if response_errors:
                 self.execute_errors += response_errors[:-1]
 
-            for x, r in enumerate(response):
-                if r is not False:
-                    cur_pool[x].result.set_result(r)
+            for x, response in enumerate(response):
+                if response is not False:
+                    cur_pool[x].result.result = response
                 else:
-                    cur_pool[x].result.set_error(True)
+                    cur_pool[x].result.error = True
 
     def execute_one_param(self):
         result = {}
@@ -178,7 +192,7 @@ def execute_one_param(self):
                 if r is not False:
                     result[cur_pool[x]] = r
 
-        self.one_param['return'].set_result(result)
+        self.one_param['return'].result = result
 
 
 def check_one_method(pool):

vk_api/upload.py

@@ -9,14 +9,17 @@
 
 
 class VkUpload(object):
+    """ Загрузка файлов через API (https://vk.com/dev/upload_files) """
+
+    __slots__ = ('vk',)
+
     def __init__(self, vk):
         """
 
         :param vk: объект VkApi
         """
 
         self.vk = vk
-        # https://vk.com/dev/upload_files
 
     def photo(self, photos, album_id,
               latitude=None, longitude=None, caption=None, description=None,

vk_api/vk_api.py

@@ -1,6 +1,6 @@
 # -*- coding: utf-8 -*-
 """
-@author: Kirill Python
+@author: python273
 @contact: https://vk.com/python273
 @license Apache License, Version 2.0, see LICENSE file
 
@@ -274,8 +274,6 @@ def vk_login(self, captcha_sid=None, captcha_key=None):
             self.storage.cookies = cookies_to_list(self.http.cookies)
             self.storage.save()
         else:
-            self.logger.info('Unknown auth error')
-
             raise AuthError(
                 'Unknown error. Please send bugreport: https://vk.com/python273'
             )
@@ -586,16 +584,18 @@ def method(self, method, values=None, captcha_sid=None, captcha_key=None,
 
 
 class VkApiMethod(object):
+
+    __slots__ = ('_vk', '_method')
+
     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)
+        return VkApiMethod(
+            self._vk,
+            (self._method + '.' if self._method else '') + method
+        )
 
     def __call__(self, **kwargs):
         return self._vk.method(self._method, kwargs)