diff --git a/artifice/csv_mixin.py b/artifice/csv_mixin.py new file mode 100644 index 0000000..e3238a0 --- /dev/null +++ b/artifice/csv_mixin.py @@ -0,0 +1,19 @@ +from csv import writer + + +class CSV_File_mixin(object): + + def write_output(self): + try: + read = open(self.filename) + raise RuntimeError("Can't write to an existing file!") + except IOError: + pass + fh = open(self.filename, "w") + + csvwriter = writer(fh, dialect='excel', delimiter=',') + for line in self.lines: + # Line is expected to be an iterable row + csvwriter.writerow(line) + + fh.close()