blob: 8588e873d84bd2d840c24cceca1e55d0f0be8c46 (
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
|
# -*- coding: utf-8 -*-
"""
SuperUser/Admin portal using flask-admin
"""
from app import app, login_manager, db
from flask import render_template, flash, url_for, redirect
import flask_login
from app.models import User
from flask_admin import Admin
from flask_admin.contrib.sqla import ModelView
admin = Admin(app, name="Admin", template_mode="bootstrap4")
"""
Initialised Admin Portal
"""
class ModelView(ModelView): # skipcq: PYL-E0102
"""
Our extended ModelView Class
Refer to Flask-Admin Docs for more details
https://flask-admin.readthedocs.io/en/latest/introduction/?highlight=ModelView#modelview-configuration-attributes
"""
def is_accessible(self):
"""This function checks if a user should be given access or not"""
try:
if flask_login.current_user.get_role() == "SUPERUSER":
return True
return False
except AttributeError:
return False
admin.add_view(ModelView(User, db.session))
|
