-
Notifications
You must be signed in to change notification settings - Fork 1
/
User.py
30 lines (26 loc) · 812 Bytes
/
User.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
from flask_login import login_manager
import flask_login
from flask import url_for
from __init__ import login_manager
#initial user
users = {'admin@gmail.com': {'password': 'admin123'}}
class User(flask_login.UserMixin):
@login_manager.user_loader
def user_loader(email):
if email not in users:
return
user = User()
user.id = email
return user
@login_manager.request_loader
def request_loader(request):
email = request.form.get('email')
if email not in users:
return
user = User()
user.id = email
user.is_authenticated = request.form['password'] == users[email]['password']
return user
# @login_manager.unauthorized_handler
# def unauthorized_handler(self):
# return False