Skip to content

Commit 9874758

Browse files
authored
Merge pull request #27 from sunsingerus/master
csv writer detructor
2 parents ba4be81 + a1d3eb2 commit 9874758

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

src/writer/csvwriter.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,9 @@ def __init__(
4444
self.path = self.path_prefix + '_' + '_'.join(self.path_suffix_parts) + '.csv'
4545
self.delete = not csv_keep_file
4646

47+
def __del__(self):
48+
self.destroy()
49+
4750
def opened(self):
4851
return bool(self.file)
4952

@@ -110,7 +113,8 @@ def close(self):
110113
self.writer = None
111114

112115
def destroy(self):
113-
if self.delete:
116+
if self.delete and os.path.isfile(self.path):
117+
self.close()
114118
os.remove(self.path)
115119

116120
if __name__ == '__main__':

0 commit comments

Comments
 (0)