cmueller/ansible
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Correct test integration ec2 cleanup script

Browse source
This commit is contained in:
James Laska 2014-03-12 19:28:57 -04:00
parent e753b39eae
commit 9cc4be2683
1 changed files with 7 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
test/integration/cleanup_ec2.py
View file

@ -15,12 +15,14 @@ def delete_aws_resources(get_func, attr, opts):
for item in get_func(): for item in get_func():
val = getattr(item, attr) val = getattr(item, attr)
if re.search(opts.match_re, val): if re.search(opts.match_re, val):
prompt_and_delete("Delete object with %s=%s? [y/n]: " % (attr, val), opts.assumeyes) prompt_and_delete(item, "Delete object with %s=%s? [y/n]: " % (attr, val), opts.assumeyes)
def prompt_and_delete(prompt, assumeyes): def prompt_and_delete(item, prompt, assumeyes):
while not assumeyes: if not assumeyes:
assumeyes = raw_input(prompt) assumeyes = raw_input(prompt).lower() == 'y'
obj.delete() assert hasattr(item, 'delete'), "Class <%s> has no delete attribute" % item.__class__
if assumeyes:
item.delete()
def parse_args(): def parse_args():
# Load details from credentials.yml # Load details from credentials.yml