AppendLineParser: Strip newlines in __iter__.

This commit is contained in:
Florian Bruhin 2015-03-07 00:26:12 +01:00
parent 2010e8115b
commit 181426b50a

View File

@ -117,7 +117,8 @@ class AppendLineParser(BaseLineParser):
def __iter__(self):
if self._fileobj is None:
raise ValueError("Iterating without open() being called!")
return itertools.chain(iter(self._fileobj), iter(self.new_data))
file_iter = (line.rstrip('\n') for line in self._fileobj)
return itertools.chain(file_iter, iter(self.new_data))
@contextlib.contextmanager
def open(self):