blob: e4bbb001f745a2f464ed6631d8befb8ef4b88c08 (
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
|
from flask_login import UserMixin
from tinydb import TinyDB, Query
db = TinyDB('db.json')
users_table = db.table('users')
UserQuery = Query()
class User(UserMixin):
def __init__(self, username, admin):
self.id = username
self.admin = admin
@staticmethod
def get(username):
result = users_table.get(UserQuery.username == username)
if result:
return User(username, result['admin'])
return None
@staticmethod
def validate_login(username, password):
user = users_table.get((UserQuery.username == username) & (UserQuery.password == password))
if user:
return User(username, user['admin'])
return None
@staticmethod
def create(username, password, admin=False):
if users_table.get(UserQuery.username == username):
return None
users_table.insert({'username': username, 'password': password, 'admin': admin})
return User(username, admin)
User.create('admin', 'dontguessthisplzahh', admin=True)
|
