58 lines
2.1 KiB
Python
58 lines
2.1 KiB
Python
from django.db import migrations
|
|
from django.forms import modelform_factory
|
|
|
|
from ccdb.utils.data import get_data_sources
|
|
|
|
|
|
class Migration(migrations.Migration):
|
|
def migrate(apps, schema_editor):
|
|
sources = get_data_sources()
|
|
if not sources:
|
|
return
|
|
|
|
c = sources['db0']
|
|
|
|
ProcessType = apps.get_model('processing', 'ProcessType')
|
|
Reagent = apps.get_model('processing', 'Reagent')
|
|
Flaw = apps.get_model('processing', 'Flaw')
|
|
Processing = apps.get_model('processing', 'Processing')
|
|
|
|
for model in [Processing, Flaw, Reagent, ProcessType]:
|
|
model.objects.all().delete()
|
|
|
|
ProcessTypeForm = modelform_factory(ProcessType, fields='__all__')
|
|
ReagentForm = modelform_factory(Reagent, fields='__all__')
|
|
|
|
for r in c.execute('SELECT * FROM tbl_lu_process_types;'):
|
|
form = ProcessTypeForm(dict(name=r[1], code=r[2],
|
|
description=r[3],
|
|
sort_order=int(r[4]) if r[4] else None))
|
|
if form.is_valid():
|
|
ProcessType.objects.create(id=r[0], **form.cleaned_data)
|
|
else:
|
|
print('process type', r[0:], form.errors.as_data())
|
|
|
|
for r in c.execute('SELECT * FROM tbl_lu_reagents;'):
|
|
form = ReagentForm(dict(name=r[1], code=r[2],
|
|
reagent_class=r[3],
|
|
sort_order=int(r[4]) if r[4] else None))
|
|
if form.is_valid():
|
|
Reagent.objects.create(id=r[0], **form.cleaned_data)
|
|
else:
|
|
print('reagent', r[0:], form.errors.as_data())
|
|
|
|
def rollback(apps, schema_editor):
|
|
ProcessType = apps.get_model('processing', 'ProcessType')
|
|
Reagent = apps.get_model('processing', 'Reagent')
|
|
|
|
for model in [Reagent, ProcessType]:
|
|
model.objects.all().delete()
|
|
|
|
dependencies = [
|
|
('processing', '0001_initial'),
|
|
('species', '0002_DATA_initial'),
|
|
]
|
|
|
|
operations = [
|
|
migrations.RunPython(migrate, rollback),
|
|
]
|