X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=sfa%2Fopenstack%2Fsecurity_group.py;h=6aced8c68802e22d65f4aae3fa3ffa97ca891c1c;hb=a0ef6e7c91c3c8fa376943d28fec8c5c204a78cd;hp=842b889662a66a684b1da8caa48254f97ebeea9e;hpb=9a249363c9d2e955c192c361bd4a1839c5652fca;p=sfa.git diff --git a/sfa/openstack/security_group.py b/sfa/openstack/security_group.py index 842b8896..6aced8c6 100644 --- a/sfa/openstack/security_group.py +++ b/sfa/openstack/security_group.py @@ -9,13 +9,15 @@ class SecurityGroup: def create_security_group(self, name): try: self.client.security_groups.create(name=name, description=name) - except Exception, ex: + except Exception as ex: logger.log_exc("Failed to add security group") + raise def delete_security_group(self, name): try: - self.client.security_groups.delete(name) - except Exception, ex: + security_group = self.client.security_groups.find(name=name) + self.client.security_groups.delete(security_group.id) + except Exception as ex: logger.log_exc("Failed to delete security group") @@ -54,14 +56,9 @@ class SecurityGroup: from_port, to_port = icmp_type[0], icmp_type[1] group = self.client.security_groups.find(name=group_name) - self.client.security_group_rules.create( - group_id=group.id, - ip_protocol=protocol, - from_port=from_port, - to_port=to_port, - cidr_ip=cidr_ip, - ) - except Exception, ex: + self.client.security_group_rules.create(group.id, \ + protocol, from_port, to_port,cidr_ip) + except Exception as ex: logger.log_exc("Failed to add rule to group %s" % group_name) @@ -78,12 +75,12 @@ class SecurityGroup: 'id': group.id, 'from_port': from_port, 'to_port': to_port, - 'cird_ip': ip, + 'cidr_ip': ip, 'ip_protocol':protocol, } rule = self.client.security_group_rules.find(**filter) if rule: self.client.security_group_rules.delete(rule) - except Exception, ex: + except Exception as ex: logger.log_exc("Failed to remove rule from group %s" % group_name)