Continue IRI API implementation

Author: EZoni

dashboard/iriapi_manager.py

@@ -1,15 +1,60 @@
+from amsc_client import Client
 from trame.widgets import vuetify3 as vuetify
 
+from error_manager import add_error
+from state_manager import state
+
+
+def update_iriapi_info():
+    print("Updating AmSC IRI API info...")
+    try:
+        # create an authenticated client
+        client = Client(token=state.iriapi_key)
+        # update Perlmutter info
+        nersc = client.facility("nersc")
+        perlmutter = nersc.resource("compute")
+        state.perlmutter_description = f"{perlmutter.description}"
+        state.perlmutter_status = f"{perlmutter.status}"
+        print(
+            f"Perlmutter status is {state.perlmutter_status} with description '{state.perlmutter_description}'"
+        )
+    except Exception as e:
+        print(f"An unexpected error occurred when connecting to AmSC IRI API:\n{e}")
+        # reset key expiration date
+        state.iriapi_key_expiration = "Unavailable"
+        # reset Perlmutter status
+        state.perlmutter_description = "Unavailable"
+        state.perlmutter_status = "unavailable"
+        title = "Unable to connect to NERSC"
+        msg = f"Error occurred when connecting to NERSC through the AmSC IRI API: {e}"
+        add_error(title, msg)
+        print(msg)
+
+
+@state.change("iriapi_key_dict")
+def load_iriapi_credentials(**kwargs):
+    # skip if triggered on server ready (all state variables marked as modified)
+    if len(state.modified_keys) == 1:
+        # return if no key file has been uploaded (redundant)
+        if state.iriapi_key_dict is not None:
+            print("Loading AmSC IRI API credentials...")
+            # store the whole content of the key file in a string
+            token_str = state.iriapi_key_dict["content"].decode("utf-8")
+            # store the client ID and key in the respective state variables
+            state.iriapi_key = token_str
+            # update AmSC IRI API info
+            update_iriapi_info()
+
 
 def load_iriapi_card():
     print("Setting AmSC IRI API card...")
     # row with component to upload input file with top padding
     with vuetify.VRow(style="padding-top: 20px;"):
         with vuetify.VCol():
             vuetify.VFileInput(
-                v_model=("sfapi_key_dict",),
+                v_model=("iriapi_key_dict",),
                 label="Token File",
-                accept=".pem",
+                accept=".txt",
                 prepend_icon="",
                 prepend_inner_icon="mdi-paperclip",
                 __properties=["accept"],
@@ -18,7 +63,7 @@ def load_iriapi_card():
     with vuetify.VRow():
         with vuetify.VCol():
             vuetify.VTextField(
-                v_model=("sfapi_key_expiration",),
+                v_model=("iriapi_key_expiration",),
                 label="Token Expiration (if expired or unavailable, please upload a valid token)",
                 readonly=True,
             )

dashboard/state_manager.py

@@ -39,11 +39,16 @@ def initialize_state():
     state.opacity = 0.05
     # HPC connection
     state.hpc_connection = "local"
+    # IRI API
+    state.iriapi_key = None
+    state.iriapi_key_dict = None
+    state.iriapi_key_expiration = "Unavailable"
     # Superfacility API
     state.sfapi_client_id = None
     state.sfapi_key = None
     state.sfapi_key_dict = None
     state.sfapi_key_expiration = "Unavailable"
+    # NERSC
     state.perlmutter_description = "Unavailable"
     state.perlmutter_status = "unavailable"
     # Simulation plots in interactive dialog