-
-
Notifications
You must be signed in to change notification settings - Fork 0
/
auth.py
55 lines (46 loc) · 1.63 KB
/
auth.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
try:
import urllib.parse as urllib_parse
from base64 import encodebytes
except ImportError:
import urllib as urllib_parse
from base64 import encodestring as encodebytes
class Auth(object):
"""
ABC for Authenticator objects.
"""
def encode_params(self, base_url, method, params):
"""Encodes parameters for a request suitable for including in a URL
or POST body. This method may also add new params to the request
if required by the authentication scheme in use."""
raise NotImplementedError()
def generate_headers(self):
"""Generates headers which should be added to the request if required
by the authentication scheme in use."""
raise NotImplementedError()
class UserPassAuth(Auth):
"""
Basic auth authentication using email/username and
password. Deprecated.
"""
def __init__(self, username, password):
self.username = username
self.password = password
def encode_params(self, base_url, method, params):
# We could consider automatically converting unicode to utf8 strings
# before encoding...
return urllib_parse.urlencode(params)
def generate_headers(self):
return {b"Authorization": b"Basic " + encodebytes(
("%s:%s" %(self.username, self.password))
.encode('utf8')).strip(b'\n')
}
class NoAuth(Auth):
"""
No authentication authenticator.
"""
def __init__(self):
pass
def encode_params(self, base_url, method, params):
return urllib_parse.urlencode(params)
def generate_headers(self):
return {}