|
340
|
1 import httplib2
|
|
|
2 import json
|
|
|
3 from urllib import urlencode
|
|
|
4
|
|
346
|
5 _credentials = None
|
|
|
6
|
|
|
7 def SetCredentials(username, password):
|
|
|
8 global _credentials
|
|
|
9 _credentials = (username, password)
|
|
|
10
|
|
|
11 def _SetupCredentials(h):
|
|
|
12 global _credentials
|
|
|
13 if _credentials != None:
|
|
|
14 h.add_credentials(_credentials[0], _credentials[1])
|
|
|
15
|
|
340
|
16 def DoGet(uri, data = {}):
|
|
|
17 d = ''
|
|
|
18 if len(data.keys()) > 0:
|
|
|
19 d = '?' + urlencode(data)
|
|
|
20
|
|
|
21 h = httplib2.Http()
|
|
346
|
22 _SetupCredentials(h)
|
|
340
|
23 resp, content = h.request(uri + d, 'GET')
|
|
|
24 if not (resp.status in [ 200 ]):
|
|
|
25 raise Exception(resp.status)
|
|
|
26 else:
|
|
|
27 try:
|
|
|
28 return json.loads(content)
|
|
|
29 except:
|
|
|
30 return content
|
|
|
31
|
|
|
32
|
|
|
33 def _DoPutOrPost(uri, method, data, contentType):
|
|
|
34 h = httplib2.Http()
|
|
346
|
35 _SetupCredentials(h)
|
|
340
|
36
|
|
|
37 if isinstance(data, str):
|
|
|
38 body = data
|
|
|
39 if len(contentType) != 0:
|
|
|
40 headers = { 'content-type' : contentType }
|
|
|
41 else:
|
|
|
42 body = json.dumps(data)
|
|
|
43 headers = { 'content-type' : 'application/json' }
|
|
|
44
|
|
|
45 resp, content = h.request(
|
|
|
46 uri, method,
|
|
|
47 body = body,
|
|
|
48 headers = headers)
|
|
|
49
|
|
|
50 if not (resp.status in [ 200, 302 ]):
|
|
|
51 raise Exception(resp.status)
|
|
|
52 else:
|
|
|
53 try:
|
|
|
54 return json.loads(content)
|
|
|
55 except:
|
|
|
56 return content
|
|
|
57
|
|
|
58
|
|
|
59 def DoDelete(uri):
|
|
|
60 h = httplib2.Http()
|
|
346
|
61 _SetupCredentials(h)
|
|
340
|
62 resp, content = h.request(uri, 'DELETE')
|
|
346
|
63
|
|
340
|
64 if not (resp.status in [ 200 ]):
|
|
|
65 raise Exception(resp.status)
|
|
|
66 else:
|
|
|
67 try:
|
|
|
68 return json.loads(content)
|
|
|
69 except:
|
|
|
70 return content
|
|
|
71
|
|
|
72
|
|
|
73 def DoPut(uri, data = {}, contentType = ''):
|
|
|
74 return _DoPutOrPost(uri, 'PUT', data, contentType)
|
|
|
75
|
|
|
76
|
|
|
77 def DoPost(uri, data = {}, contentType = ''):
|
|
|
78 return _DoPutOrPost(uri, 'POST', data, contentType)
|
