From cbdaa83b7aac76f832a5620ecea1115fbbc80bc8 Mon Sep 17 00:00:00 2001 From: Justin Riley Date: Tue, 8 May 2012 21:20:30 -0400 Subject: [PATCH] use id instead of name in authorize() and revoke() closes gh-561 --- boto/ec2/securitygroup.py | 33 ++++++++++++++++----------------- 1 file changed, 16 insertions(+), 17 deletions(-) diff --git a/boto/ec2/securitygroup.py b/boto/ec2/securitygroup.py index 0afb8a37fb..6694ecdafe 100644 --- a/boto/ec2/securitygroup.py +++ b/boto/ec2/securitygroup.py @@ -151,26 +151,26 @@ def authorize(self, ip_protocol=None, from_port=None, to_port=None, :rtype: bool :return: True if successful. """ + src_group_name = None + src_group_owner_id = None + src_group_group_id = None if src_group: cidr_ip = None - src_group_name = src_group.name + if not self.vpc_id: + src_group_name = src_group.name src_group_owner_id = src_group.owner_id if hasattr(src_group, 'group_id'): src_group_group_id = src_group.group_id else: src_group_group_id = src_group.id - else: - src_group_name = None - src_group_owner_id = None - src_group_group_id = None - status = self.connection.authorize_security_group(self.name, - src_group_name, + status = self.connection.authorize_security_group(None, + None, src_group_owner_id, ip_protocol, from_port, to_port, cidr_ip, - None, + self.id, src_group_group_id) if status: if type(cidr_ip) != list: @@ -178,31 +178,30 @@ def authorize(self, ip_protocol=None, from_port=None, to_port=None, for single_cidr_ip in cidr_ip: self.add_rule(ip_protocol, from_port, to_port, src_group_name, src_group_owner_id, single_cidr_ip, src_group_group_id) - return status def revoke(self, ip_protocol=None, from_port=None, to_port=None, cidr_ip=None, src_group=None): + src_group_name = None + src_group_owner_id = None + src_group_group_id = None if src_group: cidr_ip=None - src_group_name = src_group.name + if not self.vpc_id: + src_group_name = src_group.name src_group_owner_id = src_group.owner_id if hasattr(src_group, 'group_id'): src_group_group_id = src_group.group_id else: src_group_group_id = src_group.id - else: - src_group_name = None - src_group_owner_id = None - src_group_group_id = None - status = self.connection.revoke_security_group(self.name, - src_group_name, + status = self.connection.revoke_security_group(None, + None, src_group_owner_id, ip_protocol, from_port, to_port, cidr_ip, - None, + self.id, src_group_group_id) if status: self.remove_rule(ip_protocol, from_port, to_port, src_group_name,