Adding column only if it doesn't already exist (#5179)

apache · Jun 13, 2018 · d6846d9 · d6846d9
1 parent 7f30b48
commit d6846d9
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/superset/data/__init__.py b/superset/data/__init__.py
@@ -586,10 +586,13 @@ def load_birth_names():
     obj.main_dttm_col = 'ds'
     obj.database = utils.get_or_create_main_db()
     obj.filter_select_enabled = True
-    obj.columns.append(TableColumn(
-        column_name='num_california',
-        expression="CASE WHEN state = 'CA' THEN num ELSE 0 END"
-    ))
+
+    if not any(col.column_name == 'num_california' for col in obj.columns):
+        obj.columns.append(TableColumn(
+            column_name='num_california',
+            expression="CASE WHEN state = 'CA' THEN num ELSE 0 END"
+        ))
+
     db.session.merge(obj)
     db.session.commit()
     obj.fetch_metadata()