-
Notifications
You must be signed in to change notification settings - Fork 0
/
example.py
99 lines (70 loc) · 2.39 KB
/
example.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
93
94
95
96
97
98
99
"""
This example is based on the code example from bottle-session by
Christopher De Vries (https://github.com/devries/bottle-session).
"""
import bottle
import random
import string
import time
from bottle import request, response, HTTPResponse, template
from peewee_session import PeeweeSessionPlugin
from util import authenticator, generate_token
from peewee import SqliteDatabase
db = SqliteDatabase('test.db')
CONFIG = {'db': db}
app = bottle.app()
app.config.update(CONFIG)
session_plugin = PeeweeSessionPlugin(
cookie_lifetime='10 seconds',
db_conn=db, cookie_secret='very-s3kr3t-s4lt')
app.install(session_plugin)
csrf_token = generate_token(20)
username = "Mellanie"
PASSWORD = "StickyNotesOnTheBackofYourScreenAren'tSafe"
class User:
@staticmethod
def verify_password(user, password):
if user == username and password == PASSWORD:
return True
else:
return False
login_required = authenticator(session_plugin.session_manager,
csrf_token,
User.verify_password, app,
'/login',
redirect_success="/",
session_cookie='coldswaet-cookie'
)
@app.route('/')
@login_required()
def get_main_page(session):
if session.load('name') is None:
context = {'csrf_token': csrf_token}
return bottle.template('set_name', **context)
else:
context = {'csrf_token': csrf_token,
'name': session.load('name'),
'trigger': session.load('trigger')
}
return bottle.template('has_name', **context)
@app.route('/submit', method='POST')
def set_name(session):
session['name'] = request.forms.name.strip()
csrf_submitted = request.forms.get('csrf_token')
if csrf_submitted != csrf_token:
return template('error',
warning_message='Cross-site scripting error.')
bottle.redirect('/')
@app.route('/logout')
def logout(session):
del session['name']
response = HTTPResponse(
body="Bye!",
status=200,
)
response.delete_cookie('login_redirect',
path='/',)
return response
if __name__ == '__main__':
bottle.debug(True)
bottle.run(app=app, host='127.0.0.1', port=8080)