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
|
#!/usr/bin/env python
#
# Copyright 2009 Canonical Ltd. This software is licensed under the
# GNU Affero General Public License version 3 (see the file LICENSE).
import os
import pprint
import re
import sys
import boto
owners = {
'gary': 255383312499,
'francis': 559320013529}
_image_match = re.compile(
r'launchpad-ec2test(\d+)/image.manifest.xml$').match
if __name__ == '__main__':
if len(sys.argv) == 1:
owner = os.environ['USER']
elif len(sys.argv) == 2:
owner = sys.argv[1]
else:
raise RuntimeError('Too many arguments')
try:
owner = int(owner)
except ValueError:
owner = owners[owner]
# Get the AWS identifier and secret identifier.
aws_id = os.path.join(os.environ['HOME'], '.ec2', 'aws_id')
if not os.path.exists(aws_id):
raise RuntimeError(
"Please put your aws access key identifier and secret access "
"key identifier in %s. (On two lines).\n" % (aws_id,))
aws_file = open(aws_id, 'r')
try:
identifier = aws_file.readline().strip()
secret = aws_file.readline().strip()
finally:
aws_file.close()
# Make the EC2 connection.
conn = boto.connect_ec2(identifier, secret)
# get images
images = {}
for image in conn.get_all_images(owners=owners.values()):
match = _image_match(image.location)
if match:
val = int(match.group(1))
if val not in images:
images[val] = [image]
else:
images[val].append(image)
images = images.items()
images.sort()
# set
last, recent = images[-2:]
for tmp in (last, recent):
if len(tmp[1]) > 1:
raise ValueError(
'more than one image of value %d found: %r' % tmp)
recent = recent[1][0]
last = last[1][0]
perms = last.get_launch_permissions()
perms['user_ids'].extend(owners.values())
recent.set_launch_permissions(**perms)
# done
pprint.pprint(recent.get_launch_permissions())
|