From ea10cdd1ed836191e7ad4adca65594527b0b922e Mon Sep 17 00:00:00 2001
From: jaseg <git@jaseg.net>
Date: Sat, 5 May 2018 19:34:01 +0200
Subject: Nice spectrum plot looking as it should, including photodiode
 response compensation.

---
 firmware/spectrum_progress.py | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

(limited to 'firmware/spectrum_progress.py')

diff --git a/firmware/spectrum_progress.py b/firmware/spectrum_progress.py
index a4eb4b2..35a9d65 100644
--- a/firmware/spectrum_progress.py
+++ b/firmware/spectrum_progress.py
@@ -17,7 +17,7 @@ if __name__ == '__main__':
         step, = db.execute(
                 'SELECT MAX(step) FROM measurements WHERE run_id = (SELECT MAX(run_id) FROM runs)'
                 ).fetchone()
-        return int(step)+1
+        return int(step)
 
     def step_gen():
         while True:
@@ -28,5 +28,9 @@ if __name__ == '__main__':
             time.sleep(args.update_delay)
 
     bar = tqdm.tqdm(total=args.max_step)
-    for step in step_gen():
-        bar.update(step - bar.n)
+    while True:
+        try:
+            for step in step_gen():
+                bar.update(step - bar.n)
+        except:
+            time.sleep(args.update_delay)
-- 
cgit