Generazione di file CSV con Django (contenuto dinamico)

Question

All'interno di my view.py, ho due funzioni, una che elabora l'input da un modulo e genera un elenco filtrato e un altro che deve esportare questo elenco in CSV.

Ecco il ritorno del mio prima funzione:

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

Ecco la funzione di esportazione:

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 

non sono sicuro di come ottenere il set di query dalla mia prima funzione, che contiene un elenco degli elementi che voglio nel mio CSV e lo uso nella mia funzione export_to_csv. Oppure sarebbe il modo migliore di combinare queste due funzioni e fare clic su una casella di controllo se desidera scaricare un file CSV. Qualsiasi aiuto sarebbe apprezzato.

Answer 1

mi consiglia che unisce questi in una funzione di visualizzazione che prende un parametro aggiuntivo:

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)) 

Poi, nel vostro urls.py, mettere qualcosa di simile nel tuo urlpatterns:

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

Answer 2

IMHO, il migliore sarebbe combinarli e generare i dati CSV da un queryset esplicito. Questo potrebbe quindi essere riscritto per qualcosa di generale come (non testato):

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 

che è possibile utilizzare in questo modo:

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

-

Il codice di esempio che hai fornito assume la QuerySet è impostato nei dati della sessione, che potrebbe causare un sacco di bug e problemi di sicurezza. Se memorizzi sessioni nel tuo database, potresti finire a leggere i dati, solo per riscriverli in una forma molto meno efficiente.

Answer 3

Ho trovato un modo per farlo diverso da quello di knutin. Ho dichiarato una lista vuota chiamata csv_list = [] al di fuori delle mie funzioni, o variabile globale.

Nella mia funzione principale (quella che elabora e filtra in base all'input dell'utente), trasformo questa csv_list globale in modo che sia impostata sulla versione "aggiornata" del queryset. Poi per generare il csv, è facile come: per la chiamata in csv_list: writer.writerow ([call.src, call.dst]) Risposta ritorno

Sta funzionando ragionevolmente ora.

Answer 4

Il seguente prende un queryset Django e sputa un file CSV.

Usage ::

da utils importare dump2csv

da dummy_app.models import *

Qs = DummyModel.objects.tutto()

dump2csv.dump (qs, './data/dump.csv')

Lo script:

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)