on event draft option

oarepo · Oct 2, 2024 · e37ed53 · e37ed53
1 parent 1f8730c
commit e37ed53
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/oarepo_doi/services/components/__init__.py b/oarepo_doi/services/components/__init__.py
@@ -43,7 +43,7 @@ def create(self, identity, data=None, record=None, **kwargs):
             create_doi(self, record, data, None)
 
     def update_draft(self, identity, data=None, record=None, **kwargs):
-        if self.mode == "AUTOMATIC_DRAFT" or self.mode == "ON_EVENT":
+        if self.mode == "AUTOMATIC_DRAFT" or self.mode == "ON_EVENT_DRAFT":
             slug = community_slug_for_credentials(
                 record.parent["communities"].get("default", None)
             )
@@ -56,6 +56,7 @@ def update(self, identity, data=None, record=None, **kwargs):
             self.mode == "AUTOMATIC_DRAFT"
             or self.mode == "AUTOMATIC"
             or self.mode == "ON_EVENT"
+            or self.mode == "ON_EVENT_DRAFT"
         ):
             slug = community_slug_for_credentials(
                 record.parent["communities"].get("default", None)
@@ -70,7 +71,7 @@ def publish(self, identity, data=None, record=None, **kwargs):
             )
             self.credentials(slug)
             create_doi(self, record, record, "publish")
-        if self.mode == "AUTOMATIC_DRAFT":
+        if self.mode == "AUTOMATIC_DRAFT" or self.mode == "ON_EVENT_DRAFT":
             slug = community_slug_for_credentials(
                 record.parent["communities"].get("default", None)
             )