mirror of https://github.com/interlegis/sapl.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
107 lines
3.8 KiB
107 lines
3.8 KiB
|
|
from sapl.comissoes.models import Comissao, Composicao, Reuniao, Participacao
|
|
from sapl.materia.models import DespachoInicial, Relatoria, UnidadeTramitacao
|
|
from sapl.utils import intervalos_tem_intersecao
|
|
|
|
from difflib import SequenceMatcher
|
|
|
|
models_dependentes = [Composicao, DespachoInicial, Relatoria, UnidadeTramitacao, Reuniao]
|
|
|
|
def similar(a, b):
|
|
return SequenceMatcher(None, a, b).ratio()
|
|
|
|
def detecta_duplicados():
|
|
lst_duplicados = []
|
|
comissoes = Comissao.objects.all().order_by('id')
|
|
for c_1 in comissoes:
|
|
c_1_lst = []
|
|
for c_2 in comissoes:
|
|
if similar(c_1.nome,c_2.nome) > 0.9 and c_1.id != c_2.id:
|
|
c_1_lst.append(c_2)
|
|
comissoes = comissoes.exclude(id=c_2.id)
|
|
if c_1_lst:
|
|
c_1_lst.append(c_1)
|
|
comissoes = comissoes.exclude(id=c_1.id)
|
|
lst_duplicados.append(c_1_lst)
|
|
return lst_duplicados
|
|
|
|
|
|
def realoca_autor(principal, secundaria):
|
|
autor_principal = principal.autor.first()
|
|
autor_secundario = secundaria.autor.first()
|
|
for autoria in autor_secundario.autoria_set.all():
|
|
autoria.autor_id = autor_principal
|
|
autoria.save()
|
|
|
|
for proposicao in autor_secundario.proposicao_set.all():
|
|
proposicao.autor_id = autor_principal
|
|
proposicao.save()
|
|
|
|
for autorianorma in autor_secundario.autorianorma_set.all():
|
|
autorianorma.autor_id = autor_principal
|
|
autorianorma.save()
|
|
|
|
for documentoadministrativo in autor_secundario.documentoadministrativo_set.all():
|
|
documentoadministrativo.autor_id = autor_principal
|
|
documentoadministrativo.save()
|
|
|
|
for protocolo in autor_secundario.protocolo_set.all():
|
|
protocolo.autor_id = autor_principal
|
|
protocolo.save()
|
|
|
|
autor_secundario.delete()
|
|
|
|
|
|
def muda_models_dependentes(principal,secundaria):
|
|
for model in models_dependentes:
|
|
for obj_secundario in model.objects.filter(comissao=secundaria):
|
|
repetido = False
|
|
for obj_principal in model.objects.filter(comissao=principal):
|
|
|
|
if model == Composicao and intervalos_tem_intersecao(obj_principal.periodo.data_inicio,
|
|
obj_principal.periodo.data_fim,
|
|
obj_secundario.periodo.data_inicio,
|
|
obj_secundario.periodo.data_fim):
|
|
|
|
prim_participacoes = Participacao.objects.filter(composicao=obj_principal)
|
|
sec_participacoes = Participacao.objects.filter(composicao=obj_secundario)
|
|
for p in sec_participacoes:
|
|
if p in prim_participacoes:
|
|
p.delete()
|
|
else:
|
|
p.composicao = obj_principal
|
|
p.save()
|
|
|
|
elif model == DespachoInicial and obj_principal.materia == obj_secundario.materia:
|
|
repetido =True
|
|
elif model == Reuniao and obj_principal.numero == obj_secundario.numero:
|
|
repetido =True
|
|
else:
|
|
repetido = False
|
|
|
|
if(repetido):
|
|
obj_secundario.comissao = None
|
|
obj_secundario.delete()
|
|
|
|
else:
|
|
obj_secundario.comissao = principal
|
|
obj_secundario.save()
|
|
|
|
|
|
def junta_dulpicados(duplicados):
|
|
principal = duplicados[-1]
|
|
for secundaria in duplicados[:-1]:
|
|
muda_models_dependentes(principal,secundaria)
|
|
realoca_autor(principal, secundaria)
|
|
secundaria.delete()
|
|
|
|
|
|
def main():
|
|
lst_duplicados = detecta_duplicados()
|
|
print('Duplicados encomtrados:\n')
|
|
print(lst_duplicados)
|
|
for c in lst_duplicados:
|
|
junta_dulpicados(c)
|
|
|
|
if __name__ == '__main__':
|
|
main()
|
|
|