Added logic to handle StartupExtension classes

src/cpl_core/application/__init__.py

@@ -24,6 +24,7 @@ from .application_abc import ApplicationABC
 from .application_builder import ApplicationBuilder
 from .application_builder_abc import ApplicationBuilderABC
 from .startup_abc import StartupABC
+from .startup_extension_abc import StartupExtensionABC
 
 VersionInfo = namedtuple('VersionInfo', 'major minor micro')
 version_info = VersionInfo(major='2022', minor='6', micro='1')

src/cpl_core/application/application_builder.py

@@ -1,9 +1,10 @@
-from typing import Type, Optional, Callable
+from typing import Type, Optional, Callable, Union
 
 from cpl_core.application.application_abc import ApplicationABC
 from cpl_core.application.application_builder_abc import ApplicationBuilderABC
 from cpl_core.application.application_extension_abc import ApplicationExtensionABC
 from cpl_core.application.startup_abc import StartupABC
+from cpl_core.application.startup_extension_abc import StartupExtensionABC
 from cpl_core.configuration.configuration import Configuration
 from cpl_core.dependency_injection.service_collection import ServiceCollection
 
@@ -26,38 +27,47 @@ class ApplicationBuilder(ApplicationBuilderABC):
         self._environment = self._configuration.environment
         self._services = ServiceCollection(self._configuration)
 
-        self._extensions: list[Callable] = []
+        self._app_extensions: list[Callable] = []
+        self._startup_extensions: list[Callable] = []
 
     def use_startup(self, startup: Type[StartupABC]):
         self._startup = startup()
 
-    def use_extension(self, extension: Type[ApplicationExtensionABC]):
-        if extension not in self._extensions:
-            self._extensions.append(extension)
+    def use_extension(self, extension: Type[Union[ApplicationExtensionABC, StartupExtensionABC]]):
+        if issubclass(extension, ApplicationExtensionABC) and extension not in self._app_extensions:
+            self._app_extensions.append(extension)
+        elif issubclass(extension, StartupExtensionABC) and extension not in self._startup_extensions:
+            self._startup_extensions.append(extension)
+
+    def _build_startup(self):
+        for ex in self._startup_extensions:
+            extension = ex()
+            extension.configure_configuration(self._configuration, self._environment)
+            extension.configure_services(self._services, self._environment)
 
-    def build(self) -> ApplicationABC:
         if self._startup is not None:
             self._startup.configure_configuration(self._configuration, self._environment)
             self._startup.configure_services(self._services, self._environment)
 
+    def build(self) -> ApplicationABC:
+        self._build_startup()
+
         config = self._configuration
         services = self._services.build_service_provider()
 
-        for ex in self._extensions:
+        for ex in self._app_extensions:
             extension = ex()
             extension.run(config, services)
 
         return self._app(config, services)
 
     async def build_async(self) -> ApplicationABC:
-        if self._startup is not None:
-            await self._startup.configure_configuration(self._configuration, self._environment)
-            await self._startup.configure_services(self._services, self._environment)
+        self._build_startup()
 
         config = self._configuration
         services = self._services.build_service_provider()
 
-        for ex in self._extensions:
+        for ex in self._app_extensions:
             extension = ex()
             await extension.run(config, services)
 

src/cpl_core/application/startup_abc.py

@@ -10,8 +10,7 @@ class StartupABC(ABC):
     r"""ABC for the startup class"""
 
     @abstractmethod
-    def __init__(self, *args):
-        pass
+    def __init__(self): pass
 
     @abstractmethod
     def configure_configuration(self, config: ConfigurationABC, env: ApplicationEnvironmentABC) -> ConfigurationABC:

src/cpl_core/application/startup_extension_abc.py

@@ -0,0 +1,34 @@
+from abc import ABC, abstractmethod
+
+from cpl_core.configuration.configuration_abc import ConfigurationABC
+from cpl_core.dependency_injection.service_collection_abc import ServiceCollectionABC
+from cpl_core.environment.application_environment_abc import ApplicationEnvironmentABC
+
+
+class StartupExtensionABC(ABC):
+    r"""ABC for startup extension classes"""
+
+    @abstractmethod
+    def __init__(self): pass
+
+    @abstractmethod
+    def configure_configuration(self, config: ConfigurationABC, env: ApplicationEnvironmentABC):
+        r"""Creates configuration of application
+
+        Parameter
+        ---------
+            config: :class:`cpl_core.configuration.configuration_abc.ConfigurationABC`
+            env: :class:`cpl_core.environment.application_environment_abc`
+        """
+        pass
+
+    @abstractmethod
+    def configure_services(self, service: ServiceCollectionABC, env: ApplicationEnvironmentABC):
+        r"""Creates service provider
+
+        Parameter
+        ---------
+            services: :class:`cpl_core.dependency_injection.service_collection_abc`
+            env: :class:`cpl_core.environment.application_environment_abc`
+        """
+        pass