vNext: Add migration step for publication-dissertation

docs/releases/v14/migrate_thesis_to_dissertation.py

@@ -0,0 +1,323 @@
+# SPDX-FileCopyrightText: 2025 CERN.
+# SPDX-FileCopyrightText: 2026 Graz University of Technology.
+# SPDX-License-Identifier: MIT
+
+"""Reference helpers for the "Thesis" / "Dissertation" resource type change (v14).
+
+These functions are reference material, not a runnable script. They are here so
+you can copy and adapt the parts that fit your instance when carrying out the
+*optional* resource type change described in the v14 upgrade guide. Read them,
+lift what you need into your own `invenio shell` snippet or script, and always
+test against a copy of your data first.
+
+Two independent operations are shown:
+
+- `run_update_for_resource_type` rewrites every published record and unpublished
+  draft with resource type publication-thesis to publication-dissertation (in
+  metadata.resource_type and in any metadata.related_identifiers), going through
+  the service layer so DataCite DOI metadata is re-registered and records are
+  re-indexed.
+
+- `run_update_doi_metadata_for_resource_type` re-registers the DataCite DOI
+  metadata of every published record of a resource type, without changing the
+  record. Use it when you keep your own resource type id but changed its
+  props.datacite_general (e.g. to "Dissertation") in your vocabulary: the DataCite
+  serializer reads props.datacite_general live from the vocabulary, so only a DOI
+  update is needed.
+
+The resource type rewrite has been tested for the following scenarios:
+1. Draft with resource type publication-thesis
+2. Record with resource type publication-thesis with a DOI and no draft
+3. Record with resource type publication-thesis with no DOI and an existing draft
+4. Records with multiple versions
+"""
+
+from click import secho
+from invenio_access.permissions import system_identity
+from invenio_db import db
+from invenio_drafts_resources.resources.records.errors import DraftNotCreatedError
+from invenio_rdm_records.proxies import current_rdm_records_service as records_service
+from invenio_rdm_records.services.errors import RecordDeletedException
+from invenio_search.api import RecordsSearchV2
+
+
+def run_upgrade(migrate_record, migrate_draft):
+    """Run upgrade on selected records and drafts.
+
+    Args:
+        migrate_record (callable): Function to migrate a record.
+        migrate_draft (callable): Function to migrate a draft.
+    """
+    errored_record_ids = []
+    errored_draft_ids = []
+
+    # Handle published records
+    published_records = (
+        RecordsSearchV2(index=records_service.record_cls.index._name)
+        .filter(
+            "query_string",
+            query="metadata.resource_type.id:publication-thesis OR metadata.related_identifiers.resource_type.id:publication-thesis",
+        )
+        .source(["id"])
+        .scan()
+    )  # Only need to fetch the record IDs to make the query faster
+    # Convert the search results to a list to avoid keeping the scroll context open, as it errors out after 15 minutes
+    published_record_ids = [result["id"] for result in published_records]
+    for record_id in published_record_ids:
+        try:
+            migrate_record(record_id)
+        except Exception as error:
+            secho(f"> Error {repr(error)}", fg="red")
+            secho(f"Record {record_id} failed to update", fg="red")
+            errored_record_ids.append((record_id, error))
+
+    # Handle draft records
+    draft_records = (
+        RecordsSearchV2(index=records_service.draft_cls.index._name)
+        .filter("term", has_draft=False)
+        .filter(
+            "query_string",
+            query="metadata.resource_type.id:publication-thesis OR metadata.related_identifiers.resource_type.id:publication-thesis",
+        )
+        .source(["id"])
+        .scan()
+    )
+    # Convert the search results to a list to avoid keeping the scroll context open, as it errors out after 15 minutes
+    draft_record_ids = [result["id"] for result in draft_records]
+    for draft_id in draft_record_ids:
+        try:
+            migrate_draft(draft_id)
+        except Exception as error:
+            secho(f"> Error {repr(error)}", fg="red")
+            secho(f"Draft {draft_id} failed to update", fg="red")
+            errored_draft_ids.append((draft_id, error))
+
+    if len(errored_record_ids) > 0:
+        secho(f"Errored record IDs: {errored_record_ids}", fg="red")
+    else:
+        secho("records have been updated successfully", fg="green")
+
+    if len(errored_draft_ids) > 0:
+        secho(f"Errored draft IDs: {errored_draft_ids}", fg="red")
+    else:
+        secho("drafts have been updated successfully", fg="green")
+
+
+def run_update_for_resource_type():
+    """Run update for resource type."""
+
+    def migrate_resource_type_in_record(record_id):
+        """
+        Update resource type from publication-thesis to publication-dissertation.
+
+        We go through the service layer to automatically trigger the DOI update and re-indexing.
+        """
+        secho(f"Updating resource type for record {record_id}", fg="yellow")
+        record = records_service.read(system_identity, record_id, include_deleted=True)
+        if record.data["metadata"]["resource_type"][
+            "id"
+        ] != "publication-thesis" and not any(
+            related_identifier.get("resource_type", {}).get("id")
+            == "publication-thesis"
+            for related_identifier in record.data["metadata"].get(
+                "related_identifiers", []
+            )
+        ):
+            secho(
+                f"Skipping record <{record.id}> because it doesn't have resource-type 'publication-thesis'!",
+                fg="yellow",
+            )
+            return
+
+        try:
+            draft = records_service.read_draft(system_identity, record.id)
+            # Step 1: Update the resource type in the record via low-level API
+            # We need to make sure we don't publish the record with different metadata
+            secho(
+                f"Record <{record.id}> has an existing draft <{draft.id}>! Updating record via low-level API.",
+                fg="yellow",
+            )
+            # Update the record directly without affecting the draft
+            if (
+                record._record["metadata"]["resource_type"]["id"]
+                == "publication-thesis"
+            ):
+                record._record["metadata"]["resource_type"][
+                    "id"
+                ] = "publication-dissertation"
+            for related_identifier in record._record["metadata"].get(
+                "related_identifiers", []
+            ):
+                if (
+                    related_identifier.get("resource_type", {}).get("id")
+                    == "publication-thesis"
+                ):
+                    related_identifier["resource_type"][
+                        "id"
+                    ] = "publication-dissertation"
+            # Save the record changes and reindex
+            secho(
+                f"Record <{record.id}> has been updated... committing changes.",
+                fg="green",
+            )
+            record._record.commit()
+            # Step 2: Update the resource type in the draft
+            if draft._record["metadata"]["resource_type"]["id"] == "publication-thesis":
+                draft._record["metadata"]["resource_type"][
+                    "id"
+                ] = "publication-dissertation"
+            for related_identifier in draft._record["metadata"].get(
+                "related_identifiers", []
+            ):
+                if (
+                    related_identifier.get("resource_type", {}).get("id")
+                    == "publication-thesis"
+                ):
+                    related_identifier["resource_type"][
+                        "id"
+                    ] = "publication-dissertation"
+            # After updating the record, update the draft's fork_version_id to match the record's new version_id, to avoid conflicts when publishing
+            draft._record.fork_version_id = record._record.revision_id
+            draft._record.commit()
+            # Commit the changes for both the record and the draft in one transaction
+            db.session.commit()
+            records_service.indexer.index(record._record)
+            records_service.draft_indexer.index(draft._record)
+            secho(f"Draft <{draft.id}> has been updated successfully.", fg="green")
+            # Update DOI metadata if record has DOI
+            if (record._record.get("pids") or {}).get("doi"):
+                records_service.pids.register_or_update(
+                    system_identity, record.id, "doi", parent=False
+                )
+                secho(
+                    f"DOI metadata for record {record.id} has been updated successfully.",
+                    fg="green",
+                )
+        except DraftNotCreatedError:
+            # If the draft didn't exist, we simply edit and publish the record
+            draft = records_service.edit(system_identity, record.id)
+            if draft.data["metadata"]["resource_type"]["id"] == "publication-thesis":
+                draft.data["metadata"]["resource_type"][
+                    "id"
+                ] = "publication-dissertation"
+            for related_identifier in draft.data["metadata"].get(
+                "related_identifiers", []
+            ):
+                if (
+                    related_identifier.get("resource_type", {}).get("id")
+                    == "publication-thesis"
+                ):
+                    related_identifier["resource_type"][
+                        "id"
+                    ] = "publication-dissertation"
+            updated_draft = records_service.update_draft(
+                system_identity, draft.id, draft.data
+            )
+            record = records_service.publish(system_identity, updated_draft.id)
+        except RecordDeletedException:
+            # If the draft was deleted, we ignore it
+            # In the future, we should add include_deleted to read_draft and update the draft metadata in these cases
+            secho(f"Draft <{draft.id}> has been deleted, skipping...", fg="yellow")
+
+        secho(f"Record <{record.id}> has been updated successfully.", fg="green")
+
+    def migrate_resource_type_in_draft(draft_id):
+        """
+        Update resource type from publication-thesis to publication-dissertation.
+
+        We go through the service layer to automatically trigger the DOI update and re-indexing.
+        """
+        secho(f"Updating resource type for draft {draft_id}", fg="yellow")
+        draft = records_service.edit(system_identity, draft_id)
+        if draft.data["metadata"]["resource_type"][
+            "id"
+        ] != "publication-thesis" and not any(
+            related_identifier.get("resource_type", {}).get("id")
+            == "publication-thesis"
+            for related_identifier in draft.data["metadata"].get(
+                "related_identifiers", []
+            )
+        ):
+            secho(
+                f"Skipping draft <{draft.id}> because it doesn't have resource-type 'publication-thesis'!",
+                fg="yellow",
+            )
+            return
+
+        if draft.data["metadata"]["resource_type"]["id"] == "publication-thesis":
+            draft.data["metadata"]["resource_type"]["id"] = "publication-dissertation"
+        for related_identifier in draft.data["metadata"].get("related_identifiers", []):
+            if (
+                related_identifier.get("resource_type", {}).get("id")
+                == "publication-thesis"
+            ):
+                related_identifier["resource_type"]["id"] = "publication-dissertation"
+        updated_draft = records_service.update_draft(
+            system_identity, draft.id, draft.data
+        )
+        secho(f"Draft <{updated_draft.id}> has been updated successfully.", fg="green")
+
+    secho("Resource type update has started.", fg="green")
+
+    run_upgrade(
+        migrate_resource_type_in_record,
+        migrate_resource_type_in_draft,
+    )
+
+    secho("Resource type update has finished.", fg="green")
+
+
+def run_update_doi_metadata_for_resource_type(resource_type_id="publication-thesis"):
+    """Re-register DataCite DOI metadata for published records of a resource type.
+
+    Use this after changing the resource type's props.datacite_general in your
+    vocabulary and reloading the fixture. The DataCite serializer reads
+    props.datacite_general live from the vocabulary, so re-registering each DOI
+    pushes the new value to DataCite; the records are not otherwise changed. Only
+    version DOIs are refreshed here (pass parent=True to also update the concept
+    DOI).
+    """
+    secho(
+        f"DOI metadata update for resource type {resource_type_id} has started.",
+        fg="green",
+    )
+
+    errored_record_ids = []
+
+    published_records = (
+        RecordsSearchV2(index=records_service.record_cls.index._name)
+        .filter(
+            "query_string",
+            query=f"metadata.resource_type.id:{resource_type_id} OR metadata.related_identifiers.resource_type.id:{resource_type_id}",
+        )
+        .source(["id"])
+        .scan()
+    )
+    # Convert the search results to a list to avoid keeping the scroll context open, as it errors out after 15 minutes
+    published_record_ids = [result["id"] for result in published_records]
+    for record_id in published_record_ids:
+        try:
+            record = records_service.read(system_identity, record_id)
+            # Only records that already have a DOI can have their metadata updated
+            if not (record._record.get("pids") or {}).get("doi"):
+                secho(
+                    f"Skipping record <{record_id}> because it has no DOI!",
+                    fg="yellow",
+                )
+                continue
+            records_service.pids.register_or_update(
+                system_identity, record_id, "doi", parent=False
+            )
+            secho(
+                f"DOI metadata for record {record_id} has been updated successfully.",
+                fg="green",
+            )
+        except Exception as error:
+            secho(f"> Error {repr(error)}", fg="red")
+            secho(f"Record {record_id} failed to update", fg="red")
+            errored_record_ids.append((record_id, error))
+
+    if len(errored_record_ids) > 0:
+        secho(f"Errored record IDs: {errored_record_ids}", fg="red")
+    else:
+        secho("DOI metadata has been updated successfully", fg="green")