| Index: third_party/gsutil/third_party/boto/boto/regioninfo.py
|
| diff --git a/third_party/gsutil/third_party/boto/boto/regioninfo.py b/third_party/gsutil/third_party/boto/boto/regioninfo.py
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..6aeda122f94127ba11314839d86975b29118f4d2
|
| --- /dev/null
|
| +++ b/third_party/gsutil/third_party/boto/boto/regioninfo.py
|
| @@ -0,0 +1,187 @@
|
| +# Copyright (c) 2006-2010 Mitch Garnaat http://garnaat.org/
|
| +# Copyright (c) 2010, Eucalyptus Systems, Inc.
|
| +# 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.
|
| +import os
|
| +
|
| +import boto
|
| +from boto.compat import json
|
| +from boto.exception import BotoClientError
|
| +
|
| +
|
| +def load_endpoint_json(path):
|
| + """
|
| + Loads a given JSON file & returns it.
|
| +
|
| + :param path: The path to the JSON file
|
| + :type path: string
|
| +
|
| + :returns: The loaded data
|
| + """
|
| + with open(path, 'r') as endpoints_file:
|
| + return json.load(endpoints_file)
|
| +
|
| +
|
| +def merge_endpoints(defaults, additions):
|
| + """
|
| + Given an existing set of endpoint data, this will deep-update it with
|
| + any similarly structured data in the additions.
|
| +
|
| + :param defaults: The existing endpoints data
|
| + :type defaults: dict
|
| +
|
| + :param defaults: The additional endpoints data
|
| + :type defaults: dict
|
| +
|
| + :returns: The modified endpoints data
|
| + :rtype: dict
|
| + """
|
| + # We can't just do an ``defaults.update(...)`` here, as that could
|
| + # *overwrite* regions if present in both.
|
| + # We'll iterate instead, essentially doing a deeper merge.
|
| + for service, region_info in additions.items():
|
| + # Set the default, if not present, to an empty dict.
|
| + defaults.setdefault(service, {})
|
| + defaults[service].update(region_info)
|
| +
|
| + return defaults
|
| +
|
| +
|
| +def load_regions():
|
| + """
|
| + Actually load the region/endpoint information from the JSON files.
|
| +
|
| + By default, this loads from the default included ``boto/endpoints.json``
|
| + file.
|
| +
|
| + Users can override/extend this by supplying either a ``BOTO_ENDPOINTS``
|
| + environment variable or a ``endpoints_path`` config variable, either of
|
| + which should be an absolute path to the user's JSON file.
|
| +
|
| + :returns: The endpoints data
|
| + :rtype: dict
|
| + """
|
| + # Load the defaults first.
|
| + endpoints = load_endpoint_json(boto.ENDPOINTS_PATH)
|
| + additional_path = None
|
| +
|
| + # Try the ENV var. If not, check the config file.
|
| + if os.environ.get('BOTO_ENDPOINTS'):
|
| + additional_path = os.environ['BOTO_ENDPOINTS']
|
| + elif boto.config.get('Boto', 'endpoints_path'):
|
| + additional_path = boto.config.get('Boto', 'endpoints_path')
|
| +
|
| + # If there's a file provided, we'll load it & additively merge it into
|
| + # the endpoints.
|
| + if additional_path:
|
| + additional = load_endpoint_json(additional_path)
|
| + endpoints = merge_endpoints(endpoints, additional)
|
| +
|
| + return endpoints
|
| +
|
| +
|
| +def get_regions(service_name, region_cls=None, connection_cls=None):
|
| + """
|
| + Given a service name (like ``ec2``), returns a list of ``RegionInfo``
|
| + objects for that service.
|
| +
|
| + This leverages the ``endpoints.json`` file (+ optional user overrides) to
|
| + configure/construct all the objects.
|
| +
|
| + :param service_name: The name of the service to construct the ``RegionInfo``
|
| + objects for. Ex: ``ec2``, ``s3``, ``sns``, etc.
|
| + :type service_name: string
|
| +
|
| + :param region_cls: (Optional) The class to use when constructing. By
|
| + default, this is ``RegionInfo``.
|
| + :type region_cls: class
|
| +
|
| + :param connection_cls: (Optional) The connection class for the
|
| + ``RegionInfo`` object. Providing this allows the ``connect`` method on
|
| + the ``RegionInfo`` to work. Default is ``None`` (no connection).
|
| + :type connection_cls: class
|
| +
|
| + :returns: A list of configured ``RegionInfo`` objects
|
| + :rtype: list
|
| + """
|
| + endpoints = load_regions()
|
| +
|
| + if service_name not in endpoints:
|
| + raise BotoClientError(
|
| + "Service '%s' not found in endpoints." % service_name
|
| + )
|
| +
|
| + if region_cls is None:
|
| + region_cls = RegionInfo
|
| +
|
| + region_objs = []
|
| +
|
| + for region_name, endpoint in endpoints.get(service_name, {}).items():
|
| + region_objs.append(
|
| + region_cls(
|
| + name=region_name,
|
| + endpoint=endpoint,
|
| + connection_cls=connection_cls
|
| + )
|
| + )
|
| +
|
| + return region_objs
|
| +
|
| +
|
| +class RegionInfo(object):
|
| + """
|
| + Represents an AWS Region
|
| + """
|
| +
|
| + def __init__(self, connection=None, name=None, endpoint=None,
|
| + connection_cls=None):
|
| + self.connection = connection
|
| + self.name = name
|
| + self.endpoint = endpoint
|
| + self.connection_cls = connection_cls
|
| +
|
| + def __repr__(self):
|
| + return 'RegionInfo:%s' % self.name
|
| +
|
| + def startElement(self, name, attrs, connection):
|
| + return None
|
| +
|
| + def endElement(self, name, value, connection):
|
| + if name == 'regionName':
|
| + self.name = value
|
| + elif name == 'regionEndpoint':
|
| + self.endpoint = value
|
| + else:
|
| + setattr(self, name, value)
|
| +
|
| + def connect(self, **kw_params):
|
| + """
|
| + Connect to this Region's endpoint. Returns an connection
|
| + object pointing to the endpoint associated with this region.
|
| + You may pass any of the arguments accepted by the connection
|
| + class's constructor as keyword arguments and they will be
|
| + passed along to the connection object.
|
| +
|
| + :rtype: Connection object
|
| + :return: The connection to this regions endpoint
|
| + """
|
| + if self.connection_cls:
|
| + return self.connection_cls(region=self, **kw_params)
|
|
|