From 9f1b05a51846c93d6c50ede8dc405128a201d991 Mon Sep 17 00:00:00 2001
From: Jennifer Richards <jennifer@staff.ietf.org>
Date: Thu, 6 Jul 2023 17:27:38 -0300
Subject: [PATCH] chore: Move/copy RelatedDocuments to rfcs

---
 ...documents.py => 0009_rfc_relateddocuments.py} | 16 +++++++++++++++-
 ietf/doc/migrations/0010_move_rfc_docaliases.py  |  2 +-
 2 files changed, 16 insertions(+), 2 deletions(-)
 rename ietf/doc/migrations/{0009_move_rfc_relateddocuments.py => 0009_rfc_relateddocuments.py} (61%)

diff --git a/ietf/doc/migrations/0009_move_rfc_relateddocuments.py b/ietf/doc/migrations/0009_rfc_relateddocuments.py
similarity index 61%
rename from ietf/doc/migrations/0009_move_rfc_relateddocuments.py
rename to ietf/doc/migrations/0009_rfc_relateddocuments.py
index 2b3dcf23c..a00de56d6 100644
--- a/ietf/doc/migrations/0009_move_rfc_relateddocuments.py
+++ b/ietf/doc/migrations/0009_rfc_relateddocuments.py
@@ -7,7 +7,10 @@ def forward(apps, schema_editor):
     DocAlias = apps.get_model("doc", "DocAlias")
     Document = apps.get_model("doc", "Document")
     RelatedDocument = apps.get_model("doc", "RelatedDocument")
-    for rfc_alias in DocAlias.objects.filter(name__startswith="rfc"):
+    for rfc_alias in DocAlias.objects.filter(name__startswith="rfc").exclude(
+        docs__type__slug="rfc"
+    ):
+        # Move these over to the RFC
         RelatedDocument.objects.filter(
             relationship__slug__in=(
                 "tobcp",
@@ -21,6 +24,17 @@ def forward(apps, schema_editor):
             ),
             source__docalias=rfc_alias,
         ).update(source=Document.objects.get(name=rfc_alias.name))
+        # Duplicate references on the RFC but keep the ones on the draft as well
+        originals = list(
+            RelatedDocument.objects.filter(
+                relationship__slug__in=("refinfo", "refnorm", "refold", "refunk"),
+                source__docalias=rfc_alias,
+            )
+        )
+        for o in originals:
+            o.pk = None
+            o.source = Document.objects.get(name=rfc_alias.name)
+        RelatedDocument.objects.bulk_create(originals)
 
 
 class Migration(migrations.Migration):
diff --git a/ietf/doc/migrations/0010_move_rfc_docaliases.py b/ietf/doc/migrations/0010_move_rfc_docaliases.py
index f7be94412..af12c26a1 100644
--- a/ietf/doc/migrations/0010_move_rfc_docaliases.py
+++ b/ietf/doc/migrations/0010_move_rfc_docaliases.py
@@ -30,7 +30,7 @@ def forward(apps, schema_editor):
 
 class Migration(migrations.Migration):
     dependencies = [
-        ("doc", "0009_move_rfc_relateddocuments"),
+        ("doc", "0009_rfc_relateddocuments"),
     ]
 
     operations = [