WIP

ccdb/projects/migrations/0005_DATA_initial.py

@@ -0,0 +1,83 @@
+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:
+            print('no 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', '0004_initial_grantreport'),
+    ]
+
+    operations = [
+        migrations.RunPython(migrate, rollback),
+    ]