mirror of
https://github.com/janishutz/BiogasControllerApp.git
synced 2025-11-25 05:44:23 +00:00
Janis Hutz
7905cb851a
Added a config validator and documented code that was previously undocumented, for the plot_generator scripts, documented them.
54 lines
1.3 KiB
Python
54 lines
1.3 KiB
Python
"""This Program is CLI only!"""
|
|
|
|
import matplotlib.pyplot as plt
|
|
import csv
|
|
import os
|
|
|
|
# Get user input for various data
|
|
path = input("Path to csv-file to be plotted: ")
|
|
print("For the below, it is recommended to enter data in this format: yyyy-mm-dd-hh-mm")
|
|
date = input("Date & time at which the measurement was taken (approx.): ")
|
|
group = input("Group-name: ")
|
|
saveit = input("Should the graph be saved? (y/n) ").lower()
|
|
|
|
imp = open(path, "r")
|
|
reader = csv.reader(imp, delimiter=",")
|
|
data = list(reader)
|
|
x = []
|
|
y = []
|
|
for i in range(len(data)):
|
|
# Extract the data
|
|
extract = data.pop(0)
|
|
x.append(float(extract.pop(0)))
|
|
y.append(float(extract.pop(0)))
|
|
|
|
# Set up plot
|
|
plt.plot(x, y, color="MAGENTA")
|
|
plt.xlabel("Time")
|
|
plt.ylabel("Voltage")
|
|
|
|
plt.title(f"GC - Biogasanlage {date}")
|
|
plt.grid(True)
|
|
|
|
# Check if user wants to save the image
|
|
if saveit == "n":
|
|
print("didn't save images")
|
|
else:
|
|
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(f"{save_path}/GC-{date}-{group}.png")
|
|
|
|
print(f"Saved images to {save_path}")
|
|
plt.show()
|