-
Notifications
You must be signed in to change notification settings - Fork 0
/
login1.py
45 lines (37 loc) · 1001 Bytes
/
login1.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
import web
import re
import base64
urls = (
'/','Index',
'/login','Login'
)
app = web.application(urls,globals())
allowed = (
('jon','pass1'),
('tom','pass2')
)
class Index:
def GET(self):
if web.ctx.env.get('HTTP_AUTHORIZATION') is not None:
return 'This is the index page'
else:
raise web.seeother('/login')
class Login:
def GET(self):
auth = web.ctx.env.get('HTTP_AUTHORIZATION')
authreq = False
if auth is None:
authreq = True
else:
auth = re.sub('^Basic ','',auth)
username,password = base64.decodestring(auth).split(':')
if (username,password) in allowed:
raise web.seeother('/')
else:
authreq = True
if authreq:
web.header('WWW-Authenticate','Basic realm="Auth example"')
web.ctx.status = '401 Unauthorized'
return
if __name__=='__main__':
app.run()