Browse Source

Corrige try/except

pull/2626/head
Vinícius Cantuária 7 years ago
parent
commit
0ffc8f64bd
  1. 24
      scripts/remove_multiplos_parlamentares.py

24
scripts/remove_multiplos_parlamentares.py

@ -82,31 +82,39 @@ def transfer_purge_congressman(congressman_lists):
for pk in congressman_list[1:]: for pk in congressman_list[1:]:
parlamentar_clonado = Parlamentar.objects.get(pk=pk) parlamentar_clonado = Parlamentar.objects.get(pk=pk)
if parlamentar_principal.biografia and parlamentar_clonado.biografia: if parlamentar_principal.biografia and parlamentar_clonado.biografia:
parlamentar_principal.biografia += \ parlamentar_principal.biografia = \
f'{parlamentar_principal.biografia}\n\n------------------------\n\n{parlamentar_clonado.biografia}' f'{parlamentar_principal.biografia}' \
f'\n\n------------------------\n\n' \
f'{parlamentar_clonado.biografia}'
parlamentar_principal.save()
elif parlamentar_clonado.biografia: elif parlamentar_clonado.biografia:
parlamentar_principal.biografia = parlamentar_clonado.biografia parlamentar_principal.biografia = parlamentar_clonado.biografia
for model in models: for model in models:
for obj in model.objects.filter(parlamentar_id=pk): for obj in model.objects.filter(parlamentar_id=pk):
# TODO: Validar objeto para não repeti-lo no parlamentar principal
obj.parlamentar_id = congressman_list[0] obj.parlamentar_id = congressman_list[0]
obj.save() obj.save()
# TODO: Arrumar try/except
# TODO: Transferir para função de autor # TODO: Transferir para função de autor
try: try:
autor_principal = Autor.objects.get(parlamentar_set=parlamentar_principal) autor_principal = Autor.objects.get(parlamentar_set=parlamentar_principal)
except ObjectDoesNotExist:
try:
autor_clonado = Autor.objects.get(parlamentar_set=parlamentar_clonado) autor_clonado = Autor.objects.get(parlamentar_set=parlamentar_clonado)
for autoria in Autoria.objects.filter(autor=autor_clonado):
autoria.autor = autor_principal
autoria.save()
except ObjectDoesNotExist: except ObjectDoesNotExist:
pass
else:
autor_clonado.object_id = parlamentar_principal.id
autor_clonado.save()
else:
try: try:
autor_clonado = Autor.objects.get(parlamentar_set=parlamentar_clonado) autor_clonado = Autor.objects.get(parlamentar_set=parlamentar_clonado)
parlamentar_principal.autor = autor_clonado
except ObjectDoesNotExist: except ObjectDoesNotExist:
pass pass
else:
for autoria in Autoria.objects.filter(autor=autor_clonado):
autoria.autor = autor_principal
autoria.save()
parlamentar_clonado.delete() parlamentar_clonado.delete()

Loading…
Cancel
Save