views.py

from client.forms import ClientForm, AddressForm
from client.models import Client, Address
from django.http import HttpResponseRedirect, HttpResponse
from annoying.decorators import render_to
import csv

# CRUD Cliente

@render_to('index.html')
def index(request, message=''):
    return {'clients': Client.objects.all(), 'message': message}

@render_to('client/profile.html')
def profile(request, id):
    return {'client': Client.objects.get(id=id)}

@render_to('client/create.html')
def create_client(request):
    if request.method == 'POST':
        form = ClientForm(request.POST)
        if form.is_valid():
            c = form.save()
            for i in range(2):
                p = 'phone'+i.__str__()
                c.phone_set.create(number=request.POST[p])
            c.save()
            return HttpResponseRedirect('/')
    else:
        form = ClientForm()
    return {'form': form, 'action': '/client/', 'button': 'Salvar'}

@render_to('client/create.html')
def edit_client(request, id):
    c = Client.objects.get(id=id)
    if request.method == 'POST':
        form = ClientForm(request.POST, instance=c)
        if form.is_valid():
            form.save()
            return HttpResponseRedirect('/')
    else:
        form = ClientForm(instance=c)
    return {'form': form, 'action': '/client/edit/'+id+'/', 'button': 'Editar'}

def delete_client(request, id):
    Client.objects.get(id=id).delete()
    return HttpResponseRedirect('/')

# CRUD Endereco

@render_to('address/list.html')
def list_addresses(request):
    return {'addresses': Address.objects.all()}

@render_to('address/show.html')
def list_address(request, id):
    return {'address': Address.objects.get(id=id)}

@render_to('address/create.html')
def address(request):
    if request.method == 'POST':
        form = AddressForm(request.POST)
        if form.is_valid():
            form.save()
            return HttpResponseRedirect('/client/')
    else:
        form = AddressForm()
    return {'form': form, 'button': 'Criar'}

@render_to('address/create.html')
def edit_address(request, id):
    a = Address.objects.get(id=id)
    if request.method == 'POST':
        form = AddressForm(request.POST, instance=a)
        if form.is_valid():
            form.save()
            return HttpResponseRedirect('/client/')
    else:
        form = AddressForm(instance=a)
    return {'form': form, 'action': '/address/edit/'+id+'/', 'button': 'Editar'}

def delete_address(request, id):
    Address.objects.get(id=id).delete()
    return HttpResponseRedirect('/')

# Exporting data

def export(request):
    response = HttpResponse(content_type='text/csv')
    response['Content-Disposition'] = 'attachment; filename="clientes.csv"'
    writer = csv.writer(response)

    clients = Client.objects.filter()
    writer = csv.writer(response)
    writer.writerow(['Nome', 'Sexo', 'Data de Nascimento', 'Email', 'Fones', 'CPF', 'Endereco'])
    for client in clients:
        name = client.name.encode('utf-8')
        sex = client.sex.encode('utf-8')
        email = client.email.encode('utf-8')
        cpf = client.cpf.encode('utf-8')
        phone = client.get_phones()
        writer.writerow([name, sex, client.birthday, email, phone, cpf, client.address])

    return response