Refactoring code

Author: python273

jconfig/__init__.py

@@ -13,9 +13,4 @@
 __email__ = 'python273@ya.ru'
 __contact__ = 'https://vk.com/python273'
 
-
-import sys
-if sys.version_info[0] == 2:
-    from jconfig import *
-else:
-    from .jconfig import *
+from .jconfig import *

vk_api/__init__.py

@@ -13,13 +13,6 @@
 __email__ = 'python273@ya.ru'
 __contact__ = 'https://vk.com/python273'
 
-import sys
-
-if sys.version_info[0] == 2:
-    from vk_api import *
-    from vk_upload import *
-    from vk_tools import *
-else:
-    from .vk_api import *
-    from .vk_upload import *
-    from .vk_tools import *
+from .vk_api import *
+from .vk_upload import *
+from .vk_tools import *

vk_api/vk_tools.py

@@ -16,16 +16,19 @@
 
 
 class VkTools(object):
+    """ Содержит некоторые воспомогательные функции, которые могут понадобиться
+        при использовании API
+    """
+
+    __slots__ = ('vk',)
+
     def __init__(self, vk):
         """
-
         :param vk: объект VkApi
         """
-
         self.vk = vk
 
-    def get_all(self, method, max_count, values=None, key='items',
-                limit=None):
+    def get_all(self, method, max_count, values=None, key='items', limit=None):
         """ Получить все элементы
         Работает в методах, где в ответе есть count и items или users
         За один запрос получает max_count * 25 элементов
@@ -56,18 +59,18 @@ def get_all(self, method, max_count, values=None, key='items',
             response = self.vk.method('execute', {'code': run_code})
 
             items += response['items']
+            offset = response['offset']
 
-            if response['offset'] >= response['count']:
+            if offset >= response['count']:
                 break
 
             if limit and len(items) >= limit:
                 break
 
-            offset = response['offset']
-
         return {'count': len(items), key: items}
 
-    def get_all_slow(self, method, values=None, max_count=200, key='items'):
+    def get_all_slow(self, method, max_count, values=None, key='items',
+                     limit=None):
         """ Получить все элементы
         Работает в методах, где в ответе есть count и items или users
 
@@ -76,6 +79,8 @@ def get_all_slow(self, method, values=None, max_count=200, key='items'):
         :param max_count: максимальное количество элементов,
                             которое можно получить за один раз
         :param key: ключ элементов, которые нужно получить
+        :param limit: ограничение на кол-во получаемых элементов,
+                            но может прийти больше
         """
 
         if not values:
@@ -97,8 +102,12 @@ def get_all_slow(self, method, values=None, max_count=200, key='items'):
             response = self.vk.method(method, values)
             items += response[key]
 
+            if limit and len(items) >= limit:
+                break
+
         return {'count': len(items), key: items}
 
+
 # Полный код в файле vk_procedures
 code_get_all_items = """
 var m=%s,n=%s,b="%s",v=n;var c={count:m,offset:v}+%s;var r=API.%s(c),k=r.count,

vk_api/vk_upload.py

@@ -139,7 +139,7 @@ def document(self, file_path, title=None, tags=None, group_id=None):
 
 def open_photos(photos_paths):
     if not isinstance(photos_paths, list):
-            photos_paths = [photos_paths]
+        photos_paths = [photos_paths]
 
     photos = []