From 12175c23a7b6c2e9cc689bde1bbfd46bd559cf37 Mon Sep 17 00:00:00 2001
From: Ethan Dalool <ethan@voussoir.net>
Date: Fri, 9 Oct 2020 13:09:51 -0700
Subject: [PATCH] Add lazychain.py.

---
 voussoirkit/lazychain.py | 35 +++++++++++++++++++++++++++++++++++
 1 file changed, 35 insertions(+)
 create mode 100644 voussoirkit/lazychain.py

diff --git a/voussoirkit/lazychain.py b/voussoirkit/lazychain.py
new file mode 100644
index 0000000..adc0034
--- /dev/null
+++ b/voussoirkit/lazychain.py
@@ -0,0 +1,35 @@
+import collections
+
+class LazyChain:
+    '''
+    You may be familiar with itertools.chain, which chains two iterables into
+    one. However, I wanted a data structure where I could add more and more
+    generators into the chain without repeatedly calling
+    `chain = itertools.chain(chain, more)`.
+    '''
+    def __init__(self):
+        self.iters = collections.deque()
+
+    def __iter__(self):
+        return self
+
+    def __next__(self):
+        while self.iters:
+            try:
+                return next(self.iters[0])
+            except StopIteration:
+                self.iters.popleft()
+        raise StopIteration()
+
+    def append(self, item):
+        '''
+        Add a single item to the chain.
+        '''
+        self.iters.append(iter((item,)))
+
+    def extend(self, sequence):
+        '''
+        Add the contents of a list, tuple, generator... to the chain.
+        Make sure not to exhaust the generator outside of this chain!
+        '''
+        self.iters.append(iter(sequence))