Browse Source

Alterando script de migração

diagnosticos
Camilo Carlos (Estagiario) 15 years ago
parent
commit
4dd1827e94
  1. 77
      etc/migracao/migra.py

77
etc/migracao/migra.py

@ -66,29 +66,29 @@ def migra_assembleias(filename):
for line in reader: for line in reader:
uf = UnidadeFederativa.objects.get(sigla=line[UF_COL]) uf = UnidadeFederativa.objects.get(sigla=line[UF_COL])
municipio = Municipio.objects.get(uf=uf, is_capital=True) municipio = Municipio.objects.get(uf=uf, is_capital=True)
endereco = line[ENDERECO_COL].split('-') aux_end = line[ENDERECO_COL].split('-')
if(endereco.__len__()>1): bairro = ''
bairro = endereco[1] if(aux_end.__len__()>1):
bairro = aux_end[1].replace(' ', '', 1)
else: else:
bairro = '' bairro = ''
casa = CasaLegislativa( casa = CasaLegislativa(
municipio=municipio, municipio=municipio,
nome=line[NOME_COL], nome=line[NOME_COL],
tipo=tipo_casa, tipo=tipo_casa,
logradouro=endereco[0], logradouro=aux_end[0],
bairro=bairro, bairro=bairro,
cep=line[CEP_COL], cep=line[CEP_COL],
email=line[EMAIL_COL], email=line[EMAIL_COL],
pagina_web=line[PAGINA_COL], pagina_web=line[PAGINA_COL],
observacoes=line[OBS_COL], observacoes=line[OBS_COL],
presidente=line[PRESIDENTE_COL],
telefone=line[FONE_1_COL]
) )
if line[UF_COL] == 'DF': if line[UF_COL] == 'DF':
casa.tipo = tipo_casa = TipoCasaLegislativa.objects.filter(sigla='CT').get() casa.tipo = TipoCasaLegislativa.objects.filter(sigla='CT').get()
casa.save() casa.save()
if line[FONE_1_COL]:
fone1 = Telefone(numero=line[FONE_1_COL], tipo='F', content_object=casa)
fone1.save()
if line[FONE_2_COL]: if line[FONE_2_COL]:
fone2 = Telefone(numero=line[FONE_2_COL], tipo='I', content_object=casa) fone2 = Telefone(numero=line[FONE_2_COL], tipo='I', content_object=casa)
fone2.save() fone2.save()
@ -103,23 +103,23 @@ def migra_assembleias(filename):
nota='Telefone da Prefeitura.' nota='Telefone da Prefeitura.'
) )
fone_prefeitura.save() fone_prefeitura.save()
# Presidente será um atributo de casa legislativa
# if line[REPRESENTANTE_COL]: # if line[REPRESENTANTE_COL]:
# representante = Contato(nome=line[REPRESENTANTE_COL], content_object=casa) # representante = Contato(nome=line[REPRESENTANTE_COL], content_object=casa)
# representante.save() # representante.save()
if line[PRESIDENTE_COL]: # if line[PRESIDENTE_COL]:
mesa = MesaDiretora(casa_legislativa=casa) # mesa = MesaDiretora(casa_legislativa=casa)
mesa.save() # mesa.save()
parlamentar = Parlamentar(nome_completo=line[PRESIDENTE_COL], email=line[EMAIL_PRESIDENTE_COL]) # parlamentar = Parlamentar(nome_completo=line[PRESIDENTE_COL], email=line[EMAIL_PRESIDENTE_COL])
parlamentar.save() # parlamentar.save()
cargo_presidente = Cargo.objects.get(descricao__iexact='presidente') # cargo_presidente = Cargo.objects.get(descricao__iexact='presidente')
presidente = MembroMesaDiretora( # presidente = MembroMesaDiretora(
parlamentar=parlamentar, # parlamentar=parlamentar,
cargo=cargo_presidente, # cargo=cargo_presidente,
mesa_diretora=mesa # mesa_diretora=mesa
) # )
presidente.save() # presidente.save()
def migra_casas(filename): def migra_casas(filename):
@ -157,26 +157,23 @@ def migra_casas(filename):
except ValueError: except ValueError:
print ERROR_MSG_1 % (filename, linenum) print ERROR_MSG_1 % (filename, linenum)
continue continue
parlamentar=None
if(line[PRESIDENTE_COL]): aux_end = line[ENDERECO_COL].split('-')
parlamentar = Parlamentar(nome_completo=line[PRESIDENTE_COL], email=line[EMAIL_PRESIDENTE_COL]) bairro = ''
parlamentar.save() if(aux_end.__len__()>1):
endereco = line[ENDERECO_COL].split('-') bairro = aux_end[1].replace(' ', '', 1)
if(endereco.__len__()>1):
bairro = endereco[1]
else:
bairro = ''
casa = CasaLegislativa( casa = CasaLegislativa(
municipio=municipio, municipio=municipio,
nome='Câmara Municipal de ' + line[NOME_COL], nome='Câmara Municipal de ' + line[NOME_COL],
tipo=tipo_casa, tipo=tipo_casa,
logradouro=endereco[0], logradouro=aux_end[0],
bairro=bairro, bairro=bairro,
cep=line[CEP_COL], cep=line[CEP_COL],
email=line[EMAIL_COL], email=line[EMAIL_COL],
pagina_web=line[PAGINA_COL], pagina_web=line[PAGINA_COL],
observacoes=line[OBS_COL], observacoes=line[OBS_COL],
parlamentar=parlamentar, presidente=line[PRESIDENTE_COL],
telefone=line[FONE_1_COL]
) )
try: try:
@ -186,9 +183,6 @@ def migra_casas(filename):
print ERROR_MSG_0 % (filename, linenum) print ERROR_MSG_0 % (filename, linenum)
continue continue
if line[FONE_1_COL]:
fone1 = Telefone(numero=line[FONE_1_COL], tipo='F', content_object=casa)
fone1.save()
if line[FONE_2_COL]: if line[FONE_2_COL]:
fone2 = Telefone(numero=line[FONE_2_COL], tipo='I', content_object=casa) fone2 = Telefone(numero=line[FONE_2_COL], tipo='I', content_object=casa)
fone2.save() fone2.save()
@ -208,19 +202,6 @@ def migra_casas(filename):
representante = Contato(nome=line[REPRESENTANTE_COL], content_object=casa) representante = Contato(nome=line[REPRESENTANTE_COL], content_object=casa)
representante.save() representante.save()
if line[PRESIDENTE_COL]:
mesa = MesaDiretora(casa_legislativa=casa)
mesa.save()
parlamentar = Parlamentar(nome_completo=line[PRESIDENTE_COL], email=line[EMAIL_PRESIDENTE_COL])
parlamentar.save()
cargo_presidente = Cargo.objects.get(descricao__iexact='presidente')
presidente = MembroMesaDiretora(
parlamentar=parlamentar,
cargo=cargo_presidente,
mesa_diretora=mesa
)
presidente.save()
def migra_cnpj(filename): def migra_cnpj(filename):
# identificação das colunas no arquivo CSV # identificação das colunas no arquivo CSV
COD_TSE_COL = 0 COD_TSE_COL = 0

Loading…
Cancel
Save