From 508ce49a00fe61c47b4f8e892a1e56fe603d3b42 Mon Sep 17 00:00:00 2001 From: Patrick Michaud Date: Tue, 11 Apr 2017 15:54:01 -0700 Subject: [PATCH] Shimming in things for non-pg backends --- sqlshare_rest/backend/base.py | 6 ++++++ sqlshare_rest/views/sql.py | 1 + 2 files changed, 7 insertions(+) diff --git a/sqlshare_rest/backend/base.py b/sqlshare_rest/backend/base.py index 370d05e..872fae1 100644 --- a/sqlshare_rest/backend/base.py +++ b/sqlshare_rest/backend/base.py @@ -21,6 +21,12 @@ def get_query_plan(self, sql, user): """ return "" + def run_named_cursor_query(self, *args, **kwargs): + return self.run_query(*args, **kwargs) + + def finish_named_cursor(self, *args, **kwargs): + pass + def run_query(self, sql, user, params=None, return_cursor=False, query=None): self._not_implemented("run_query") diff --git a/sqlshare_rest/views/sql.py b/sqlshare_rest/views/sql.py index 2a73c47..4d3d87e 100644 --- a/sqlshare_rest/views/sql.py +++ b/sqlshare_rest/views/sql.py @@ -41,6 +41,7 @@ def response_for_query(sql, user, download_name): disposition = 'attachment; filename="%s"' % download_name response = StreamingHttpResponse(stream_query(cursor, user), content_type='text/csv') + response['Content-Disposition'] = disposition return response except Exception as ex: