Index newly ingested volume

readux_ingest_ecds/models.py

@@ -9,6 +9,7 @@
 from .services.metadata_services import metadata_from_file, clean_metadata
 from .helpers import get_iiif_models
 from .storages import TmpStorage
+from .mail import send_email_on_success, send_email_on_failure
 
 Manifest = get_iiif_models()['Manifest']
 ImageServer = get_iiif_models()['ImageServer']
@@ -213,6 +214,22 @@ def create_canvases(self):
 
         upload_trigger_file(self.trigger_file)
 
+    def success(self):
+        LOGGER.info(f'SUCCESS!!! {self.manifest.pid}')
+        send_email_on_success(creator=self.creator, manifest=self.manifest)
+        self.manifest.save()
+        if os.environ["DJANGO_ENV"] != 'test':
+            from apps.iiif.manifests.documents import ManifestDocument
+            index = ManifestDocument()
+            index.update(self.manifest, True, 'index')
+        self.delete()
+
+    def failure(self, exc):
+        LOGGER.info(f'FAIL!!! {self.manifest.pid}')
+        send_email_on_failure(bundle=self.bundle.name, creator=self.creator, exception=str(exc))
+        self.delete()
+
+
 class Bulk(models.Model):
     id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
     collections = models.ManyToManyField(

readux_ingest_ecds/tasks.py

@@ -37,10 +37,7 @@ def on_success(self, retval, task_id, args, kwargs):
             None: The return value of this handler is ignored.
         """
         ingest = Local.objects.get(id=args[0])
-        LOGGER.info(f'SUCCESS!!! {ingest.manifest.pid}')
-        send_email_on_success(creator=ingest.creator, manifest=ingest.manifest)
-        ingest.manifest.save()
-        ingest.delete()
+        ingest.success()
 
     def on_failure(self, exc, task_id, args, kwargs, einfo):
         """Error handler.
@@ -58,9 +55,7 @@ def on_failure(self, exc, task_id, args, kwargs, einfo):
             None: The return value of this handler is ignored.
         """
         ingest = Local.objects.get(id=args[0])
-        LOGGER.info(f'FAIL!!! {ingest.manifest.pid}')
-        send_email_on_failure(bundle=ingest.bundle.name, creator=ingest.creator, exception=str(exc))
-        ingest.delete()
+        ingest.failure(exc)
 
 app = Celery('readux_ingest_ecds', result_extended=True)
 app.config_from_object('django.conf:settings')