Imagens no script de verificacao de arquivos media

scripts/verificacao_arquivos_faltando_media.py

@@ -2,15 +2,41 @@
 
 # Dependência:
 # pip install terminaltables
-
-from terminaltables import AsciiTable
+from os.path import isfile
 
 from datetime import datetime
-from os.path import isfile
+from django.contrib.contenttypes.models import ContentType
+from django.core.urlresolvers import reverse
+from terminaltables import AsciiTable
 
+from sigi.apps.casas.models import CasaLegislativa
 from sigi.apps.convenios.models import Anexo as AnexoConvenios
-from sigi.apps.ocorrencias.models import Anexo as AnexoOcorrencias
 from sigi.apps.diagnosticos.models import Anexo as AnexoDiagnosticos
+from sigi.apps.ocorrencias.models import Anexo as AnexoOcorrencias
+from sigi.apps.parlamentares.models import Parlamentar
+from sigi.apps.servidores.models import Servidor
+
+
+def print_table(msg, relacao):
+    print "\n%s:\n" % msg
+    table = AsciiTable(relacao)
+    print table.table
+
+
+# IMAGENS FALTANDO
+imagens_faltando = [[u"SITUAÇÃO DO ARQUIVO DA FOTO  ", "URL", "OBJETO"]]
+for cl in (CasaLegislativa, Parlamentar, Servidor):
+    content_type = ContentType.objects.get_for_model(cl)
+    url_base = "admin:%s_%s_change" % (content_type.app_label, content_type.model)
+    for a in cl.objects.all():
+        if a.foto:
+            url = reverse(url_base, args=(a.pk,))
+            imagens_faltando.append([
+                "PRESENTE" if isfile(a.foto.path) else "FALTANDO",
+                url,
+                unicode(a)])
+
+print_table(u"Relação de todas as fotos de Casas Legislativas, Parlamentares e Servidores", imagens_faltando)
 
 
 novos_faltando = []
@@ -18,8 +44,7 @@ antigos_faltando = []
 inicio = datetime(2014, 9, 1)
 
 for cl in (AnexoConvenios, AnexoOcorrencias, AnexoDiagnosticos):
-    todos = cl.objects.all()
-    for a in todos:
+    for a in cl.objects.all():
         if not isfile(a.arquivo.path):
             nome = a.arquivo.name.split('/')[-1]
             if a.data_pub > inicio:
@@ -37,11 +62,5 @@ for cl in (AnexoConvenios, AnexoOcorrencias, AnexoDiagnosticos):
 novos_faltando = [["UF", "Arquivo"]] + sorted(novos_faltando)
 antigos_faltando = [['URL DO CONVENIO', 'NOME DO ARQUIVO', 'DATA']] + sorted(antigos_faltando)
 
-
-def print_table(msg, relacao):
-    print "\n%s:\n" % msg
-    table = AsciiTable(relacao)
-    print table.table
-
 print_table(u'Anexos de Novas Ocorrências faltando (desde %s)' % inicio, novos_faltando)
 print_table(u'Anexos de Convênios Antigos faltando (até %s)' % inicio, antigos_faltando)