VkTools.get_all_items rename to VkTools.get_all

Add "key" param in VkTools.get_all

Author: python273

vk_api/vk_tools.py

@@ -6,15 +6,16 @@ class VkTools(object):
     def __init__(self, vk):
         self.vk = vk
 
-    def get_all_items(self, method, values=None, max_count=200):
+    def get_all(self, method, values=None, max_count=200, key='items'):
         ''' Получить все элементы
-            Работает в методах, где в ответе есть count и items
+            Работает в методах, где в ответе есть items или users
             За один запрос получает max_count * 25 элементов
 
         :param method: метод
         :param values: параметры
         :param max_count: максимальное количество элементов,
                             которое можно получить за один раз
+        :param key: ключ элементов, которые нужно получить
         '''
 
         if values:
@@ -26,13 +27,12 @@ def get_all_items(self, method, values=None, max_count=200):
         offset = 0
 
         while True:
-            run_code = code_get_all_items % (max_count, offset, json.dumps(values), method, method)
+            run_code = code_get_all_items % (
+                                        max_count, offset, json.dumps(values),
+                                        key, method, method
+                                    )
 
-            try:
-                response = self.vk.method('execute', {'code': run_code})
-            except Exception as e:
-                print run_code
-                print e
+            response = self.vk.method('execute', {'code': run_code})
 
             items += response['items']
 
@@ -41,16 +41,17 @@ def get_all_items(self, method, values=None, max_count=200):
 
             offset = response['offset']
 
-        return {'count': len(items), 'items': items}
+        return {'count': len(items), key: items}
 
-    def get_all_items_slow(self, method, values=None, max_count=200):
+    def get_all_slow(self, method, values=None, max_count=200, key='items'):
         ''' Получить все элементы
-            Работает в методах, где в ответе есть count и items
+            Работает в методах, где в ответе есть count и items или users
 
         :param method: метод
         :param values: параметры
         :param max_count: максимальное количество элементов,
                             которое можно получить за один раз
+        :param key: ключ элементов, которые нужно получить
         '''
 
         if not values:
@@ -62,17 +63,17 @@ def get_all_items_slow(self, method, values=None, max_count=200):
 
         response = self.vk.method(method, values)
         count = response['count']
-        items = response['items']
+        items = response[key]
 
         for i in xrange(max_count, count + 1, max_count):
             values.update({
                 'offset': i
             })
 
             response = self.vk.method(method, values)
-            items += response['items']
+            items += response[key]
 
-        return {'count': len(items), 'items': items}
+        return {'count': len(items), key: items}
 
 # Полный код в файле vk_procedures
-code_get_all_items = 'var z=%s,x=%s,y=%s,p={"count":z}+y,r=API.%s(p),c=r["count"],j=r["items"],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["items"];i=i+1;};return{"count":c,"items":j,"offset":o,"end":o+z>=c};'
+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};'

vk_procedures

@@ -1,16 +1,17 @@
 ==========
 CODE: get items x25
-INPUT: max_count, start_offset, params:in_JSON, method, method
+INPUT: max_count, start_offset, params:in_JSON, key, 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["items"];
+var items = res[key];
 
 var offset = 0;
 
@@ -21,7 +22,7 @@ while(i < 25 && offset < count) {
     params = params + my_params;
 
     res = API.%s(params);
-    items = items + res["items"];
+    items = items + res[key];
 
     i = i + 1;
 }