Add fobj.close to CsvLazyDictWriter

turicas · Nov 30, 2018 · bbb2c57 · bbb2c57
1 parent 04557b0
commit bbb2c57
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/rows/utils.py b/rows/utils.py
@@ -582,6 +582,13 @@ def writerow(self, row):
  self.writerow = self.writer.writerow
  return self.writerow(row)
 
+ def __del__(self):
+ self.close()
+
+ def close(self):
+ if self._fobj and not self._fobj.closed:
+ self._fobj.close()
+
 
 def execute_command(command):
  """Execute a command and return its output"""