From 5f6236c80bbe143c283bcb0e8bd36b4195f1aaf2 Mon Sep 17 00:00:00 2001
From: Maxime Beauchemin <maximebeauchemin@gmail.com>
Date: Thu, 17 Dec 2015 18:04:53 -0800
Subject: [PATCH] Export to csv or json

---
 panoramix/templates/panoramix/explore.html | 13 +++++++++++++
 panoramix/views.py                         |  7 +++++++
 panoramix/viz.py                           |  8 ++++++++
 3 files changed, 28 insertions(+)

diff --git a/panoramix/templates/panoramix/explore.html b/panoramix/templates/panoramix/explore.html
index 40e7bb2f23..9f9793a433 100644
--- a/panoramix/templates/panoramix/explore.html
+++ b/panoramix/templates/panoramix/explore.html
@@ -36,6 +36,13 @@
         <span class="btn btn-info pull-right disabled query"
           data-toggle="modal" data-target="#query_modal">query</span>
         <span class="btn btn-warning pull-right notbtn" id="timer">0 sec</span>
+        <div class="btn-group pull-right" role="group">
+          <span class="btn btn-default disabled">
+            <i class="fa fa-file-text"></i>
+          </span>
+          <span class="btn btn-default" id="csv">.csv</span>
+          <span class="btn btn-default" id="json">.json</span>
+        </div>
         <hr/>
     </div>
     <div class="row">
@@ -229,6 +236,12 @@
     $('legend').click(function () {
       toggle_fieldset($(this), true);
     });
+    $('#json').click(function () {
+      window.location = '{{ viz.json_endpoint | safe }}';
+    });
+    $('#csv').click(function () {
+      window.location = '{{ viz.csv_endpoint | safe }}';
+    });
     $("#viz_type").change(function() {$("#query").submit();});
     collapsed_fieldsets = get_collapsed_fieldsets();
     for(var i=0; i<collapsed_fieldsets.length;i++){
diff --git a/panoramix/views.py b/panoramix/views.py
index 5d74bb6017..4c99a12fea 100644
--- a/panoramix/views.py
+++ b/panoramix/views.py
@@ -399,6 +399,13 @@ class Panoramix(BaseView):
         obj = viz.viz_types[viz_type](
             datasource,
             form_data=request.args)
+        if request.args.get("csv") == "true":
+            status = 200
+            payload = obj.get_csv()
+            return Response(
+                payload,
+                status=status,
+                mimetype="application/csv")
         if request.args.get("json") == "true":
             status = 200
             if config.get("DEBUG"):
diff --git a/panoramix/viz.py b/panoramix/viz.py
index 576f88cd34..3f42103bda 100644
--- a/panoramix/viz.py
+++ b/panoramix/viz.py
@@ -208,6 +208,10 @@ class BaseViz(object):
         }
         return json.dumps(payload)
 
+    def get_csv(self):
+        df = self.get_df()
+        return df.to_csv(index=False)
+
     def get_json_data(self):
         return json.dumps([])
 
@@ -215,6 +219,10 @@ class BaseViz(object):
     def json_endpoint(self):
         return self.get_url(json="true")
 
+    @property
+    def csv_endpoint(self):
+        return self.get_url(csv="true")
+
     def get_data_attribute(self):
         content = {
             'viz_name': self.viz_type,