From 37c176101385a5add2ee86938173ffae874963f8 Mon Sep 17 00:00:00 2001
From: Breno Teixeira <brenotx@gmail.com>
Date: Mon, 24 Jun 2013 09:51:18 -0300
Subject: [PATCH] =?UTF-8?q?adi=C3=A7=C3=A3o=20da=20op=C3=A7=C3=A3o=20de=20?=
 =?UTF-8?q?exportar=20contatos=20intelegis=20no=20formato=20CSV?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 sigi/apps/casas/templates/casas/carrinho.html | 22 ++++++++++++++++++-
 sigi/apps/casas/views.py                      | 19 +++++++++++-----
 2 files changed, 34 insertions(+), 7 deletions(-)

diff --git a/sigi/apps/casas/templates/casas/carrinho.html b/sigi/apps/casas/templates/casas/carrinho.html
index 04e5a5f..08c8846 100644
--- a/sigi/apps/casas/templates/casas/carrinho.html
+++ b/sigi/apps/casas/templates/casas/carrinho.html
@@ -188,9 +188,29 @@
 				</li>
 			</ul>
 			</fieldset>
+
+			<fieldset><legend>Contato Interlegis</legend>
+			<ul id="sortable" class="tabs-conteudo">
+				<li><span class="ui-icon ui-icon-arrowthick-2-n-s"></span>
+					<input type="checkbox" name="itens_csv_selected"
+						value="Nome contato" class="action-select" checked="checked" />
+					<label>Nome</label>
+				</li>
+				<li><span class="ui-icon ui-icon-arrowthick-2-n-s"></span>
+					<input type="checkbox" name="itens_csv_selected"
+						value="Cargo contato" class="action-select" checked="checked" />
+					<label>Cargo</label>
+				</li>
+				<li><span class="ui-icon ui-icon-arrowthick-2-n-s"></span>
+					<input type="checkbox" name="itens_csv_selected"
+						value="Email contato" class="action-select" checked="checked" />
+					<label>Email</label>
+				</li>
+			</ul>
+			</fieldset>
 			<ul class="botoes">
 				<li><input type="submit" value="Exportar CSV" /></li>
-			</ul>			
+			</ul>
 		</form>	
 	</div>
 	<div id="tabs-4">
diff --git a/sigi/apps/casas/views.py b/sigi/apps/casas/views.py
index be79661..a114c96 100644
--- a/sigi/apps/casas/views.py
+++ b/sigi/apps/casas/views.py
@@ -3,7 +3,7 @@ from django.http import HttpResponse, HttpResponseRedirect
 from django.shortcuts import render_to_response
 from geraldo.generators import PDFGenerator
 
-from sigi.apps.casas.models import CasaLegislativa
+from sigi.apps.casas.models import CasaLegislativa, Funcionario
 from sigi.apps.casas.reports import CasasLegislativasLabels
 from sigi.apps.casas.reports import CasasLegislativasLabelsSemPresidente
 from sigi.apps.casas.reports import CasasLegislativasReport
@@ -302,7 +302,8 @@ def export_csv(request):
     response = HttpResponse(mimetype='text/csv')
     response['Content-Disposition'] = 'attachment; filename=casas.csv'
     
-    csv_writer = csv.writer(response)
+    writer = csv.writer(response)
+
     casas = carrinhoOrGet_for_qs(request)
     if not casas or not request.POST:
         return HttpResponseRedirect('../')
@@ -311,11 +312,11 @@ def export_csv(request):
     atributos2 = [s.encode("utf-8") for s in atributos]
 
     try:
-        atributos2.insert(atributos2.index(u'Município'), u'UF')
+        atributos2.insert(atributos2.index('Município'), u'UF')
     except ValueError:
         pass
         
-    csv_writer.writerow(atributos2)
+    writer.writerow(atributos2)
         
     for casa in casas:
         lista = []
@@ -330,7 +331,7 @@ def export_csv(request):
                 lista.append(casa.nome.encode("utf-8"))
             elif u"Município" == atributo:
                 lista.append(unicode(casa.municipio.uf.sigla).encode("utf-8"))
-                lista.append(unicode(casa.municipio.nome).encode("utf-8"))            
+                lista.append(unicode(casa.municipio.nome).encode("utf-8"))
             elif u"Presidente" == atributo:
 		#TODO: Esse encode deu erro em 25/04/2012. Comentei para que o usuário pudesse continuar seu trabalho
                 # É preciso descobrir o porque do erro e fazer a correção definitiva.
@@ -352,9 +353,15 @@ def export_csv(request):
                 lista.append(casa.total_parlamentares)
             elif u"Última alteração de endereco" == atributo:
                 lista.append(casa.ult_alt_endereco)
+            elif u"Nome contato" == atributo:
+                lista.append(casa.funcionario_set.filter(setor="contato_interlegis")[0].nome.encode("utf-8"))
+            elif u"Cargo contato" == atributo:
+                lista.append(casa.funcionario_set.filter(setor="contato_interlegis")[0].cargo.encode("utf-8"))
+            elif u"Email contato" == atributo:
+                lista.append(casa.funcionario_set.filter(setor="contato_interlegis")[0].email.encode("utf-8"))
             else:
                 pass
                                 
-        csv_writer.writerow(lista)
+        writer.writerow(lista)
     
     return response