blob: 7e441b2293ad8b5f77935486ba9ec2aa590b26d9 (
plain)
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
|
from app import app
from database import database_get
from flask_login import LoginManager, UserMixin, current_user
class User(UserMixin):
def __init__(self, id, email, password_hash):
self.id = unicode(id)
self.email = email
self.password_hash = password_hash
self.authenticated = False
def is_authenticated(self):
return self.authenticated
def is_active(self):
return True
def is_anonymous(self):
return False
def get_id(self):
return self.id
login_manager = LoginManager()
login_manager.init_app(app)
@login_manager.user_loader
def load_user(uid):
conn = database_get()
curs = conn.cursor()
curs.execute("SELECT * FROM users WHERE id = (?)", [uid])
row = curs.fetchone()
if row is None:
return None
else:
return User(int(row[0]), row[1], row[2])
|
