feat: add datacenter to deployment info

ai4os · Jan 25, 2024 · 7773a02 · 7773a02
1 parent 3d11fdb
commit 7773a02
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/ai4papi/nomad/common.py b/ai4papi/nomad/common.py
@@ -109,6 +109,7 @@ def get_deployment(
         'active_endpoints': None,
         'main_endpoint': None,
         'alloc_ID': None,
+        'datacenter': None,
     }
 
     # Retrieve tasks
@@ -212,9 +213,12 @@ def get_deployment(
 
         a = Nomad.allocation.get_allocation(allocs[idx]['ID'])
 
-        # Add ID and status
+        # Add ID
         info['alloc_ID'] = a['ID']
 
+        # Add datacenter
+        info['datacenter'] = Nomad.node.get_node(a['NodeID'])['Datacenter']
+
         # Replace Nomad status with a more user-friendly status
         if a['ClientStatus'] == 'pending':
             info['status'] = 'starting'