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) |