Index: third_party/gsutil/boto/boto/ec2/blockdevicemapping.py |
diff --git a/third_party/gsutil/boto/boto/ec2/blockdevicemapping.py b/third_party/gsutil/boto/boto/ec2/blockdevicemapping.py |
new file mode 100644 |
index 0000000000000000000000000000000000000000..ca0e9373c53c65c827c8b22a21d62208e85abe68 |
--- /dev/null |
+++ b/third_party/gsutil/boto/boto/ec2/blockdevicemapping.py |
@@ -0,0 +1,141 @@ |
+# Copyright (c) 2009-2012 Mitch Garnaat http://garnaat.org/ |
+# Copyright (c) 2012 Amazon.com, Inc. or its affiliates. All Rights Reserved |
+# |
+# Permission is hereby granted, free of charge, to any person obtaining a |
+# copy of this software and associated documentation files (the |
+# "Software"), to deal in the Software without restriction, including |
+# without limitation the rights to use, copy, modify, merge, publish, dis- |
+# tribute, sublicense, and/or sell copies of the Software, and to permit |
+# persons to whom the Software is furnished to do so, subject to the fol- |
+# lowing conditions: |
+# |
+# The above copyright notice and this permission notice shall be included |
+# in all copies or substantial portions of the Software. |
+# |
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS |
+# OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABIL- |
+# ITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT |
+# SHALL THE AUTHOR BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, |
+# WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, |
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS |
+# IN THE SOFTWARE. |
+# |
+ |
+ |
+class BlockDeviceType(object): |
+ """ |
+ Represents parameters for a block device. |
+ """ |
+ |
+ def __init__(self, |
+ connection=None, |
+ ephemeral_name=None, |
+ no_device=False, |
+ volume_id=None, |
+ snapshot_id=None, |
+ status=None, |
+ attach_time=None, |
+ delete_on_termination=False, |
+ size=None, |
+ volume_type=None, |
+ iops=None): |
+ self.connection = connection |
+ self.ephemeral_name = ephemeral_name |
+ self.no_device = no_device |
+ self.volume_id = volume_id |
+ self.snapshot_id = snapshot_id |
+ self.status = status |
+ self.attach_time = attach_time |
+ self.delete_on_termination = delete_on_termination |
+ self.size = size |
+ self.volume_type = volume_type |
+ self.iops = iops |
+ |
+ def startElement(self, name, attrs, connection): |
+ pass |
+ |
+ def endElement(self, name, value, connection): |
+ if name == 'volumeId': |
+ self.volume_id = value |
+ elif name == 'virtualName': |
+ self.ephemeral_name = value |
+ elif name == 'NoDevice': |
+ self.no_device = (value == 'true') |
+ elif name == 'snapshotId': |
+ self.snapshot_id = value |
+ elif name == 'volumeSize': |
+ self.size = int(value) |
+ elif name == 'status': |
+ self.status = value |
+ elif name == 'attachTime': |
+ self.attach_time = value |
+ elif name == 'deleteOnTermination': |
+ self.delete_on_termination = (value == 'true') |
+ elif name == 'volumeType': |
+ self.volume_type = value |
+ elif name == 'iops': |
+ self.iops = int(value) |
+ else: |
+ setattr(self, name, value) |
+ |
+# for backwards compatibility |
+EBSBlockDeviceType = BlockDeviceType |
+ |
+ |
+class BlockDeviceMapping(dict): |
+ """ |
+ Represents a collection of BlockDeviceTypes when creating ec2 instances. |
+ |
+ Example: |
+ dev_sda1 = BlockDeviceType() |
+ dev_sda1.size = 100 # change root volume to 100GB instead of default |
+ bdm = BlockDeviceMapping() |
+ bdm['/dev/sda1'] = dev_sda1 |
+ reservation = image.run(..., block_device_map=bdm, ...) |
+ """ |
+ |
+ def __init__(self, connection=None): |
+ """ |
+ :type connection: :class:`boto.ec2.EC2Connection` |
+ :param connection: Optional connection. |
+ """ |
+ dict.__init__(self) |
+ self.connection = connection |
+ self.current_name = None |
+ self.current_value = None |
+ |
+ def startElement(self, name, attrs, connection): |
+ if name == 'ebs' or name == 'virtualName': |
+ self.current_value = BlockDeviceType(self) |
+ return self.current_value |
+ |
+ def endElement(self, name, value, connection): |
+ if name == 'device' or name == 'deviceName': |
+ self.current_name = value |
+ elif name == 'item': |
+ self[self.current_name] = self.current_value |
+ |
+ def build_list_params(self, params, prefix=''): |
+ i = 1 |
+ for dev_name in self: |
+ pre = '%sBlockDeviceMapping.%d' % (prefix, i) |
+ params['%s.DeviceName' % pre] = dev_name |
+ block_dev = self[dev_name] |
+ if block_dev.ephemeral_name: |
+ params['%s.VirtualName' % pre] = block_dev.ephemeral_name |
+ else: |
+ if block_dev.no_device: |
+ params['%s.Ebs.NoDevice' % pre] = 'true' |
+ if block_dev.snapshot_id: |
+ params['%s.Ebs.SnapshotId' % pre] = block_dev.snapshot_id |
+ if block_dev.size: |
+ params['%s.Ebs.VolumeSize' % pre] = block_dev.size |
+ if block_dev.delete_on_termination: |
+ params['%s.Ebs.DeleteOnTermination' % pre] = 'true' |
+ else: |
+ params['%s.Ebs.DeleteOnTermination' % pre] = 'false' |
+ if block_dev.volume_type: |
+ params['%s.Ebs.VolumeType' % pre] = block_dev.volume_type |
+ if block_dev.iops is not None: |
+ params['%s.Ebs.Iops' % pre] = block_dev.iops |
+ i += 1 |