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.
61 lines
2.0 KiB
61 lines
2.0 KiB
from django.db import connection
|
|
from django.db import models
|
|
|
|
|
|
import legacy
|
|
from parlamentares.models import (
|
|
Legislatura, SessaoLegislativa, NivelInstrucao)
|
|
|
|
|
|
mappings = (
|
|
(Legislatura, 'num_legislatura', [
|
|
('dat_inicio', 'data_inicio'),
|
|
('dat_fim', 'data_fim'),
|
|
('dat_eleicao', 'data_eleicao'), ]
|
|
),
|
|
(SessaoLegislativa, 'cod_sessao_leg', [
|
|
('num_legislatura', 'legislatura'),
|
|
('num_sessao_leg', 'numero'),
|
|
('tip_sessao_leg', 'tipo'),
|
|
('dat_inicio', 'data_inicio'),
|
|
('dat_fim', 'data_fim'),
|
|
('dat_inicio_intervalo', 'data_inicio_intervalo'),
|
|
('dat_fim_intervalo', 'data_fim_intervalo'), ]
|
|
),
|
|
(NivelInstrucao, 'cod_nivel_instrucao', [
|
|
('des_nivel_instrucao', 'nivel_instrucao')]
|
|
),
|
|
)
|
|
|
|
|
|
def migrate():
|
|
|
|
for model, pk, field_pairs in mappings:
|
|
|
|
print 'Migrating %s...' % model.__name__
|
|
|
|
# clear all model entries
|
|
model.objects.all().delete()
|
|
# resets id sequence
|
|
sql_reset_seq = 'ALTER SEQUENCE %s_id_seq RESTART WITH 1;' % model._meta.db_table
|
|
cursor = connection.cursor()
|
|
cursor.execute(sql_reset_seq)
|
|
|
|
legacy_model = getattr(legacy.models, model.__name__)
|
|
for old in legacy_model.objects.all().order_by(pk):
|
|
old_id = getattr(old, pk)
|
|
new = model()
|
|
while not new.id:
|
|
for old_field, new_field in field_pairs:
|
|
value = getattr(old, old_field)
|
|
# check for a relation
|
|
model_field = model._meta.get_field(new_field)
|
|
if isinstance(model_field, models.ForeignKey):
|
|
value = model_field.related_model.objects.get(id=value)
|
|
setattr(new, new_field, value)
|
|
new.save()
|
|
assert new.id <= old_id, 'New id exceeds old one. Be sure your new table was just created!'
|
|
if new.id < old_id:
|
|
new.delete()
|
|
new = model()
|
|
|
|
|