Manage keyboard interrupt on demo

valletw · Dec 11, 2022 · f559d45 · f559d45
1 parent 58f9646
commit f559d45
Showing 1 changed file with 14 additions and 11 deletions.
diff --git a/demo/main.py b/demo/main.py
@@ -27,14 +27,17 @@ def parse_args():
         export = ExportCSV(params.export)
     # Run data dump process.
     with UMmeter(params.tty) as meter:
-        while True:
-            data = meter.get_data()
-            now = datetime.now()
-            if export is not None:
-                export.update(now, data)
-            print(
-                f"[{data['model']}] {now.time()}"
-                f" {data['voltage']:1.04f}V {data['intensity']:1.04f}A"
-                f" {data['power']:1.04f}W {data['resistance']}Ohm"
-            )
-            sleep(params.refresh)
+        try:
+            while True:
+                data = meter.get_data()
+                now = datetime.now()
+                if export is not None:
+                    export.update(now, data)
+                print(
+                    f"[{data['model']}] {now.time()}"
+                    f" {data['voltage']:1.04f}V {data['intensity']:1.04f}A"
+                    f" {data['power']:1.04f}W {data['resistance']}Ohm"
+                )
+                sleep(params.refresh)
+        except KeyboardInterrupt:
+            pass