82 lines
3.4 KiB
Python
82 lines
3.4 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']
|
|
|
|
Project = apps.get_model('projects', 'Project')
|
|
Grant = apps.get_model('projects', 'Grant')
|
|
GrantReport = apps.get_model('projects', 'GrantReport')
|
|
|
|
Project.objects.all().delete()
|
|
Grant.objects.all().delete()
|
|
GrantReport.objects.all().delete()
|
|
|
|
ProjectForm = modelform_factory(Project, fields=('name', 'code',
|
|
'iacuc_number',
|
|
'description',
|
|
'sort_order'))
|
|
GrantForm = modelform_factory(Grant, fields=('title', 'code',
|
|
'description', 'sort_order'))
|
|
GrantReportForm = modelform_factory(GrantReport, fields=('grant', 'title',
|
|
'report_type',
|
|
'description',
|
|
'due_date',
|
|
'submitted_date',
|
|
'sort_order'))
|
|
|
|
for r in c.execute('SELECT * FROM tbl_lu_projects;'):
|
|
form = ProjectForm(dict(name=r[1], code=r[2],
|
|
iacuc_number=r[3], description=r[4],
|
|
sort_order=int(r[5])))
|
|
if form.is_valid():
|
|
Project.objects.create(id=r[0], **form.cleaned_data)
|
|
else:
|
|
print('project', r[0:], form.errors.as_data())
|
|
|
|
for r in c.execute('SELECT * FROM tbl_lu_grants;'):
|
|
form = GrantForm(dict(title=r[1], code=r[2], description=r[3], sort_order=r[4]))
|
|
if form.is_valid():
|
|
Grant.objects.create(id=r[0], **form.cleaned_data)
|
|
else:
|
|
print('grant', r[0:], form.errors.as_data())
|
|
|
|
for r in c.execute('SELECT * FROM tbl_hash_project_grants;'):
|
|
p = Project.objects.get(id=r[0])
|
|
g = Grant.objects.get(id=r[1])
|
|
p.grants.add(g)
|
|
p.save()
|
|
|
|
for r in c.execute('SELECT *, report_due_date AS "due_date [dtdt]" FROM tbl_lu_grant_reports;'):
|
|
form = GrantReportForm(dict(grant=r[0], title=r[1], report_type=r[2],
|
|
description=r[3], due_date=r[8],
|
|
submitted_date=r[5], sort_order=r[7]))
|
|
if form.is_valid():
|
|
form.save() # No PK field in Andre's file
|
|
else:
|
|
print('grant report', r[0:], form.errors.as_data())
|
|
|
|
|
|
def rollback(apps, schema_editor):
|
|
Project = apps.get_model('projects', 'Project')
|
|
Grant = apps.get_model('projects', 'Grant')
|
|
GrantReport = apps.get_model('projects', 'GrantReport')
|
|
|
|
for model in [Project, Grant, GrantReport]:
|
|
model.objects.all().delete()
|
|
|
|
dependencies = [
|
|
('projects', '0001_initial'),
|
|
]
|
|
|
|
operations = [
|
|
migrations.RunPython(migrate, rollback),
|
|
]
|