FIX admin panel

Author: aledruetta

prototipo/projeto/ext/admin/__init__.py

@@ -1,16 +1,18 @@
 from flask_admin import Admin
-
 from projeto.ext.admin.views import EstacaoView, SensorView, UserView
 from projeto.ext.api.models import Estacao, Sensor
-from projeto.ext.auth import UserAuth
+from projeto.ext.auth.models import UserAuth
 from projeto.ext.db import db
 
-admin = Admin(name='FlaskAPI', template_mode='bootstrap3')
+admin = Admin(name="Dashboard", template_mode="bootstrap3")
 
+views = [
+    UserView(UserAuth, db.session),
+    EstacaoView(Estacao, db.session),
+    SensorView(Sensor, db.session),
+]
+admin.add_views(*views)
 
-def init_app(app):
-    admin.add_view(EstacaoView(Estacao, db.session))
-    admin.add_view(SensorView(Sensor, db.session))
-    admin.add_view(UserView(UserAuth, db.session))
 
+def init_app(app):
     admin.init_app(app)

prototipo/projeto/ext/admin/views.py

@@ -6,6 +6,8 @@
 class AdminView(ModelView):
     @login_required
     def is_accessible(self):
+        print(current_user.is_authenticated)
+        print(current_user.is_admin)
         if current_user.is_authenticated and current_user.is_admin:
             return True
         return False
@@ -16,6 +18,7 @@ def inaccessible_callback(self, name, **kwargs):
 
 
 class UserView(AdminView):
+    column_list = ("id", "email", "is_admin")
     column_sortable_list = ()