From 6b7faf392685845de71a3fcd35ec6bd03dcc1ddb Mon Sep 17 00:00:00 2001
From: Henrik Levkowetz <henrik@levkowetz.com>
Date: Mon, 28 Mar 2016 19:32:49 +0000
Subject: [PATCH] Added migrations for the submit check display work.  -
 Legacy-Id: 10998

---
 .../migrations/0007_submission_draft.py       | 21 ++++++++++++
 .../0008_data_for_submission_draft_field.py   | 31 ++++++++++++++++++
 .../migrations/0009_submissioncheck_symbol.py | 20 ++++++++++++
 .../0010_data_set_submission_check_symbol.py  | 32 +++++++++++++++++++
 4 files changed, 104 insertions(+)
 create mode 100644 ietf/submit/migrations/0007_submission_draft.py
 create mode 100644 ietf/submit/migrations/0008_data_for_submission_draft_field.py
 create mode 100644 ietf/submit/migrations/0009_submissioncheck_symbol.py
 create mode 100644 ietf/submit/migrations/0010_data_set_submission_check_symbol.py

diff --git a/ietf/submit/migrations/0007_submission_draft.py b/ietf/submit/migrations/0007_submission_draft.py
new file mode 100644
index 000000000..92fc98198
--- /dev/null
+++ b/ietf/submit/migrations/0007_submission_draft.py
@@ -0,0 +1,21 @@
+# -*- coding: utf-8 -*-
+from __future__ import unicode_literals
+
+from django.db import models, migrations
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('doc', '0012_auto_20160207_0537'),
+        ('submit', '0006_remove_submission_idnits_message'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='submission',
+            name='draft',
+            field=models.ForeignKey(null=True, blank=True, to='doc.Document'),
+            preserve_default=True,
+        ),
+    ]
diff --git a/ietf/submit/migrations/0008_data_for_submission_draft_field.py b/ietf/submit/migrations/0008_data_for_submission_draft_field.py
new file mode 100644
index 000000000..fe1ff10e5
--- /dev/null
+++ b/ietf/submit/migrations/0008_data_for_submission_draft_field.py
@@ -0,0 +1,31 @@
+# -*- coding: utf-8 -*-
+from __future__ import unicode_literals
+
+from django.db import migrations
+
+def populate_submission_draft(apps, schema_editor):
+    Submission = apps.get_model('submit', 'Submission')
+    Document = apps.get_model('doc', 'Document')
+    print("")
+    for submission in Submission.objects.filter(state_id='posted'):
+        if submission.draft == None:
+            try:
+                draft = Document.objects.get(name=submission.name)
+            except Document.DoesNotExist:
+                print( "Failed to find %s-%s" % (submission.name, submission.rev))
+                continue
+            submission.draft = draft
+            submission.save()
+
+def backward(apps, schema_editor):
+    pass                                # nothing to do
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('submit', '0007_submission_draft'),
+    ]
+
+    operations = [
+        migrations.RunPython(populate_submission_draft, backward)
+    ]
diff --git a/ietf/submit/migrations/0009_submissioncheck_symbol.py b/ietf/submit/migrations/0009_submissioncheck_symbol.py
new file mode 100644
index 000000000..e6f7178d2
--- /dev/null
+++ b/ietf/submit/migrations/0009_submissioncheck_symbol.py
@@ -0,0 +1,20 @@
+# -*- coding: utf-8 -*-
+from __future__ import unicode_literals
+
+from django.db import models, migrations
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('submit', '0008_data_for_submission_draft_field'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='submissioncheck',
+            name='symbol',
+            field=models.CharField(default=b'', max_length=64),
+            preserve_default=True,
+        ),
+    ]
diff --git a/ietf/submit/migrations/0010_data_set_submission_check_symbol.py b/ietf/submit/migrations/0010_data_set_submission_check_symbol.py
new file mode 100644
index 000000000..886b83772
--- /dev/null
+++ b/ietf/submit/migrations/0010_data_set_submission_check_symbol.py
@@ -0,0 +1,32 @@
+# -*- coding: utf-8 -*-
+from __future__ import unicode_literals
+
+from django.db import migrations
+from ietf.submit.checkers import DraftIdnitsChecker, DraftYangChecker
+
+def set_submission_check_symbol(apps, schema_editor):
+    SubmissionCheck = apps.get_model('submit','SubmissionCheck')
+    for s in SubmissionCheck.objects.all():
+        if not s.symbol:
+            if s.checker == "idnits check":
+                s.symbol = DraftIdnitsChecker.symbol
+            if s.checker == 'yang validation':
+                s.symbol = DraftYangChecker.symbol
+            s.save()
+
+def backward(apps, schema_editor):
+    SubmissionCheck = apps.get_model('submit','SubmissionCheck')
+    for s in SubmissionCheck.objects.all():
+        if s.symbol != "":
+            s.symbol = ""
+            s.save()
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('submit', '0009_submissioncheck_symbol'),
+    ]
+
+    operations = [
+        migrations.RunPython(set_submission_check_symbol, backward)
+    ]