mj add another admin
This commit is contained in:
7
app.py
7
app.py
@@ -21,17 +21,18 @@ login_manager.login_view = 'auth.login'
|
||||
login_manager.login_message = 'Παρακαλώ συνδεθείτε για να συνεχίσετε.'
|
||||
|
||||
class User(UserMixin):
|
||||
def __init__(self, id, username):
|
||||
def __init__(self, id, username, clan_id):
|
||||
self.id = id
|
||||
self.username = username
|
||||
self.clan_id = clan_id
|
||||
|
||||
@login_manager.user_loader
|
||||
def load_user(user_id):
|
||||
conn = get_db()
|
||||
row = conn.execute('SELECT id, username FROM users WHERE id = ?', (user_id,)).fetchone()
|
||||
row = conn.execute('SELECT id, username, clan_id FROM users WHERE id = ?', (user_id,)).fetchone()
|
||||
conn.close()
|
||||
if row:
|
||||
return User(row['id'], row['username'])
|
||||
return User(row['id'], row['username'], row['clan_id'])
|
||||
return None
|
||||
|
||||
# Make current_user available in all templates
|
||||
|
||||
Reference in New Issue
Block a user