pythermostat/pythermostat/plot.py

@@ -5,13 +5,15 @@ import matplotlib.animation as animation
 from threading import Thread, Lock
 from pythermostat.client import Client
 
-TIME_WINDOW = 300.0
 
-tec = Client()
-target_temperature = tec.get_pid()[0]['target']
-print("Channel 0 target temperature: {:.3f}".format(target_temperature))
+def main():
+    TIME_WINDOW = 300.0
 
-class Series:
+    tec = Client()
+    target_temperature = tec.get_pid()[0]['target']
+    print("Channel 0 target temperature: {:.3f}".format(target_temperature))
+
+    class Series:
         def __init__(self, conv=lambda x: x):
             self.conv = conv
             self.x_data = []
@@ -28,7 +30,7 @@ class Series:
             self.x_data = self.x_data[drop:]
             self.y_data = self.y_data[drop:]
 
-series = {
+    series = {
         # 'adc': Series(),
         # 'sens': Series(lambda x: x * 0.0001),
         'temperature': Series(),
@@ -41,12 +43,12 @@ series = {
         'tec_i': Series(),
         'tec_u_meas': Series(),
         # 'interval': Series(),
-}
-series_lock = Lock()
+    }
+    series_lock = Lock()
 
-quit = False
+    quit = False
 
-def recv_data(tec):
+    def recv_data(tec):
         global last_packet_time
         while True:
             data = tec.get_report()
@@ -65,16 +67,16 @@ def recv_data(tec):
                 break
             time.sleep(0.05)
 
-thread = Thread(target=recv_data, args=(tec,))
-thread.start()
+    thread = Thread(target=recv_data, args=(tec,))
+    thread.start()
 
-fig, ax = plt.subplots()
+    fig, ax = plt.subplots()
 
-for k, s in series.items():
+    for k, s in series.items():
         s.plot, = ax.plot([], [], label=k)
-legend = ax.legend()
+    legend = ax.legend()
 
-def animate(i):
+    def animate(i):
         min_x, max_x, min_y, max_y = None, None, None, None
         
         series_lock.acquire()
@@ -119,13 +121,17 @@ def animate(i):
             margin_y = 0.01 * (max_y - min_y)
             ax.set_ylim(min_y - margin_y, max_y + margin_y)
 
-    global legend
+        nonlocal legend
         legend.remove()
         legend = ax.legend()
 
-ani = animation.FuncAnimation(
+    ani = animation.FuncAnimation(
         fig, animate, interval=1, blit=False, save_count=50)
 
-plt.show()
-quit = True
-thread.join()
+    plt.show()
+    quit = True
+    thread.join()
+
+
+if __name__ == "__main__":
+    main()