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"""