Skip to content

Instantly share code, notes, and snippets.

@citmusa
Last active November 9, 2016 02:30
Show Gist options
  • Save citmusa/4a30ad4a72920b6637dcfa9c3893416a to your computer and use it in GitHub Desktop.
Save citmusa/4a30ad4a72920b6637dcfa9c3893416a to your computer and use it in GitHub Desktop.
Dump database to .json
import csv
from django.db.models.loading import get_model
def dump(qs, outfile_path):
"""
Takes in a Django queryset and spits out a CSV file.
Usage::
>> from utils import dump2csv
>> from dummy_app.models import *
>> qs = DummyModel.objects.all()
>> dump2csv.dump(qs, './data/dump.csv')
Based on a snippet by zbyte64::
http://www.djangosnippets.org/snippets/790/
"""
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)
#examples
python manage.py dumpdata --indent=2 > functional_tests/fixtures/alldb.json
python manage.py dumpdata auth.User --indent=2 > functional_tests/fixtures/auth_user.json
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment