Shimming in things for non-pg backends

uw-it-aca · Apr 11, 2017 · 508ce49 · 508ce49
1 parent 93632db
commit 508ce49
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git 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
@@ -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: