1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
|
##############################################################################
# Copyright (c) 2017 ZTE Coreporation and others.
# feng.xiaowei@zte.com.cn
# All rights reserved. This program and the accompanying materials
# are made available under the terms of the Apache License, Version 2.0
# which accompanies this distribution, and is available at
# http://www.apache.org/licenses/LICENSE-2.0
##############################################################################
from neutronclient.neutron import client
from deploy.common import query
import keystoneauth
class Neutron(keystoneauth.ClientBase):
def __init__(self, version='2', openrc=None):
super(Neutron, self).__init__(client.Client, version, openrc)
def create_network(self, name, body):
if not self.get_network_by_name(name):
return self._create_network(name, body)
else:
print('admin_ext [{}] already exist'.format(name))
return None
def create_subnet(self, body=None):
if not self.get_subnet_by_name(body):
return self._create_subnet(body)
else:
print ('subnet [{}] already exist'.format(body))
return None
def list_networks(self):
return self.client.list_networks()['networks']
def list_subnets(self):
return self.client.list_subnets()['subnets']
def get_network_by_name(self, name):
return query.find(lambda nw: nw['name'] == name, self.list_networks())
def get_subnet_by_name(self, body):
def _same_subnet(this, that):
for item in ['name', 'network_id']:
if this[item] != that[item]:
return False
return True
return query.find(lambda n: _same_subnet(n, body['subnets'][0]),
self.list_subnets())
def _create_network(self, name, body):
try:
nid = self.client.create_network(body=body)['network']['id']
print('_create_admin_ext_net [{}] with id: {}'.format(name, nid))
return nid
except Exception, e:
print('_create_admin_ext_net [{}] fail with: {}'.format(name, e))
return None
def _create_subnet(self, body):
print('_create_subnet with body: {}'.format(body))
try:
snid = self.client.create_subnet(body)['subnets'][0]['id']
print('_create_subnet success with id={}'.format(snid))
return snid
except Exception, e:
print('_create_subnet fail with: {}'.format(e))
return None
def _list_security_groups(self):
return self.client.list_security_groups()['security_groups']
def get_security_group_by_name(self, name):
return query.find(lambda nw: nw['name'] == name, self._list_security_groups())
def _check_security_group_rule_conflict(self, security_group, body):
newrule = body['security_group_rule']
rules = security_group['security_group_rules']
for rule in rules:
is_same = True
for key in newrule.keys():
if key in rule and newrule[key] != rule[key]:
is_same = False
break
if is_same:
print('The rule already exists in the security group %s' % security_group['id'])
return True
return False
def create_security_group_rule(self, security_group, body):
if not self._check_security_group_rule_conflict(security_group, body):
try:
rule = self.client.create_security_group_rule(body=body)
print('create_security_group_rule success with id %s' % rule['security_group_rule']['id'])
except Exception, e:
print('create_security_group_rule fail with exception %s' % e)
|
