Add support for Amazon Machine Learning

README.rst

@@ -92,6 +92,7 @@ At the moment, boto supports:
   * Amazon Simple Email Service (SES) (Python 3)
   * Amazon Cognito Identity (Python 3)
   * Amazon Cognito Sync (Python 3)
+  * Amazon Machine Learning (Python 3)
 
 * Monitoring
 

boto/__init__.py

@@ -1083,6 +1083,22 @@ def connect_ec2containerservice(aws_access_key_id=None,
     )
 
 
+def connect_machinelearning(aws_access_key_id=None,
+                            aws_secret_access_key=None,
+                            **kwargs):
+    """
+    Connect to Amazon Machine Learning service
+    rtype: :class:`boto.machinelearning.layer1.MachineLearningConnection`
+    :return: A connection to the Amazon Machine Learning service
+    """
+    from boto.machinelearning.layer1 import MachineLearningConnection
+    return MachineLearningConnection(
+        aws_access_key_id=aws_access_key_id,
+        aws_secret_access_key=aws_secret_access_key,
+        **kwargs
+    )
+
+
 def storage_uri(uri_str, default_scheme='file', debug=0, validate=True,
                 bucket_storage_uri_class=BucketStorageUri,
                 suppress_consec_slashes=True, is_latest=False):

boto/auth.py

@@ -321,6 +321,8 @@ def headers_to_sign(self, http_request):
         in the StringToSign.
         """
         host_header_value = self.host_header(self.host, http_request)
+        if http_request.headers.get('Host'):
+            host_header_value = http_request.headers['Host']
         headers_to_sign = {'Host': host_header_value}
         for name, value in http_request.headers.items():
             lname = name.lower()

boto/connection.py

@@ -931,7 +931,8 @@ def _mexe(self, request, sender=None, override_num_retries=None,
                 # not include the port.
                 if 's3' not in self._required_auth_capability():
                     if not getattr(self, 'anon', False):
-                        self.set_host_header(request)
+                        if not request.headers.get('Host'):
+                            self.set_host_header(request)
                 boto.log.debug('Final headers: %s' % request.headers)
                 request.start_time = datetime.now()
                 if callable(sender):

boto/endpoints.json

@@ -279,6 +279,10 @@
         "us-east-1": "opsworks.us-east-1.amazonaws.com",
         "eu-central-1": "opsworks.eu-central-1.amazonaws.com"
     },
+    "machinelearning": {
+        "us-east-1": "machinelearning.us-east-1.amazonaws.com",
+        "us-west-2": "machinelearning.us-west-2.amazonaws.com"
+    },
     "rds": {
         "ap-northeast-1": "rds.ap-northeast-1.amazonaws.com",
         "ap-southeast-1": "rds.ap-southeast-1.amazonaws.com",

boto/machinelearning/__init__.py

@@ -0,0 +1,42 @@
+# Copyright (c) 2015 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.
+#
+from boto.regioninfo import RegionInfo, get_regions
+
+
+def regions():
+    """
+    Get all available regions for the Amazon Machine Learning.
+
+    :rtype: list
+    :return: A list of :class:`boto.regioninfo.RegionInfo`
+    """
+    from boto.machinelearning.layer1 import MachineLearningConnection
+    return get_regions('machinelearning',
+                       connection_cls=MachineLearningConnection)
+
+
+def connect_to_region(region_name, **kw_params):
+    for region in regions():
+        if region.name == region_name:
+            return region.connect(**kw_params)
+    return None

boto/machinelearning/exceptions.py

@@ -0,0 +1,51 @@
+# Copyright (c) 2015 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.
+#
+
+from boto.exception import BotoServerError
+
+
+class InternalServerException(BotoServerError):
+    pass
+
+
+class LimitExceededException(BotoServerError):
+    pass
+
+
+class IdempotentParameterMismatchException(BotoServerError):
+    pass
+
+
+class ResourceInUseException(BotoServerError):
+    pass
+
+
+class ResourceNotFoundException(BotoServerError):
+    pass
+
+
+class PredictorNotMountedException(BotoServerError):
+    pass
+
+
+class InvalidInputException(BotoServerError):
+    pass