Merge pull request #119 from chrisburr/settings-lifetime-func

Add lifetime_function to ServiceSettingsBase

src/diracx/core/settings.py

@@ -6,8 +6,9 @@
     "ServiceSettingsBase",
 )
 
+import contextlib
 from pathlib import Path
-from typing import TYPE_CHECKING, Any, Self, TypeVar
+from typing import TYPE_CHECKING, Any, AsyncIterator, Self, TypeVar
 
 from authlib.jose import JsonWebKey
 from pydantic import AnyUrl, BaseSettings, SecretStr, parse_obj_as
@@ -58,3 +59,8 @@ class ServiceSettingsBase(BaseSettings, allow_mutation=False):
     @classmethod
     def create(cls) -> Self:
         raise NotImplementedError("This should never be called")
+
+    @contextlib.asynccontextmanager
+    async def lifetime_function(self) -> AsyncIterator[None]:
+        """A context manager that can be used to run code at startup and shutdown."""
+        yield

src/diracx/routers/__init__.py

@@ -52,6 +52,7 @@ def create_app_inner(
         cls = type(service_settings)
         assert cls not in available_settings_classes
         available_settings_classes.add(cls)
+        app.lifetime_functions.append(service_settings.lifetime_function)
         app.dependency_overrides[cls.create] = partial(lambda x: x, service_settings)
 
     # Override the configuration source