azure/azure.py

   1 #!/usr/bin/python2.6
   2
   3 """A simple test API for accessing Windows Azure blob storage.
   4
   5 Parts of the code are modeled after boto (a library for accessing Amazon Web
   6 Services), but this code is far less general and is meant only as a
   7 proof-of-concept."""
   8
   9 import base64, hashlib, hmac, httplib, os, time
  10
  11 # The version of the Azure API we implement; sent in the x-ms-version header.
  12 API_VERSION = '2009-09-19'
  13
  14 def uri_decode(s):
  15     # TODO
  16     return s
  17
  18 def add_auth_header(headers, method, path):
  19     header_order = ['Content-Encoding', 'Content-Language', 'Content-Length',
  20                     'Content-MD5', 'Content-Type', 'Date', 'If-Modified-Since',
  21                     'If-Match', 'If-None-Match', 'If-Unmodified-Since',
  22                     'Range']
  23
  24     if not headers.has_key('Date'):
  25         headers['Date'] = time.strftime("%a, %d %b %Y %H:%M:%S GMT",
  26                                         time.gmtime())
  27     if not headers.has_key('x-ms-version'):
  28         headers['x-ms-version'] = API_VERSION
  29
  30     StringToSign = method + "\n"
  31     for h in header_order:
  32         if h in headers:
  33             StringToSign += headers[h] + "\n"
  34         else:
  35             StringToSign += "\n"
  36
  37     # Add Canonicalized Headers
  38     canonized = []
  39     for (k, v) in headers.items():
  40         k = k.lower()
  41         if k.startswith('x-ms-'):
  42             canonized.append((k, v))
  43     canonized.sort()
  44     for (k, v) in canonized:
  45         StringToSign += "%s:%s\n" % (k, v)
  46
  47     # Add CanonicalizedHeaders Resource
  48     account_name = os.environ['AZURE_ACCOUNT_NAME']
  49     account_name = 'bluesky'
  50     resource = "/" + account_name
  51     if '?' not in path:
  52         resource += path
  53     else:
  54         (path, params) = path.split('?', 1)
  55         params = [p.split('=') for p in params.split("&")]
  56         params = dict((k.lower(), uri_decode(v)) for (k, v) in params)
  57         resource += path
  58         for k in sorted(params):
  59             resource += "\n%s:%s" % (k, params[k])
  60     StringToSign += resource
  61
  62     # print "String to sign:", repr(StringToSign)
  63
  64     secret_key = os.environ['AZURE_SECRET_KEY']
  65     secret_key = base64.b64decode(secret_key)
  66     h = hmac.new(secret_key, digestmod=hashlib.sha256)
  67     h.update(StringToSign)
  68
  69     signature = base64.b64encode(h.digest())
  70     headers['Authorization'] = "SharedKey %s:%s" % (account_name, signature)
  71
  72 def generate_request(path, method='GET', body="", headers={}):
  73     host = os.environ['AZURE_ACCOUNT_NAME'] + ".blob.core.windows.net"
  74     #auth_path = conn.calling_format.build_auth_path(self.bucket, key)
  75
  76     headers = headers.copy()
  77     headers['Content-Length'] = str(len(body))
  78     if len(body) > 0:
  79         headers['Content-MD5'] = base64.b64encode(hashlib.md5(body).digest())
  80     add_auth_header(headers, method, path)
  81
  82     req = "%s %s HTTP/1.1\r\nHost: %s\r\n" % (method, path, host)
  83     req = req + ''.join("%s: %s\r\n" % h for h in headers.items()) + "\r\n"
  84     print req
  85
  86     conn = httplib.HTTPConnection(host)
  87     conn.request(method, path, body, headers)
  88     response = conn.getresponse()
  89     print "Response:", response.status
  90     print "Headers:", response.getheaders()
  91     print "Body:", response.read()
  92
  93 if False:
  94     generate_request("/?comp=list")