Merge pull request #96 from xkord/devel

Added date time limit for tools.py get_all method

Author: python273

vk_api/tools.py

@@ -30,7 +30,7 @@ def __init__(self, vk):
         self.vk = vk
 
     def get_all_iter(self, method, max_count, values=None, key='items',
-                     limit=None):
+                     limit=None, stop_fn=None):
         """ Получить все элементы.
         Работает в методах, где в ответе есть count и items или users.
         За один запрос получает max_count * 25 элементов
@@ -51,6 +51,9 @@ def get_all_iter(self, method, max_count, values=None, key='items',
         :param limit: ограничение на кол-во получаемых элементов,
                             но может прийти больше
         :type limit: int
+
+        :param stop_fn: функция, отвечающая за выход из цикла
+        :type stop_fn: func
         """
 
         values = values.copy() if values else {}
@@ -80,7 +83,11 @@ def get_all_iter(self, method, max_count, values=None, key='items',
             if limit and items_count >= limit:
                 break
 
-    def get_all(self, method, max_count, values=None, key='items', limit=None):
+            if stop_fn and stop_fn(items):
+                break
+
+    def get_all(self, method, max_count, values=None, key='items', limit=None,
+                stop_fn=None):
         """ Использовать только если нужно загрузить все объекты в память.
 
             Eсли вы можете обрабатывать объекты по частям, то лучше
@@ -90,12 +97,13 @@ def get_all(self, method, max_count, values=None, key='items', limit=None):
             все данные в память
         """
 
-        items = list(self.get_all_iter(method, max_count, values, key, limit))
+        items = list(self.get_all_iter(method, max_count, values, key, limit,
+                                       stop_fn))
 
         return {'count': len(items), key: items}
 
     def get_all_slow_iter(self, method, max_count, values=None, key='items',
-                          limit=None):
+                          limit=None, stop_fn=None):
         """ Получить все элементы (без использования execute)
         Работает в методах, где в ответе есть count и items или users
 
@@ -115,6 +123,9 @@ def get_all_slow_iter(self, method, max_count, values=None, key='items',
         :param limit: ограничение на кол-во получаемых элементов,
                             но может прийти больше
         :type limit: int
+
+        :param stop_fn: функция, отвечающая за выход из цикла
+        :type stop_fn: func
         """
 
         values = values.copy() if values else {}
@@ -138,8 +149,11 @@ def get_all_slow_iter(self, method, max_count, values=None, key='items',
             if limit and items_count >= limit:
                 break
 
+            if stop_fn and stop_fn(items):
+                break
+
     def get_all_slow(self, method, max_count, values=None, key='items',
-                     limit=None):
+                     limit=None, stop_fn=None):
         """ Использовать только если нужно загрузить все объекты в память.
 
             Eсли вы можете обрабатывать объекты по частям, то лучше
@@ -150,7 +164,8 @@ def get_all_slow(self, method, max_count, values=None, key='items',
         """
 
         items = list(
-            self.get_all_slow_iter(method, max_count, values, key, limit)
+            self.get_all_slow_iter(method, max_count, values, key, limit, 
+                                   stop_fn)
         )
         return {'count': len(items), key: items}