simple_worker_pip/cloudcache.py at master · CloudMarc/simple_worker_pip · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
# SimpleWorker For Python
import time
from datetime import datetime
import json
import urllib2
import urllib
#from poster.encode import multipart_encode
#from poster.streaminghttp import register_openers
import hmac
#import sha
import base64
import hashlib

#class RequestWithMethod(urllib2.Request):
#    """Workaround for using DELETE with urllib2"""
#    def __init__(self, url, method, data=None, headers={},\
#        origin_req_host=None, unverifiable=False):
#        self._method = method
#        urllib2.Request.__init__(self, url, data, headers,\
#                 origin_req_host, unverifiable)
#
#    def get_method(self):
#        if self._method:
#            return self._method
#        else:
#            return urllib2.Request.get_method(self)
#

class CloudCache:
  def __init__(self, host, port, version, akey, skey):
    #self.url = "http://"+host+":"+str(port) + "/"+str(version)+"/"
    self.url = "http://"+host+":"+str(port) + "/"
    #print "url = " + self.url
    self.host = host
    self.port = port
    self.akey = akey
    self.skey = skey
    self.version= version
    self.project_id = ''
    self.headers = {}
    self.headers['Accept'] = "application/json"
    self.headers['Accept-Encoding'] = "gzip, deflate"
    self.headers['User-Agent'] = "CloudCache Python Pip v0.4"

  def __get(self, url, headers = {}):
    headers = dict(headers.items() + self.headers.items())
    #print "__get, url = " + url
    #print "__get , headers = " + str(headers)
    req = urllib2.Request(url, None, headers)
    ret = urllib2.urlopen(req)
    return ret.read()

  def __generate_timestamp(self, gmtime):
    return time.strftime("%Y-%m-%dT%H:%M:%SZ", gmtime)


  def __generate_signature(self, service, operation, timestamp, secret_access_key):
    #my_sha_hmac = hmac.new(secret_access_key, service + operation + timestamp, sha)
    my_sha_hmac = hmac.new(secret_access_key, service + operation + timestamp, hashlib.sha1)
    my_b64_hmac_digest = base64.encodestring(my_sha_hmac.hexdigest()).strip()
    #print "signature:  " + my_b64_hmac_digest
    return my_b64_hmac_digest


  def auth(self):
    ts = self.__generate_timestamp(time.gmtime())
    sig = self.__generate_signature("CloudCache", "auth", ts, self.skey)

    url = "http://" + self.host + ":" + self.port + "/auth"

    headers = {'User-Agent' : "CloudCache Pip version 0.4", 'signature' : sig, 'timestamp' : ts, 'akey' : self.akey}

    req = urllib2.Request(url, None, headers)
    ret = urllib2.urlopen(req)
    return str(ret.read())

  def getmulti(self, keys):
    ts = self.__generate_timestamp(time.gmtime())
    sig = self.__generate_signature("CloudCache", "multi", ts, self.skey)

    url = "http://" + self.host + ":" + self.port + "/getmulti"
    #print "getmulti url = " + url

    headers = {'User-Agent' : "CloudCache Pip version 0.4", 'signature' : sig, 'timestamp' : ts, 'akey' : self.akey}

    #headers['HTTP_KEYS'] = json.dumps(keys)
    headers['keys'] = json.dumps(keys)

    #print "getmulti headers:  " + str(headers)
    req = urllib2.Request(url, None, headers)
    ret = urllib2.urlopen(req)
    return str(ret.read())

  def get(self, key):
    ts = self.__generate_timestamp(time.gmtime())
    sig = self.__generate_signature("CloudCache", "GET", ts, self.skey)

    url = "http://" + self.host + ":" + self.port + "/"+key
    #print "get url = " + url

    headers = {'User-Agent' : "CloudCache Pip version 0.4", 'signature' : sig, 'timestamp' : ts, 'akey' : self.akey}

    req = urllib2.Request(url, None, headers)
    ret = urllib2.urlopen(req)
    return str(ret.read())


  def set(self, key, val, ttl = 0):
    ts  = self.__generate_timestamp(time.gmtime())
    sig = self.__generate_signature("CloudCache", "POST", ts, self.skey)

    url = "http://" + self.host + ":" + self.port + "/"+key
    #print "set (post) url = " + url
    headers = {'User-Agent' : "CloudCache Pip version 0.4", 'signature' : sig, 'timestamp' : ts, 'akey' : self.akey}
    if ttl != 0:
      headers['ttl'] = ttl

    data = {"val" : val}
    data = urllib.urlencode(data)

    req = urllib2.Request(url, data, headers)
    ret = urllib2.urlopen(req)

    return str(ret.read())