[get_df] Updating multi-statement logic (#5517)

This commit is contained in:
John Bodley 2018-07-31 14:52:03 -07:00 committed by GitHub
parent 8c822f1e9b
commit e7d05127d1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 1 deletions

View File

@ -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],