69 lines
2.4 KiB
Python
69 lines
2.4 KiB
Python
|
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
|