Adds missing flavor information in the server list long command

josecastroleon · josecastroleon · commit 429b43a331cd · 2017-03-13T16:32:15.000+01:00
Closes-Bug: #1672396

Change-Id: Ie2a664fd1c3db1b8269ea079df181f87afc702a7
diff --git a/openstackclient/compute/v2/server.py b/openstackclient/compute/v2/server.py
@@ -917,6 +917,8 @@ def take_action(self, parsed_args):
                 'Networks',
                 'Image Name',
                 'Image ID',
+                'Flavor Name',
+                'Flavor ID',
                 'OS-EXT-AZ:availability_zone',
                 'OS-EXT-SRV-ATTR:host',
                 'Metadata',
@@ -930,6 +932,8 @@ def take_action(self, parsed_args):
                 'Networks',
                 'Image Name',
                 'Image ID',
+                'Flavor Name',
+                'Flavor ID',
                 'Availability Zone',
                 'Host',
                 'Properties',
@@ -977,8 +981,19 @@ def take_action(self, parsed_args):
         except Exception:
             pass
 
-        # Populate image_name and image_id attributes of server objects
-        # so that we can display "Image Name" and "Image ID" columns.
+        flavors = {}
+        # Create a dict that maps flavor_id to flavor object.
+        # Needed so that we can display the "Flavor Name" column.
+        # "Flavor Name" is not crucial, so we swallow any exceptions.
+        try:
+            flavors_list = compute_client.flavors.list()
+            for i in flavors_list:
+                flavors[i.id] = i
+        except Exception:
+            pass
+
+        # Populate image_name, image_id, flavor_name and flavor_id attributes
+        # of server objects so that we can display those columns.
         for s in data:
             if 'id' in s.image:
                 image = images.get(s.image['id'])
@@ -988,6 +1003,14 @@ def take_action(self, parsed_args):
             else:
                 s.image_name = ''
                 s.image_id = ''
+            if 'id' in s.flavor:
+                flavor = flavors.get(s.flavor['id'])
+                if flavor:
+                    s.flavor_name = flavor.name
+                s.flavor_id = s.flavor['id']
+            else:
+                s.flavor_name = ''
+                s.flavor_id = ''
 
         table = (column_headers,
                  (utils.get_item_properties(
diff --git a/openstackclient/tests/unit/compute/v2/test_server.py b/openstackclient/tests/unit/compute/v2/test_server.py
@@ -957,6 +957,8 @@ class TestServerList(TestServer):
         'Networks',
         'Image Name',
         'Image ID',
+        'Flavor Name',
+        'Flavor ID',
         'Availability Zone',
         'Host',
         'Properties',
@@ -1027,6 +1029,12 @@ def setUp(self):
             for s in self.servers
         ]
 
+        Flavor = collections.namedtuple('Flavor', 'id name')
+        self.flavors_mock.list.return_value = [
+            Flavor(id=s.flavor['id'], name=self.flavor.name)
+            for s in self.servers
+        ]
+
         for s in self.servers:
             self.data.append((
                 s.id,
@@ -1046,6 +1054,8 @@ def setUp(self):
                 server._format_servers_list_networks(s.networks),
                 self.image.name,
                 s.image['id'],
+                self.flavor.name,
+                s.flavor['id'],
                 getattr(s, 'OS-EXT-AZ:availability_zone'),
                 getattr(s, 'OS-EXT-SRV-ATTR:host'),
                 s.Metadata,
