Accept data from a generator in arvados.CollectionWriter.write()

diff --git a/sdk/python/arvados.py b/sdk/python/arvados.py
index f51b768d5b1b4d8d8421f075a63da3a982629872..354ef629f8641d27f20935c4b57de0df5808520c 100644 (file)
--- a/sdk/python/arvados.py
+++ b/sdk/python/arvados.py
@@ -713,6 +713,10 @@ class CollectionWriter(object):
         map(lambda x: self.write_directory_tree(*x), todo)
 
     def write(self, newdata):
+        if hasattr(newdata, '__iter__'):
+            for s in newdata:
+                self.write(s)
+            return
         self._data_buffer += [newdata]
         self._data_buffer_len += len(newdata)
         self._current_stream_length += len(newdata)