From 364c23dd487fbd4f9f848a88d48db7ef4442c716 Mon Sep 17 00:00:00 2001 From: Andrew Williams Date: Wed, 26 May 2010 09:25:34 +0100 Subject: [PATCH] Allow model to be defined on CSV export --- utils.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/utils.py b/utils.py index 352bb3c..e32c049 100644 --- a/utils.py +++ b/utils.py @@ -1,7 +1,7 @@ import csv from django.db.models.loading import get_model -def dump(qs, outfile_path): +def dump(qs, outfile_path, model=None): """ Takes in a Django queryset and spits out a CSV file. @@ -17,7 +17,9 @@ def dump(qs, outfile_path): http://www.djangosnippets.org/snippets/790/ """ - model = qs.model + + if not model: + model = qs.model writer = csv.writer(open(outfile_path, 'w')) headers = []