hmac_client.py

import requests
import time
import hashlib
import sys
import hmac
import logging

try: # for Python 3
    from http.client import HTTPConnection
except ImportError:
    from httplib import HTTPConnection
HTTPConnection.debuglevel = 1

logging.basicConfig() # you need to initialize logging, otherwise you will not see anything from requests
logging.getLogger().setLevel(logging.DEBUG)
requests_log = logging.getLogger("requests.packages.urllib3")
requests_log.setLevel(logging.DEBUG)
requests_log.propagate = True

def get():
    path_and_query = "/?TIMESTAMP="+str(int(time.time()))+"&ACCOUNT_ID=admin"
    host = "http://localhost:5000"
    sig=hmac.new("secret", digestmod=hashlib.sha1, msg=path_and_query).hexdigest()
    req = requests.get(host+path_and_query, headers={'X-Auth-Signature': sig})
    print req.content


def post():
    path_and_query = "/?TIMESTAMP="+str(int(time.time()))+"&ACCOUNT_ID=admin"
    host = "http://localhost:5000"
    sig=hmac.new("secret", digestmod=hashlib.sha1, msg=path_and_query).hexdigest()
    req = requests.post(host+path_and_query, headers={'X-Auth-Signature': sig}, data={'n': '1'})
    print req.content


def put():
    path_and_query = "/?TIMESTAMP="+str(int(time.time()))+"&ACCOUNT_ID=admin"
    host = "http://localhost:5000"
    sig=hmac.new("secret", digestmod=hashlib.sha1, msg=path_and_query).hexdigest()
    req = requests.put(host+path_and_query, headers={'X-Auth-Signature': sig}, data={'n': '1'})
    print req.content

if __name__ == '__main__':
    if len(sys.argv) > 1:
        method = sys.argv[1]
        if method == 'post':
            post()
        elif method == 'get':
            get()
        elif method == 'put':
            put()
        else:
            print "NA"
    else:
        print "get|put|post"