Fix VkTools

Author: python273

vk_api/utils.py

@@ -8,6 +8,11 @@
 Copyright (C) 2016
 """
 
+try:
+    import simplejson as json
+except ImportError:
+    import json
+
 
 def search_re(reg, string):
     """ Поиск по регулярке """
@@ -42,3 +47,10 @@ def code_from_number(prefix, postfix, number):
         return
 
     return number[prefix_len:-postfix_len]
+
+
+def sjson_dumps(*args, **kwargs):
+    kwargs['ensure_ascii'] = False
+    kwargs['separators'] = (',', ':')
+
+    return json.dumps(*args, **kwargs)

vk_api/vk_requestspool.py

@@ -3,11 +3,7 @@
 import sys
 from collections import namedtuple
 
-try:
-    import simplejson as json
-except ImportError:
-    import json
-
+from .utils import sjson_dumps
 
 if sys.version_info.major == 2:
     range = xrange
@@ -73,7 +69,7 @@ def method(self, method, values=None):
         """ Добавляет запрос в пулл
 
         :param method: метод
-        :type method: basestring
+        :type method: str
 
         :param values: параметры
         :type values: dict
@@ -96,13 +92,13 @@ def method_one_param(self, method, key, values, default_values=None):
         """ Использовать, если изменяется значение только одного параметра
 
         :param method: метод
-        :type method: basestring
+        :type method: str
 
         :param default_values: одинаковые значения для запросов
         :type default_values: dict
 
         :param key: ключ изменяющегося параметра
-        :type key: basestring
+        :type key: str
 
         :param values: список значений изменяющегося параметра (max: 25)
         :type values: list
@@ -178,13 +174,6 @@ def execute_one_param(self):
         self.one_param['return'].set_result(result)
 
 
-def sjson_dumps(*args, **kwargs):
-    kwargs['ensure_ascii'] = False
-    kwargs['separators'] = (',', ':')
-
-    return json.dumps(*args, **kwargs)
-
-
 def check_one_method(pool):
     """ Возвращает True, если все запросы в пулле к одному методу """
 
@@ -233,12 +222,6 @@ def gen_code_many_methods(pool):
 
 
 # Полный код в файле 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,
-j=r[b],i=1;while(i<25&&v+m<=k){v=i*m+n;c.offset=v;j=j+API.%s(c)[b];i=i+1;}
-return {count:k,items:j,offset:v+m};
-""".replace('\n', '')
-
 code_requestspoll_one_method = """
 var p=%s,i=0,r=[];while(i<p.length){r.push(API.%s(p[i]));i=i+1;}return r;
 """.replace('\n', '')

vk_api/vk_tools.py

@@ -10,6 +10,8 @@
 
 import sys
 
+from .utils import sjson_dumps
+
 
 if sys.version_info.major == 2:
     range = xrange
@@ -35,12 +37,21 @@ def get_all_iter(self, method, max_count, values=None, key='items',
         За один запрос получает max_count * 25 элементов
 
         :param method: метод
+        :type method: str
+
         :param values: параметры
+        :type values: dict
+
         :param max_count: максимальное количество элементов,
                             которое можно получить за один раз
+        :type max_count: int
+
         :param key: ключ элементов, которые нужно получить
+        :type key: str
+
         :param limit: ограничение на кол-во получаемых элементов,
                             но может прийти больше
+        :type limit: int
         """
 
         if values:
@@ -125,3 +136,11 @@ def get_all_slow(self, method, max_count, values=None, key='items',
 
         items = list(self.get_all_slow(method, max_count, values, key, limit))
         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,
+j=r[b],i=1;while(i<25&&v+m<=k){v=i*m+n;c.offset=v;j=j+API.%s(c)[b];i=i+1;}
+return {count:k,items:j,offset:v+m};
+""".replace('\n', '')