From b8aa37321d6ac0ebd9f2237c8d2ed6594b614b57 Mon Sep 17 00:00:00 2001
From: wyzula-jan <133381102+wyzula-jan@users.noreply.github.com>
Date: Thu, 31 Aug 2023 13:56:24 +0200
Subject: [PATCH] fix: config.yaml can be passed as a console argument to
 extreme.py

---
 bec_widgets/examples/extreme/config.yaml |  4 ++--
 bec_widgets/examples/extreme/extreme.py  | 22 +++++++++++++++++++---
 2 files changed, 21 insertions(+), 5 deletions(-)

diff --git a/bec_widgets/examples/extreme/config.yaml b/bec_widgets/examples/extreme/config.yaml
index e5274faa..a01184b7 100644
--- a/bec_widgets/examples/extreme/config.yaml
+++ b/bec_widgets/examples/extreme/config.yaml
@@ -1,6 +1,6 @@
-xy_pairs_1: [["samx", ["gauss_bpm", "gauss_adc1"]]]
 xy_pairs: [["samx", ["gauss_bpm", "gauss_adc1"]],
            ["samx", ["gauss_adc1", "gauss_adc2"]],
            ["samx", ["gauss_adc2"]],
+           ["samx", ["gauss_adc1"]],
            ["samx", ["gauss_bpm", "gauss_adc2"]],
-           ["samx", ["gauss_bpm"]]]
+           ["samx", ["gauss_bpm", "gauss_adc1", "gauss_adc2"]]]
diff --git a/bec_widgets/examples/extreme/extreme.py b/bec_widgets/examples/extreme/extreme.py
index ed5ea4f9..df4fcad1 100644
--- a/bec_widgets/examples/extreme/extreme.py
+++ b/bec_widgets/examples/extreme/extreme.py
@@ -283,13 +283,29 @@ class PlotApp(QWidget):
 
 if __name__ == "__main__":
     import yaml
+    import argparse
+
     from bec_widgets import ctrl_c
     from bec_widgets.bec_dispatcher import bec_dispatcher
 
-    with open("config.yaml", "r") as file:
-        config = yaml.safe_load(file)
+    parser = argparse.ArgumentParser(description="Plotting App")
+    parser.add_argument(
+        "--config", "-c", help="Path to the .yaml configuration file", default="config.yaml"
+    )
+    args = parser.parse_args()
 
-    xy_pairs = config["xy_pairs"]
+    try:
+        with open(args.config, "r") as file:
+            config = yaml.safe_load(file)
+            xy_pairs = config.get("xy_pairs", [])
+    except FileNotFoundError:
+        print(f"The file {args.config} was not found.")
+        exit(1)
+    except Exception as e:
+        print(f"An error occurred while loading the config file: {e}")
+        exit(1)
+
+        # TODO PUT RAISE ERROR HERE to check for xy_pairs
 
     # BECclient global variables
     client = bec_dispatcher.client