class FileManager: def __init__(self, filename): #DIRECTORY PATH HERE self.path = os.path.dirname(os.path.abspath(__file__)).replace( '\\', '/') + "/Tables/" self.filename = filename try: os.makedirs(self.path) except FileExistsError: pass print("Saving data in: ", self.path + filename + ".csv") self.file = open(self.path + filename + ".csv", 'w', newline="") self.writer = csv.writer(self.file) self.headers = None self.types = None self.params = None self.rows_per_iter = None def set_values(self, headers, types, params, chunk_size): self.headers = headers self.types = types self.params = params self.chunk_size = chunk_size self.DG = DataGenerator(types, params, chunk_size) def write_headers(self): self.writer.writerow(self.headers) def write(self): self.writer.writerows(self.DG.count()) # PRINT WIP ############# def close_file(self): self.file.close()