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
|
#!/usr/bin/env python
from argparse import ArgumentParser
import logging
from boto.ec2 import regions
from jujupy import SimpleEnvironment
from substrate import AWSAccount
def parse_args(argv=None):
parser = ArgumentParser('Clean up leftover resources.')
parser.add_argument('env', help='The juju environment to use.')
parser.add_argument('--verbose', '-v', action='count', default=0)
parser.add_argument('-a', '--all-regions', action='store_true',
help='Take action on all regions.')
return parser.parse_args(argv)
def get_regions(args, env):
if args.all_regions:
return [region.name for region in filter(
lambda x: '-gov-' not in x.name, regions())]
return [env.get_region()]
def clean(args):
env = SimpleEnvironment.from_config(args.env)
selected_regions = get_regions(args, env)
for region in selected_regions:
with AWSAccount.from_boot_config(env, region=region) as substrate:
logging.info('Cleaning resources in {}.'.format(substrate.region))
all_groups = dict(substrate.iter_security_groups())
instance_groups = dict(substrate.iter_instance_security_groups())
non_instance_groups = dict((k, v) for k, v in all_groups.items()
if k not in instance_groups)
unclean = substrate.delete_detached_interfaces(
non_instance_groups.keys())
logging.info('Unable to delete {} groups'.format(len(unclean)))
for group_id in unclean:
logging.debug('Cannot delete {}'.format(all_groups[group_id]))
for group_id in unclean:
non_instance_groups.pop(group_id, None)
substrate.destroy_security_groups(non_instance_groups.values())
def main():
args = parse_args()
log_level = max(logging.WARN - args.verbose * 10, logging.DEBUG)
logging.basicConfig(level=log_level)
logging.getLogger('boto').setLevel(logging.CRITICAL)
clean(args)
if __name__ == '__main__':
main()
|