Merge remote-tracking branch 'gh3/patch-1'

1 files changed, 16 insertions, 0 deletions

diff --git a/src/croniter/croniter.py b/src/croniter/croniter.py
index 8b55b6e..bd7c4ef 100644
--- a/src/croniter/croniter.py
+++ b/src/croniter/croniter.py
@@ -141,6 +141,22 @@ class croniter(object):
             return datetime.datetime.fromtimestamp(self.cur)
         return self.cur
 
+    # iterator protocol, to enable direct use of croniter objects in a loop, like "for dt in croniter('5 0 * * *'): ..."
+    # or for combining multiple croniters into single dates feed using 'itertools' module
+    def __iter__(self): return self
+    __next__ = next = get_next          
+
+    def all_next(self, ret_type=float):
+        "Generator of all consecutive dates. Can be used instead of implicit call to __iter__, whenever non-default 'ret_type' has to be specified."
+        while True: yield self._get_next(ret_type, is_prev=False)
+
+    def all_prev(self, ret_type=float):
+        "Generator of all previous dates."
+        while True: yield self._get_next(ret_type, is_prev=True)
+    
+    iter = all_next             # alias, you can call .iter() instead of .all_next()
+    
+    
     def _get_next(self, ret_type=float, is_prev=False):
         expanded = self.expanded[:]