From e7d05127d16231475ffdb22da9c784fd346d09cf Mon Sep 17 00:00:00 2001 From: John Bodley <4567245+john-bodley@users.noreply.github.com> Date: Tue, 31 Jul 2018 14:52:03 -0700 Subject: [PATCH] [get_df] Updating multi-statement logic (#5517) --- superset/models/core.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/superset/models/core.py b/superset/models/core.py index 007049827b..9264fcfe36 100644 --- a/superset/models/core.py +++ b/superset/models/core.py @@ -738,8 +738,12 @@ class Database(Model, AuditMixinNullable, ImportMixin): with closing(engine.raw_connection()) as conn: with closing(conn.cursor()) as cursor: - for sql in sqls: + for sql in sqls[:-1]: self.db_engine_spec.execute(cursor, sql) + cursor.fetchall() + + self.db_engine_spec.execute(cursor, sqls[-1]) + df = pd.DataFrame.from_records( data=list(cursor.fetchall()), columns=[col_desc[0] for col_desc in cursor.description],