add csv download field to search results

nonstop/templates/nonstop/search.html

@@ -42,6 +42,9 @@
             <a href="?{{ qs }}&page={{ tracks.next_page_number }}">→</a>
         {% endif %}
     </span>
+    <span class="download">
+      <a href="csv?{{ qs }}" download="nonstop.csv">CSV</a>
+    </span>
 </div>
 {% endif %}
 

nonstop/urls.py

 from django.conf.urls import url
 
-from .views import SomaDayArchiveView, SomaDayArchiveCsvView, RedirectTodayView, TrackDetailView, ArtistDetailView, ArtistListView, StatisticsView, UploadTracksView, RecentTracksView, QuickLinksView, SearchView, CleanupView
+from .views import SomaDayArchiveView, SomaDayArchiveCsvView, RedirectTodayView, TrackDetailView, ArtistDetailView, ArtistListView, StatisticsView, UploadTracksView, RecentTracksView, QuickLinksView, SearchView, CleanupView, SearchCsvView
 
 urlpatterns = [
     # Example: /2012/nov/10/
@@ -19,6 +19,7 @@ urlpatterns = [
     url(r'^upload/$', UploadTracksView.as_view(), name='nonstop-upload-tracks'),
     url(r'^recent/$', RecentTracksView.as_view(), name='nonstop-recent-tracks'),
     url(r'^search/$', SearchView.as_view(), name='nonstop-search'),
+    url(r'^search/csv$', SearchCsvView.as_view(), name='nonstop-search-csv'),
     url(r'^quick-links/$', QuickLinksView.as_view(), name='nonstop-quick-links'),
     url(r'^cleanup/$', CleanupView.as_view(), name='nonstop-cleanup'),
 ]

nonstop/views.py

@@ -248,9 +248,7 @@ class QuickLinksView(TemplateView):
 class SearchView(TemplateView):
     template_name = 'nonstop/search.html'
 
-    def get_context_data(self, **kwargs):
-        ctx = super(SearchView, self).get_context_data(**kwargs)
-        ctx['form'] = TrackSearchForm(self.request.GET)
+    def get_queryset(self):
         queryset = Track.objects.all()
 
         q = self.request.GET.get('q')
@@ -267,15 +265,20 @@ class SearchView(TemplateView):
             else:
                 queryset = queryset.filter(nonstop_zones=zone)
 
-        qs = self.request.GET.copy()
-        qs.pop('page', None)
-        ctx['qs'] = qs.urlencode()
-
         order = self.request.GET.get('order_by') or 'title'
         if order:
             if 'added_to_nonstop_timestamp' in order:
                 queryset = queryset.filter(added_to_nonstop_timestamp__isnull=False)
             queryset = queryset.order_by(order)
+        return queryset
+
+    def get_context_data(self, **kwargs):
+        ctx = super(SearchView, self).get_context_data(**kwargs)
+        ctx['form'] = TrackSearchForm(self.request.GET)
+        queryset = self.get_queryset()
+        qs = self.request.GET.copy()
+        qs.pop('page', None)
+        ctx['qs'] = qs.urlencode()
 
         tracks = Paginator(queryset.select_related(), 20)
 
@@ -290,6 +293,22 @@ class SearchView(TemplateView):
         return ctx
 
 
+class SearchCsvView(SearchView):
+    def get(self, request, *args, **kwargs):
+        out = StringIO()
+        writer = csv.writer(out)
+        writer.writerow(['Title', 'Artist', 'Zones', 'Language', 'Instru', 'CFWB'])
+        for track in self.get_queryset():
+            writer.writerow([
+                track.title.encode('utf-8', 'replace') if track.title else 'Inconnu',
+                track.artist.name.encode('utf-8', 'replace') if (track.artist and track.artist.name) else 'Inconnu',
+                ' + '.join([x.title.encode('utf-8') for x in track.nonstop_zones.all()]),
+                track.language or '',
+                track.instru and 'instru' or '',
+                track.cfwb and 'cfwb' or ''])
+        return HttpResponse(out.getvalue(), content_type='text/csv; charset=utf-8')
+
+
 class CleanupView(TemplateView):
     template_name = 'nonstop/cleanup.html'