Prettify dictionary keys

fabric-testbed · Oct 18, 2023 · 2c9d5dc · 2c9d5dc
1 parent 5b460a6
commit 2c9d5dc
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/fabrictestbed_extensions/fablib/resources.py b/fabrictestbed_extensions/fablib/resources.py
@@ -757,6 +757,15 @@ def site_to_dict(
         }
         if not latlon:
             d.pop("location")
+
+        # Prettify the dict's keys so that `fields` and
+        # `filter_function` arguments to list_sites() will work.
+        for k, v in d.items():
+            pretty_key = self.site_pretty_names.get(k)
+            if not pretty_key:
+                continue
+            d[pretty_key] = d.pop(k)
+
         return d
 
     def site_to_dictXXX(self, site):