update redis config

csst_dag/dfs.py

 import redis
 import toml
 import os
-
+from csst_dfs_client import plan, level0
+from astropy.table import Table
 
 CONFIG = toml.load(os.path.join(os.path.dirname(__file__), "config", "config.toml"))
 
 
 class DFS:
-    def __init__(self, location="naoc"):
-        if location not in CONFIG.keys():
-            print("Available DFS locations: ", list(CONFIG.keys()))
-            raise ValueError(f"Unknown dfs location: {location}")
-        self.location = location
-        self.config = CONFIG[location]
-        print("Setting DFS config:")
-        for k, v in self.config["dfs"].items():
-            if k not in os.environ.keys():
+    def __init__(self, location=None):
+        # try each location
+        print("Test all locations:")
+        status_list = []
+        for loc in CONFIG.keys():
+            # print("Setting DFS config:")
+            for k, v in CONFIG[loc]["dfs"].items():
                 os.environ.setdefault(k, str(v))
-                print(f" - Set {k}: {v}")
-            else:
-                print(f" - {k}: {os.environ[k]}")
+            # print("Setting redis config:")
+            redis = Redis(location=loc)
+            this_dfs_status = plan.find(instrument="").success
+            try:
+                this_redis_status = redis.ping()
+            except BaseException:
+                this_redis_status = False
+            this_status = dict(
+                location=loc,
+                status=this_dfs_status and this_redis_status,
+                dfs=this_dfs_status,
+                redis=this_redis_status,
+            )
+            status_list.append(this_status)
+        status_table = Table(status_list)
+        print(status_table)
 
-        self.redis = Redis(location)
+        if status_table["status"].sum() == 0:
+            raise ValueError("No DFS location is available")
+        elif status_table["status"].sum() > 1:
+            print("Multiple DFS locations are available, please specify one")
+            raise ValueError("Multiple DFS locations are available")
+        else:
+            self.location = status_table["location"][status_table["status"]][0]
+            self.config = CONFIG[self.location]
+            print(f"Using DFS location: {location}")
+            for k, v in CONFIG[loc]["dfs"].items():
+                os.environ.setdefault(k, str(v))
+            # print("Setting redis config:")
+            self.redis = Redis(location=loc)
 
 
 class Redis(redis.Redis):