···

       
12
       
12
       
 
       


     

       
13
       
13
       
 
       
parser = argparse.ArgumentParser(description='Create an EBS-backed NixOS AMI')

     

       
14
       
14
       
 
       
parser.add_argument('--region', dest='region', required=True, help='EC2 region to create the image in')

     

       
15
       
15
       
-
       
parser.add_argument('--channel', dest='channel', default="13.10", help='Channel to use')

     

       
15
       
15
       
+
       
parser.add_argument('--channel', dest='channel', default="14.12", help='Channel to use')

     

       
16
       
16
       
 
       
parser.add_argument('--keep', dest='keep', action='store_true', help='Keep NixOps machine after use')

     

       
17
       
17
       
 
       
parser.add_argument('--hvm', dest='hvm', action='store_true', help='Create HVM image')

     

       
18
       
18
       
 
       
parser.add_argument('--key', dest='key_name', action='store_true', help='Keypair used for HVM instance creation', default="rob")

     
···

       
54
       
54
       
 
       
except Exception:

     

       
55
       
55
       
 
       
    depl = db.create_deployment()

     

       
56
       
56
       
 
       
    depl.name = "ebs-creator"

     

       
57
       
57
       
-
       
depl.auto_response = "y"

     

       
57
       
57
       
+
       
depl.logger.set_autoresponse("y")

     

       
58
       
58
       
 
       
depl.nix_exprs = [os.path.abspath("./ebs-creator.nix"), os.path.abspath("./ebs-creator-config.nix")]

     

       
59
       
59
       
 
       
if not args.keep: depl.destroy_resources()

     

       
60
       
60
       
 
       
depl.deploy(allow_reboot=True)

     
···

       
140
       
140
       
 
       
        )

     

       
141
       
141
       
 
       
if not args.hvm:

     

       
142
       
142
       
 
       
    common_args['kernel_id']=aki.id

     

       
143
       
143
       
+
       


     

       
143
       
144
       
 
       
ami_id = m._conn.register_image(**common_args)

     

       
144
       
145
       
 
       


     

       
145
       
146
       
 
       
print >> sys.stderr, "registered AMI {0}".format(ami_id)

     
···

       
185
       
186
       
 
       
test_depl.machines['machine'].run_command("nixos-version")

     

       
186
       
187
       
 
       


     

       
187
       
188
       
 
       
# Log the AMI ID.

     

       
188
       
188
       
-
       
f = open("{0}.{1}.ami-id".format(args.region, image_type), "w")

     

       
189
       
189
       
-
       
f.write("{0}".format(ami_id))

     

       
190
       
190
       
-
       
f.close()

     

       
189
       
189
       
+
       
f = open("ec2-amis.nix".format(args.region, image_type), "w")

     

       
190
       
190
       
+
       
f.write("{\n")

     

       
191
       
191
       
 
       


     

       
192
       
192
       
 
       
for dest in [ 'us-east-1', 'us-west-1', 'us-west-2', 'eu-west-1', 'eu-central-1', 'ap-southeast-1', 'ap-southeast-2', 'ap-northeast-1', 'sa-east-1']:

     

       
193
       
193
       
+
       
    copy_image = None

     

       
193
       
194
       
 
       
    if args.region != dest:

     

       
194
       
194
       
-
       
        print >> sys.stderr, "copying image from region {0} to {1}".format(args.region, dest)

     

       
195
       
195
       
-
       
        conn = boto.ec2.connect_to_region(dest)

     

       
196
       
196
       
-
       
        copy_image = conn.copy_image(args.region, ami_id, ami_name, description=None, client_token=None)

     

       
195
       
195
       
+
       
        try:

     

       
196
       
196
       
+
       
            print >> sys.stderr, "copying image from region {0} to {1}".format(args.region, dest)

     

       
197
       
197
       
+
       
            conn = boto.ec2.connect_to_region(dest)

     

       
198
       
198
       
+
       
            copy_image = conn.copy_image(args.region, ami_id, ami_name, description=None, client_token=None)

     

       
199
       
199
       
+
       
        except :

     

       
200
       
200
       
+
       
            print >> sys.stderr, "FAILED!"

     

       
197
       
201
       
 
       


     

       
198
       
202
       
 
       
        # Log the AMI ID.

     

       
199
       
199
       
-
       
        f = open("{0}.{1}.ami-id".format(dest, image_type), "w")

     

       
200
       
200
       
-
       
        f.write("{0}".format(copy_image.image_id))

     

       
201
       
201
       
-
       
        f.close()

     

       
203
       
203
       
+
       
        if copy_image != None:

     

       
204
       
204
       
+
       
            f.write('  "{0}"."{1}".{2} = "{3}";\n'.format(args.channel,dest,"hvm" if args.hvm else "ebs",copy_image.image_id))

     

       
205
       
205
       
+
       
    else:

     

       
206
       
206
       
+
       
        f.write('  "{0}"."{1}".{2} = "{3}";\n'.format(args.channel,args.region,"hvm" if args.hvm else "ebs",ami_id))

     

       
202
       
207
       
 
       


     

       
203
       
208
       
 
       


     

       
209
       
209
       
+
       
f.write("}\n")

     

       
210
       
210
       
+
       
f.close()

     

       
211
       
211
       
+
       


     

       
204
       
212
       
 
       
if not args.keep:

     

       
213
       
213
       
+
       
    test_depl.logger.set_autoresponse("y")

     

       
205
       
214
       
 
       
    test_depl.destroy_resources()

     

       
206
       
215
       
 
       
    test_depl.delete()

     

       
207
       
216