diff --git a/plot_generator/plot_generator.py b/plot_generator/plot_generator.py new file mode 100644 index 0000000..b2e1986 --- /dev/null +++ b/plot_generator/plot_generator.py @@ -0,0 +1,50 @@ +"""This Program is CLI only!""" + +import matplotlib.pyplot as plt +import numpy as np +import csv +import os + +path = input("Path to csv-file to be plotted: ") +date = input("Date & time at which the measurement was taken (approx.): ") +saveit = input("Should the graph be saved? (y/n) ").lower() + +imp = open(path, "r") +reader = csv.reader(imp, delimiter=',') +rohdaten = list(reader) +lenght = len(rohdaten) +x = [] +y = [] +for i in range(lenght): + extract = rohdaten.pop(0) + x.append(float(extract.pop(0))) + y.append(float(extract.pop(0))) + +plt.plot(x, y, color="MAGENTA") +plt.xlabel("Time") +plt.ylabel("Voltage") +titel = f"GC - Biogasanlage {date}" +plt.title(titel) +plt.grid(True) +if saveit == "y": + pos = 0 + for letter in path[::-1]: + if letter == "/": + break + else: + pos += 1 + pos_c = len(path) - pos + ppath = path[:pos_c] + save_path = f"{ppath}graphs/" + try: + os.mkdir(save_path) + except FileExistsError: + pass + plt.savefig(save_path) + os.rename(f"{save_path}/.png", f"{save_path}/GC-{date}.png") + print(f"saved images to {save_path}") +else: + print("didn't save images") +plt.show() + +