summaryrefslogtreecommitdiff
path: root/savannaclient/api/cluster_templates.py
diff options
context:
space:
mode:
Diffstat (limited to 'savannaclient/api/cluster_templates.py')
-rw-r--r--savannaclient/api/cluster_templates.py8
1 files changed, 5 insertions, 3 deletions
diff --git a/savannaclient/api/cluster_templates.py b/savannaclient/api/cluster_templates.py
index fef2cc9..c18de32 100644
--- a/savannaclient/api/cluster_templates.py
+++ b/savannaclient/api/cluster_templates.py
@@ -26,18 +26,20 @@ class ClusterTemplateManager(base.ResourceManager):
resource_class = ClusterTemplate
def create(self, name, plugin_name, hadoop_version, description,
- cluster_configs, node_groups, anti_affinity):
- # expecting node groups to be array of dictionaries
+ cluster_configs, node_groups, anti_affinity, net_id=None):
data = {
'name': name,
'plugin_name': plugin_name,
'hadoop_version': hadoop_version,
'description': description,
'cluster_configs': cluster_configs,
- 'node_groups': node_groups,
+ 'node_groups': [ng.as_dict() for ng in node_groups],
'anti_affinity': anti_affinity
}
+ if net_id:
+ data.update({'neutron_management_network': net_id})
+
return self._create('/cluster-templates', data, 'cluster_template')
def list(self):