fix exception when dag labels is none

epoch8 · Feb 13, 2021 · 3c628f3 · 3c628f3
1 parent e3efa78
commit 3c628f3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/airflow_exporter/prometheus_exporter.py b/airflow_exporter/prometheus_exporter.py
@@ -77,11 +77,11 @@ def get_dag_labels(dag_id):
         return [], []
 
     labels = dag.params.get('labels')
-    labels = {k:v for k,v in labels.items() if not k.startswith('__')}
 
     if labels is None:
         return [], []
 
+    labels = {k:v for k,v in labels.items() if not k.startswith('__')}
     return list(labels.keys()), list(labels.values())