From d6846d96ff08b4a4322075f2735107f2ed4d2490 Mon Sep 17 00:00:00 2001 From: michellethomas Date: Wed, 13 Jun 2018 11:49:43 -0700 Subject: [PATCH] Adding column only if it doesn't already exist (#5179) --- superset/data/__init__.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/superset/data/__init__.py b/superset/data/__init__.py index 4f79be842a90b..7bae20bae984e 100644 --- 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()