Allow model to be defined on CSV export

This commit is contained in:
2010-05-26 09:25:34 +01:00
parent 3aee1ba1e7
commit 364c23dd48

View File

@@ -1,7 +1,7 @@
import csv import csv
from django.db.models.loading import get_model 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. Takes in a Django queryset and spits out a CSV file.
@@ -17,6 +17,8 @@ def dump(qs, outfile_path):
http://www.djangosnippets.org/snippets/790/ http://www.djangosnippets.org/snippets/790/
""" """
if not model:
model = qs.model model = qs.model
writer = csv.writer(open(outfile_path, 'w')) writer = csv.writer(open(outfile_path, 'w'))