Add user admin

avilaton · Jun 21, 2018 · 7450a10 · 7450a10
1 parent 74e84dd
commit 7450a10
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/app/admin/__init__.py b/app/admin/__init__.py
@@ -3,7 +3,7 @@
 def register_admin_views(admin):
 
     # admin views
-    from .views import MyModelView
+    from .views import MyModelView, UserModelView
     from app.models import Agency
     from app.models import Calendar
     from app.models import CalendarDate
@@ -13,6 +13,7 @@ def register_admin_views(admin):
     from app.models import Stop
     from app.models import StopTime
     from app.models import ShapePath
+    from app.models import User
     from app import db
 
 
@@ -23,3 +24,4 @@ def register_admin_views(admin):
     admin.add_view(MyModelView(Route, db.session))
     admin.add_view(MyModelView(Trip, db.session))
     admin.add_view(MyModelView(Stop, db.session))
+    admin.add_view(UserModelView(User, db.session))
diff --git a/app/admin/views.py b/app/admin/views.py
@@ -15,6 +15,11 @@ def inaccessible_callback(self, name, **kwargs):
         return current_app.login_manager.unauthorized()
 
 
+class UserModelView(MyModelView):
+    excluded_list_columns = ['password_hash']
+    excluded_form_columns = ['password_hash']
+
+
 # Create customized index view class that handles login & registration
 class MyAdminIndexView(AdminIndexView):