Skip to content
Snippets Groups Projects
Commit fafda4b6 authored by Daniel Hillier's avatar Daniel Hillier
Browse files

Check writer context manager was successfully created before closing

parent 51c7ec9a
No related branches found
No related tags found
1 merge request!3And incremental support to the witer
......@@ -226,6 +226,7 @@
self._file = output_file
self._close = close
self.encoding = encoding
self.writer_cm = None
def __enter__(self):
self.writer_cm = elementtreemod._get_writer(self._file, encoding=self.encoding)
......@@ -233,6 +234,7 @@
return _IncrementalFileWriter(writer)
def __exit__(self, type, value, traceback):
self.writer_cm.__exit__(type, value, traceback)
if self.writer_cm:
self.writer_cm.__exit__(type, value, traceback)
if self._close:
self._file.close()
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment