vk_api.VkTools fix

Author: python273

vk_api/vk_tools.py

@@ -24,11 +24,11 @@ def __init__(self, vk):
 
         self.vk = vk
 
-    def get_all(self, method, values=None, max_count=200, key='items',
+    def get_all(self, method, max_count, values=None, key='items',
                 limit_count=None):
         """ Получить все элементы
-            Работает в методах, где в ответе есть items или users
-            За один запрос получает max_count * 25 элементов
+        Работает в методах, где в ответе есть count и items или users
+        За один запрос получает max_count * 25 элементов
 
         :param method: метод
         :param values: параметры
@@ -48,15 +48,18 @@ def get_all(self, method, values=None, max_count=200, key='items',
         offset = 0
 
         while True:
-            run_code = code_get_all_items % (max_count, offset,
-                                             json.dumps(values), key,
+            run_code = code_get_all_items % (max_count, offset, key,
+                                             json.dumps(values),
                                              method, method)
 
             response = self.vk.method('execute', {'code': run_code})
 
             items += response['items']
 
-            if response['end'] or (limit_count and len(items) >= limit_count):
+            if response['offset'] >= response['count']:
+                break
+
+            if limit_count and len(items) >= limit_count:
                 break
 
             offset = response['offset']
@@ -65,7 +68,7 @@ def get_all(self, method, values=None, max_count=200, key='items',
 
     def get_all_slow(self, method, values=None, max_count=200, key='items'):
         """ Получить все элементы
-            Работает в методах, где в ответе есть count и items или users
+        Работает в методах, где в ответе есть count и items или users
 
         :param method: метод
         :param values: параметры
@@ -97,7 +100,7 @@ def get_all_slow(self, method, values=None, max_count=200, key='items'):
 
 # Полный код в файле vk_procedures
 code_get_all_items = """
-var z=%s,x=%s,y=%s,k="%s",p={"count":z}+y,r=API.%s(p),c=r["count"],j=r[k],o=0,
-i=1;while(i<25&&o<c){o=i*z+x;p={"count":z,"offset":o}+y;r=API.%s(p);j=j+r[k];i
-=i+1;};return{"count":c,"items":j,"offset":o,"end":o+z>=c};
+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', '')

vk_procedures

@@ -1,31 +1,27 @@
 ==========
 CODE: get items x25
-INPUT: max_count, start_offset, params:in_JSON, key, method, method
+INPUT: MAX_COUNT, START_OFFSET, KEY, MY_PARAMS:json, METHOD, METHOD
 ----------
-var max_count = %s;
-var start_offset = %s;
-var my_params = %s;
-var key = %s;
-
-var params = {"count": max_count} + my_params;
-
-var res = API.%s(params);
-var count = res["count"];
-var items = res[key];
-
-var offset = 0;
-
-var i = 1;
-while(i < 25 && offset < count) {
+var max_count = %s,
+    start_offset = %s,
+    key = "%s",
+    offset = start_offset;
+ 
+var params = {count: max_count, offset: offset} + %s;
+ 
+var res = API.%s(params),
+    count = res.count,
+    items = res[key],
+    i = 1;
+ 
+while(i < 25 && offset + max_count <= count) {
     offset = i * max_count + start_offset;
-    params = {"count": max_count, "offset": offset};
-    params = params + my_params;
-
-    res = API.%s(params);
-    items = items + res[key];
-
+    params.offset = offset;
+ 
+    items = items + API.%s(params)[key];
+ 
     i = i + 1;
 }
-
-return {"count": count, "items": items, "offset": offset, "end": offset >= count};
+ 
+return {count: count, items: items, offset: offset + max_count};
 ==========