Add flavor argument to router

This allows the router to take a flavor argument that can
be either the ID or the name of a flavor to use when creating
a router.

Partially-Implements: blueprint multi-l3-backends
Change-Id: I100fc75de6d41900f6452f356f0b7b741cd177ce

1 files changed, 7 insertions, 0 deletions

diff --git a/neutronclient/neutron/v2_0/router.py b/neutronclient/neutron/v2_0/router.py
index 200feee..b431daa 100644
--- a/neutronclient/neutron/v2_0/router.py
+++ b/neutronclient/neutron/v2_0/router.py
@@ -71,6 +71,9 @@ class CreateRouter(neutronV20.CreateCommand):
         parser.add_argument(
             '--description',
             help=_('Description of router.'))
+        parser.add_argument(
+            '--flavor',
+            help=_('ID or name of flavor.'))
         utils.add_boolean_argument(
             parser, '--distributed', dest='distributed',
             help=_('Create a distributed router.'))
@@ -82,6 +85,10 @@ class CreateRouter(neutronV20.CreateCommand):
 
     def args2body(self, parsed_args):
         body = {'admin_state_up': parsed_args.admin_state}
+        if parsed_args.flavor:
+            _flavor_id = neutronV20.find_resourceid_by_name_or_id(
+                self.get_client(), 'flavor', parsed_args.flavor)
+            body['flavor_id'] = _flavor_id
         neutronV20.update_dict(parsed_args, body,
                                ['name', 'tenant_id', 'distributed', 'ha',
                                 'description'])