Simplify if/else and keep mypy happy

apache · Nov 7, 2022 · 4df4406 · 4df4406
1 parent d289217
commit 4df4406
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/airflow/api_connexion/schemas/dag_schema.py b/airflow/api_connexion/schemas/dag_schema.py
@@ -77,9 +77,10 @@ class Meta:
     @staticmethod
     def get_owners(obj: DagModel):
         """Convert owners attribute to DAG representation"""
-        if not getattr(obj, 'owners', None):
+        if obj.owners is None:
             return []
-        return obj.owners.split(",")
+        else:
+            return obj.owners.split(",")
 
     @staticmethod
     def get_token(obj: DagModel):
@@ -126,9 +127,10 @@ def get_tags(obj: DAG):
     @staticmethod
     def get_owners(obj: DAG):
         """Convert owners attribute to DAG representation"""
-        if not getattr(obj, 'owner', None):
+        if obj.owner is None:
             return []
-        return obj.owner.split(",")
+        else:
+            return obj.owner.split(",")
 
     @staticmethod
     def get_is_paused(obj: DAG):