bt2/setup.py.in: improve legibility of our_get_config_vars a bit

[babeltrace.git] / src / bindings / python / bt2 / setup.py.in

diff --git a/src/bindings/python/bt2/setup.py.in b/src/bindings/python/bt2/setup.py.in
index cadefb728b10569ffc1297d85027a582324cc9a3..e256e42ed75a6f8cd6ba849b991925180339e592 100644 (file)
--- a/src/bindings/python/bt2/setup.py.in
+++ b/src/bindings/python/bt2/setup.py.in
@@ -93,21 +93,22 @@ def our_get_config_vars(*args):
     }
 
     if len(args) == 0:
+        # Return a dict with all config vars.
         all_config_vars = original_get_config_vars()
         for name in overridden_config_vars:
             all_config_vars[name] = overridden_config_vars[name]
-        return all_config_vars
-
-    subset_config_vars = []
-    for name in args:
-        if name not in overridden_config_vars:
-            [var] = original_get_config_vars(name)
-            subset_config_vars.append(var)
-            continue
-
-        subset_config_vars.append(overridden_config_vars[name])
 
-    return subset_config_vars
+        return all_config_vars
+    else:
+        # Return a list with the requested config vars.
+        subset_config_vars = []
+        for name in args:
+            if name in overridden_config_vars:
+                subset_config_vars.append(overridden_config_vars[name])
+            else:
+                subset_config_vars.append(original_get_config_vars(name)[0])
+
+        return subset_config_vars
 
 
 sysconfig.get_config_vars = our_get_config_vars