From 6a6b9c462600cec23ed57762f452e667cdc4d79d Mon Sep 17 00:00:00 2001
From: Benjamin Bertrand <benjamin.bertrand@esss.se>
Date: Mon, 2 Jul 2018 11:20:42 +0200
Subject: [PATCH] Add Task table to admin view

---
 app/admin/views.py | 5 +++++
 app/factory.py     | 3 ++-
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/app/admin/views.py b/app/admin/views.py
index 998718f..d159e87 100644
--- a/app/admin/views.py
+++ b/app/admin/views.py
@@ -85,3 +85,8 @@ class NetworkAdmin(AdminModelView):
     form_overrides = {
         'vlan_name': fields.StringField,
     }
+
+
+class TaskAdmin(AdminModelView):
+    column_display_pk = True
+    can_create = False
diff --git a/app/factory.py b/app/factory.py
index afea592..b611657 100644
--- a/app/factory.py
+++ b/app/factory.py
@@ -17,7 +17,7 @@ from . import settings, models
 from .extensions import (db, migrate, login_manager, ldap_manager, bootstrap,
                          admin, mail, jwt, toolbar, session_redis_store, fsession, cache)
 from .admin.views import (AdminModelView, ItemAdmin, UserAdmin, TokenAdmin,
-                          NetworkAdmin)
+                          NetworkAdmin, TaskAdmin)
 from .main.views import bp as main
 from .inventory.views import bp as inventory
 from .network.views import bp as network
@@ -112,6 +112,7 @@ def create_app(config=None):
     admin.add_view(AdminModelView(models.Mac, db.session))
     admin.add_view(AdminModelView(models.Cname, db.session))
     admin.add_view(AdminModelView(models.Tag, db.session))
+    admin.add_view(TaskAdmin(models.Task, db.session))
 
     app.register_blueprint(main)
     app.register_blueprint(inventory, url_prefix='/inventory')
-- 
GitLab