-
Notifications
You must be signed in to change notification settings - Fork 0
/
utils.py
93 lines (74 loc) · 2.36 KB
/
utils.py
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
import sys, base64, urllib2, json, getpass, requests
from requests.auth import HTTPBasicAuth
from HTMLParser import HTMLParser
def initConnection():
global user
global password
global url
user = '2nnztrtz0ue'
password = getpass.getpass("Password : ")
url = 'https://api.myintervals.com/'
def startConnection(urlString):
urlRequest = url + urlString
#formatOutput(urlRequest)
request = urllib2.Request(urlRequest)
authHeader = encodeHeader()
request.add_header('Authorization', authHeader)
request.add_header('Accept', 'application/json')
try:
handle = urllib2.urlopen(request)
except IOError, e: # here we shouldn't fail if the username/password is right
print "It looks like the username or password is wrong."
print e
sys.exit(1)
jsonObject = json.load(handle)
handle.close()
return jsonObject
def getConnection(urlString):
urlRequest = url + urlString
try:
request = requests.get(urlRequest, auth=(user,password))
except IOError, e:
print e
sys.exit(1)
return request.text
def postConnection(urlString, infos):
urlRequest = url + urlString
headers = {'content-type':'application/json'}
try:
request = requests.post(urlRequest, data=json.dumps(infos), auth=(user, password), headers=headers)
except IOError, e:
print "It looks like the username or password is wrong"
print e
sys.exit(1)
return request.text
def putConnection(urlString, data):
urlRequest = url + urlString
request = requests.put(urlString, json.dumps(data))
try:
print "oups"
except IOError, e:
print "It looks like the username or password is wrong"
print e
sys.exit(1)
jsonObject = json.load(handle)
handle.close()
return jsonObject
def encodeHeader():
base64string = base64.encodestring('%s:%s' % (user, password))[:-1]
authheader = "Basic %s" % base64string
return authheader
def stripTags(html):
s = PPStripper()
s.feed(html)
return s.get_data()
def formatOutput(jsonObject):
print json.dumps(jsonObject, indent=4)
class PPStripper(HTMLParser):
def __init__(self):
self.reset()
self.fed = []
def handle_data(self, d):
self.fed.append(d)
def get_data(self):
return ''.join(self.fed)