defer dagbag load until first list call

ms32035 · Aug 29, 2019 · 4c861de · 4c861de
1 parent 72fa82c
commit 4c861de
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/dag-dependencies-plugin/dag_dependencies_plugin.py b/dag-dependencies-plugin/dag_dependencies_plugin.py
@@ -10,7 +10,7 @@
 
 
 class DAGDependenciesView(BaseView):
-    dagbag = models.DagBag(settings.DAGS_FOLDER)
+    dagbag = None
     plugins_folder = conf.get("core", "plugins_folder")
     template_folder = os.path.join(plugins_folder, "dag-dependencies-plugin")
     route_base = "/"
@@ -34,8 +34,11 @@ def render(self, template, **context):
     def list(self):
         title = "DAG Dependencies"
 
+        if DAGDependenciesView.dagbag is None:
+            DAGDependenciesView.dagbag = models.DagBag(settings.DAGS_FOLDER)
+
         if datetime.utcnow() > self.last_refresh + timedelta(
-                seconds=self.refresh_interval
+            seconds=self.refresh_interval
         ):
             DAGDependenciesView.dagbag.collect_dags()
             self.nodes, self.edges = self._generate_graph()