-
Notifications
You must be signed in to change notification settings - Fork 1
/
oauth_http_server.py
81 lines (71 loc) · 2.4 KB
/
oauth_http_server.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
import threading
import traceback
import socket
from logging import getLogger
logger = getLogger('oauth_http_server')
try:
from BaseHTTPServer import HTTPServer, BaseHTTPRequestHandler
except ImportError:
from http.server import HTTPServer, BaseHTTPRequestHandler
try:
import urlparse
except ImportError:
from urllib import parse as urlparse
def find_unused_port():
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM, 0)
sock.bind(('', 0))
sock.listen(socket.SOMAXCONN)
ipaddr, port = sock.getsockname()
sock.close()
return port
class OAuthHandler(BaseHTTPRequestHandler):
def do_GET(self):
path = [i for i in self.path.split('/') if i]
logger.debug("Path: %s" % path)
if path[0] != 'oauth':
raise ValueError("Unable to process non-OAuth request.")
callback_data = "?".join(path[1].split('?')[1:])
parsed_data = {k: v[0]
for k, v in urlparse.parse_qs(callback_data).items()}
try:
self.server.callback(parsed_data)
except Exception as e:
logger.exception("Error calling oAuth login callback.")
self.send_response(500)
self.end_headers()
self.wfile.writelines(traceback.format_exc(e))
return
self.send_response(200)
self.end_headers()
self.wfile.write(self.server.success_msg)
t = threading.Thread(target=self.server.shutdown)
t.daemon = True
t.start()
class OAuthServer(HTTPServer):
allow_reuse_address = 0
def __init__(self, host='localhost', port=None, handler_class=OAuthHandler, callback=None):
if not callable(callback):
raise TypeError("Must provide a callable callback.")
if port is None:
port = find_unused_port()
# Because for some reason super won't work for the HTTPServer
HTTPServer.__init__(self, (host, port), handler_class)
self.host, self.port = host, port
self.callback = callback
def get_oauth_callback_url(self):
return 'http://%s:%d/oauth/callback' % (self.host, self.port)
success_msg = \
"""
<html>
<head>
<title>Successful login!</title>
</head>
<body>
<h1>Login successful!</h1>
<p>
You have been logged in!
You can now close this window and enjoy the app.
</p>
</body>
</html>
"""