#49 XmlAssembler should set end_of_doc when flushing a new doc

geopython · Aug 7, 2016 · f121329 · f121329
1 parent 16e9646
commit f121329
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/stetl/filters/xmlassembler.py b/stetl/filters/xmlassembler.py
@@ -50,15 +50,15 @@ def consume_element(self, packet):
         # Always move the data (element) from packet
         element = packet.consume()
 
-        if element is None or packet.is_end_of_stream() is True:
-            return packet
-
-        self.total_element_count += 1
+        if element:
+            self.total_element_count += 1
+            self.element_arr.append(element)
 
-        self.element_arr.append(element)
         return packet
 
     def flush_elements(self, packet):
+        packet.set_end_of_doc()
+
         if len(self.element_arr) == 0:
             return packet