Generowanie pliku CSV za pomocą Django (zawartość dynamiczna)

Question

Wewnątrz mojego pliku view.py mam dwie funkcje, jedną, która przetwarza dane wejściowe z formularza i wyświetla przefiltrowaną listę, a drugą, która ma wyeksportować tę listę do pliku CSV.

Oto powrót mojej pierwszej funkcji:

return render_to_response('templateX.html', 
{ 
'queryset': queryset, 
'filter_form': filter_form, 
'validated': validated, 
}, 
context_instance = RequestContext(request) 
) 

Oto funkcja eksportujący:

def export_to_csv(request): 
    # get the response object, this can be used as a stream. 
    response = HttpResponse(mimetype='text/csv') 
    # force download. 
    response['Content-Disposition'] = 'attachment;filename=export.csv' 
    # the csv writer 
    writer = csv.writer(response) 
    qs = request.session['queryset'] 
    for cdr in qs: 
     writer.writerow([cdr['calldate'], cdr['src'], cdr['dst'], ]) 
    return response 

Nie jestem pewien, jak uzyskać queryset z mojej pierwszej funkcji, który zawiera listę elementów, które chcę w moim pliku CSV i wykorzystuję w funkcji export_to_csv. A może najlepszym sposobem byłoby połączenie tych dwóch funkcji i aby użytkownik kliknął pole wyboru, czy chce pobrać plik CSV. Każda pomoc będzie doceniona.

Answer 1

Polecam łącząc je w jednej funkcji widoku, który bierze dodatkowy parametr:

def my_view(request, exportCSV): 
    # ... Figure out `queryset` here ... 

    if exportCSV: 
     response = HttpResponse(mimetype='text/csv') 
     response['Content-Disposition'] = 'attachment;filename=export.csv' 
     writer = csv.writer(response) 
     for cdr in queryset: 
      writer.writerow([cdr['calldate'], cdr['src'], cdr['dst'], ]) 
     return response 
    else: 
     return render_to_response('templateX.html', {'queryset': queryset, 
      'filter_form': filter_form, 'validated': validated}, 
      context_instance = RequestContext(request)) 

Następnie w swojej urls.py, umieścić coś takiego w swoim urlpatterns:

url(r'^form', 'my_view', {"exportCSV": False}, name="form"), 
url(r'^csv', 'my_view', {"exportCSV": True}, name="export"), 

Answer 2

IMHO, najlepiej byłoby je połączyć i wygenerować dane CSV z jawnego zestawu zapytań. To może następnie zostać przepisany na coś jak generał (nie testowane):

def export_to_csv(request, queryset, fields): 
    response = ... 
    writer = csv.writer(response) 
    for obj in queryset: 
     writer.writerow([getattr(obj, f) for f in fields]) 
    return response 

których można używać tak:

def my_view(request): 
    calls = Call.objects.all() 
    return export_to_csv(request, calls, fields = ('calldate', 'src', 'dst')) 

-

Przykładowy kod podałeś zakłada QuerySet jest ustawić w danych sesji, co może spowodować mnóstwo błędów, a także problemy z bezpieczeństwem. Jeśli przechowujesz sesje w bazie danych, możesz skończyć odczyt danych, po prostu zapisać je w znacznie mniej wydajnej formie.

Answer 3

Znalazłem sposób na zrobienie czegoś innego niż knutin. Deklarowałem pustą listę o nazwie csv_list = [] poza moimi funkcjami lub zmienną globalną.

W mojej głównej funkcji (tej, która przetwarza i filtruje na podstawie danych wprowadzonych przez użytkownika), robię tę csv_list globalną, aby została ustawiona na "zaktualizowaną" wersję zestawu zapytań. Następnie w celu wygenerowania pliku CSV, to tak proste, jak: dla wywołania w csv_list: writer.writerow ([call.src, call.dst]) Pasmo powrót

To działa rozsądnie teraz.

Answer 4

Poniższy przyjmuje zestaw zapytań Django i wypluwa plik CSV.

Zastosowanie ::

z utils import dump2csv

z dummy_app.models import *

qs = DummyModel.objects.wszystko()

dump2csv.dump (QS './data/dump.csv')

Skrypt:

import csv 
from django.db.models.loading import get_model 

def dump(qs, outfile_path): 

    model = qs.model 
writer = csv.writer(open(outfile_path, 'w')) 

headers = [] 
for field in model._meta.fields: 
    headers.append(field.name) 
writer.writerow(headers) 

for obj in qs: 
    row = [] 
    for field in headers: 
     val = getattr(obj, field) 
     if callable(val): 
      val = val() 
     if type(val) == unicode: 
      val = val.encode("utf-8") 
     row.append(val) 
    writer.writerow(row)