added admin view

1 files changed, 20 insertions, 0 deletions

diff --git a/app/views/admin.py b/app/views/admin.py
new file mode 100644
index 0000000..b971a1a
--- /dev/null
+++ b/app/views/admin.py
@@ -0,0 +1,20 @@
+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")
+
+class ModelView(ModelView):
+    def is_accessible(self):
+        try:
+            if flask_login.current_user.get_role() == "SUPERUSER":
+                return True
+            return False
+        except AttributeError:
+            return False
+
+admin.add_view(ModelView(User, db.session))
\ No newline at end of file