Added ServiceProvider

src/sh_edraft/service/service_provider.py

@@ -0,0 +1,68 @@
+from collections import Callable
+from typing import Type
+
+from termcolor import colored
+
+from sh_edraft.service.base.provider_base import ProviderBase
+from sh_edraft.service.base.service_base import ServiceBase
+from sh_edraft.service.model.provide_state import ProvideState
+
+
+class ServiceProvider(ProviderBase):
+
+    def __init__(self):
+        ProviderBase.__init__(self)
+
+    def create(self):
+        pass
+
+    @staticmethod
+    def _create_instance(service: type[ServiceBase], args: tuple) -> ServiceBase:
+        instance = service()
+        try:
+            instance.init(args)
+            return instance
+        except Exception as e:
+            print(colored(f'Argument error\n{e}', 'red'))
+
+    def add_transient(self, service: Type[ServiceBase], *args):
+        self._transient_services.append(ProvideState(service, args))
+
+    def add_scoped(self, service: Type[ServiceBase], *args):
+        self._transient_services.append(ProvideState(service, args))
+
+    def add_singleton(self, service: Type[ServiceBase], *args):
+        for known_service in self._singleton_services:
+            if type(known_service) == type(service):
+                raise Exception(f'Service from type {type(service)} already exists')
+
+        self._singleton_services.append(self._create_instance(service, args))
+
+    def get_service(self, instance_type: type) -> Callable[ServiceBase]:
+        for state in self._transient_services:
+            if state.service == instance_type:
+                return self._create_instance(state.service, state.args)
+
+        for state in self._scoped_services:
+            if type(state.service) == instance_type:
+                return self._create_instance(state.service, state.args)
+
+        for service in self._singleton_services:
+            if type(service) == instance_type:
+                return service
+
+    def remove_service(self, instance_type: type):
+        for state in self._transient_services:
+            if state.service == instance_type:
+                self._transient_services.remove(state)
+                return
+
+        for state in self._scoped_services:
+            if type(state.service) == instance_type:
+                self._scoped_services.remove(state)
+                return
+
+        for service in self._singleton_services:
+            if type(service) == instance_type:
+                self._singleton_services.remove(service)
+                return