diff --git a/arkindex/documents/tasks.py b/arkindex/documents/tasks.py index 2dd9855b9ba7fb4d9400db8973c18d6046354de1..5c2db2fd9fcd327c47beb8b351a882b5bfc9669b 100644 --- a/arkindex/documents/tasks.py +++ b/arkindex/documents/tasks.py @@ -21,7 +21,7 @@ from arkindex.documents.models import ( Transcription, TranscriptionEntity, ) -from arkindex.ponos.models import Task +from arkindex.ponos.models import Artifact, Task from arkindex.process.models import Process, ProcessDatasetSet, ProcessElement, WorkerActivity, WorkerRun from arkindex.training.models import DatasetElement, DatasetSet from arkindex.users.models import User @@ -67,6 +67,8 @@ def corpus_delete(corpus_id: str) -> None: Selection.objects.filter(element__corpus_id=corpus_id), corpus.memberships.all(), corpus.exports.all(), + # Delete task artifacts + Artifact.objects.filter(task__process__corpus_id=corpus_id), # ProcessDatasetSet M2M ProcessDatasetSet.objects.filter(set__dataset__corpus_id=corpus_id), ProcessDatasetSet.objects.filter(process__corpus_id=corpus_id), diff --git a/arkindex/sql_validation/corpus_delete.sql b/arkindex/sql_validation/corpus_delete.sql index 2ebb6b6f75be165e6be9d6bac416a10ec36d74bc..d1abf7227d10e150dff9ea9767008c4dbd459c30 100644 --- a/arkindex/sql_validation/corpus_delete.sql +++ b/arkindex/sql_validation/corpus_delete.sql @@ -152,6 +152,15 @@ DELETE FROM "documents_corpusexport" WHERE "documents_corpusexport"."corpus_id" = '{corpus_id}'::uuid; +DELETE +FROM "ponos_artifact" +WHERE "ponos_artifact"."id" IN + (SELECT U0."id" + FROM "ponos_artifact" U0 + INNER JOIN "ponos_task" U1 ON (U0."task_id" = U1."id") + INNER JOIN "process_process" U2 ON (U1."process_id" = U2."id") + WHERE U2."corpus_id" = '{corpus_id}'::uuid); + DELETE FROM "process_processdatasetset" WHERE "process_processdatasetset"."id" IN diff --git a/arkindex/sql_validation/corpus_delete_top_level_type.sql b/arkindex/sql_validation/corpus_delete_top_level_type.sql index 80ae4c77e85a44b11f33a3ef17803d55a9fc6d7b..2c86ce4af3da5250b848e740e206bfb62638c858 100644 --- a/arkindex/sql_validation/corpus_delete_top_level_type.sql +++ b/arkindex/sql_validation/corpus_delete_top_level_type.sql @@ -156,6 +156,15 @@ DELETE FROM "documents_corpusexport" WHERE "documents_corpusexport"."corpus_id" = '{corpus_id}'::uuid; +DELETE +FROM "ponos_artifact" +WHERE "ponos_artifact"."id" IN + (SELECT U0."id" + FROM "ponos_artifact" U0 + INNER JOIN "ponos_task" U1 ON (U0."task_id" = U1."id") + INNER JOIN "process_process" U2 ON (U1."process_id" = U2."id") + WHERE U2."corpus_id" = '{corpus_id}'::uuid); + DELETE FROM "process_processdatasetset" WHERE "process_processdatasetset"."id" IN