Export data with django-excel

dj-export-excel.py

''' Exporta planilhas em Excel '''
from datetime import datetime
import django_excel as excel
from django.http import HttpResponseBadRequest
from .models import Person

MDATA = datetime.now().strftime('%Y-%m-%d')


def export_data_person(request, atype):
    if atype == "sheet":
        return excel.make_response_from_a_table(
            Person, 'xls', file_name="contatos")
    elif atype == "custom":
        query_sets = Person.objects.all()
        column_names = ['id', 'name', 'email']
        return excel.make_response_from_query_sets(
            query_sets,
            column_names,
            'xls',
            file_name="contatos-" + MDATA
        )
    else:
        return HttpResponseBadRequest(
            "Bad request. please put one of these " +
            "in your url suffix: sheet, book or custom")