pt
/
superset
mirror of https://github.com/apache/superset.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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

This commit is contained in:
michellethomas 2018-06-13 11:49:43 -07:00 committed by John Bodley
parent 7f30b48955
commit d6846d96ff
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
superset/data/__init__.py
View File

@ -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()