Skip to content

nfcharles/python-aws-sign

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

35 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

AWS Sign

AWS Sign provides classes for signing AWS HTTP requests adhering to the signature version 4 specification.

Additionally, a tornado based HTTP client capable of sending signature v4 signed requests is provided.

Classes

ServiceConstants

Request signing requires many inputs, one of which is a set of values representing the AWS service : this includes, but not limited:

  • host
  • service name
  • region
  • signing algorithm

ServiceConstants provides a logical grouping of these values.

Services that require more constants should subclass ServiceConstants. As an example, consider a class representing APIGateway. ServiceConstants doesn't sufficiently represent all the constants needed to sign requests so we must create a sublcass.

class APIGatewayServiceConstants(Sigv4ServiceConstants):
    # Parsed by 'from_url' method.  Matched group array is passed as *args list to
    # constructor so ordinal positions of match values must match constructor args.
    URL_REGEX = re.compile(r"""(https)://      # scheme
                               (\w+            # api prefix
                               \.
                               ([\w\-]+)       # service
                               \.
                               ([\w\-]+)       # region
                               .amazonaws.com)
                               \/
                               ([\w]+)$        # stage""", re.X)
        
    def __init__(self, *args):
        super(APIGatewayServiceConstants, self).__init__(*args[:4])
        self.stage = args[4]

    def path(self, *args):
        return '%s/%s' % (self.stage, '/'.join(args))
        
    def __str__(self):
        return '%s\nstage=%s' % (super(APIGatewayServiceConstants, self).__str__(), self.stage)

Consider another example using dynamodb

class DynamoDBServiceConstants(Sigv4ServiceConstants):
    __REQUIRED_HEADERS = {'content-type':None, 'x-amz-target': None}

    def __init__(self, *args):
        super(DynamoDBServiceConstants, self).__init__(*args)
        self.__headers = self._merge(super(DynamoDBServiceConstants, self).headers,
                                     self.__REQUIRED_HEADERS)

    @property
    def headers(self):
        return self.__headers

Note that DynamoDBServiceConstants has inherited host and x-amz-date headers from Sigv4ServiceConstants instance.

>>> sc = DynamoDBServiceConstants.from_url('https://dynamodb.us-west-2.amazonaws.com')
>>> pprint.pprint(sc.headers)
{'content-type': None,
 'host': 'dynamodb.us-west-2.amazonaws.com',
 'x-amz-date': None,
 'x-amz-target': None}

Authorization

auth.Authorization encapsulates the signing behavior. Invoke the header instance method to get the HTTP 'Authorization' header data necessary for signed requests. Also note that signing values can be encoded in querystring parameters.

...

a = auth.Authorization(ServiceConstants(*args), Credentials())

# Sign request
http_headers['Authorization'] = a.header(**kwargs)

...

Client

An tornado based HTTP client is provided that implicitly supports signature version 4 signing.

from boto3 import session
from aws_sign.v4 import Sigv4ServiceConstants
from aws_sign.client import http
from tornado.httpclient import HTTPError

import pprint
import json


# Define APIGatewayServiceConstants
...

try:
    creds    = session.Session().get_credentials()
    endpoint = 'https://12345abcde.execute-api.us-west-2.amazonaws.com/test'
    defaults = {'headers': {'content-type': 'application/x-www-form-urlencoded'}}

    client = http.get_instance(endpoint, APIGatewayServiceConstants, defaults, sign=True, creds=creds)
    resp = client.get('/test/foo')

    pprint.pprint(json.loads(resp.body))
    # {'foo': ['bar', 'baz']}

except HTTPError as e:
    print e.response
except Exception, e:
    print e 

License

AWS Sign is free software and is released under the terms of the MIT license (http://opensource.org/licenses/mit-license.php), as specified in the accompanying LICENSE.txt file.