Move the search_drafts options override to startup from ext

invenio_config_tuw/ext.py

@@ -10,7 +10,6 @@
 from flask.config import Config
 from flask_minify import Minify
 from flask_security.signals import user_registered
-from invenio_rdm_records.services.search_params import MyDraftsParam
 
 from . import config
 from .auth.utils import auto_trust_user
@@ -48,18 +47,6 @@ def auto_trust_new_user(sender, user, **kwargs):
     auto_trust_user(user)
 
 
-def override_search_drafts_options(app):
-    """Override the "search drafts" options to show all accessible drafts."""
-    # doing this via config is (currently) not possible, as the `search_drafts`
-    # property can't be overridden with a config item (unlike `search`, above it)
-    # cf. https://github.com/inveniosoftware/invenio-rdm-records/blob/maint-10.x/invenio_rdm_records/services/config.py#L327-L332
-    try:
-        service = app.extensions["invenio-rdm-records"].records_service
-        service.config.search_drafts.params_interpreters_cls.remove(MyDraftsParam)
-    except ValueError:
-        pass
-
-
 class InvenioConfigTUW(object):
     """Invenio-Config-TUW extension."""
 

invenio_config_tuw/startup.py

@@ -20,6 +20,7 @@ from flask import Blueprint, current_app, flash, render_template, request
 from flask_login import current_user, login_required
 from flask_menu import register_menu
 from invenio_db import db
+from invenio_rdm_records.services.search_params import MyDraftsParam
 
 from .curation import CurationForm
 from .formatters import CustomFormatter
@@ -83,6 +84,18 @@ def register_smtp_error_handler(app):
         )
 
 
+def override_search_drafts_options(app):
+    """Override the "search drafts" options to show all accessible drafts."""
+    # doing this via config is (currently) not possible, as the `search_drafts`
+    # property can't be overridden with a config item (unlike `search`, above it)
+    # cf. https://github.com/inveniosoftware/invenio-rdm-records/blob/maint-10.x/invenio_rdm_records/services/config.py#L327-L332
+    try:
+        service = app.extensions["invenio-rdm-records"].records_service
+        service.config.search_drafts.params_interpreters_cls.remove(MyDraftsParam)
+    except ValueError:
+        pass
+
+
 def create_curation_settings_blueprint(app):
     """Register the curation settings view after the app has been initialized.
 

pyproject.toml

@@ -72,11 +72,11 @@ invenio_config_tuw_settings = "invenio_config_tuw.startup:create_curation_settin
 
 [project.entry-points."invenio_base.finalize_app"]
 invenio_config_tuw_mail_handler = "invenio_config_tuw.startup:register_smtp_error_handler"
-invenio_config_tuw_search_drafts = "invenio_config_tuw.ext:override_search_drafts_options"
+invenio_config_tuw_search_drafts = "invenio_config_tuw.startup:override_search_drafts_options"
 
 [project.entry-points."invenio_base.api_finalize_app"]
 invenio_config_tuw_mail_handler = "invenio_config_tuw.startup:register_smtp_error_handler"
-invenio_config_tuw_search_drafts = "invenio_config_tuw.ext:override_search_drafts_options"
+invenio_config_tuw_search_drafts = "invenio_config_tuw.startup:override_search_drafts_options"
 
 [project.entry-points."invenio_celery.tasks"]
 invenio_config_tuw = "invenio_config_tuw.tasks"