mirror of
https://github.com/ets-labs/python-dependency-injector.git
synced 2025-05-15 21:33:46 +03:00
Refactor schema module
This commit is contained in:
Roman Mogylatov
parent
19786d3c7a
commit
8cac34b1c0
|
|
@ -9,123 +9,129 @@ from . import containers, providers
|
||||||
Schema = Dict[Any, Any]
|
Schema = Dict[Any, Any]
|
||||||
|
|
||||||
|
|
||||||
def build_schema(schema: Schema) -> Dict[str, providers.Provider]:
|
class SchemaProcessorV1:
|
||||||
"""Build provider schema."""
|
|
||||||
container = containers.DynamicContainer()
|
|
||||||
_create_providers(container, schema['providers'])
|
|
||||||
_setup_injections(container, schema['providers'])
|
|
||||||
return container.providers
|
|
||||||
|
|
||||||
|
def __init__(self, schema: Schema) -> None:
|
||||||
|
self._schema = schema
|
||||||
|
self._container = containers.DynamicContainer()
|
||||||
|
|
||||||
def _create_providers(
|
def process(self):
|
||||||
container: containers.Container,
|
"""Process schema."""
|
||||||
providers_data: Dict[str, Any],
|
self._create_providers(self._schema)
|
||||||
):
|
self._setup_injections(self._schema)
|
||||||
for provider_name, data in providers_data.items():
|
|
||||||
provider_type = _get_provider_cls(data['provider'])
|
|
||||||
args = []
|
|
||||||
|
|
||||||
provides = data.get('provides')
|
def get_providers(self):
|
||||||
if provides:
|
"""Return providers."""
|
||||||
provides = _import_string(provides)
|
return self._container.providers
|
||||||
|
|
||||||
|
def _create_providers(self, schema: Schema, container: Optional[containers.Container] = None) -> None:
|
||||||
|
if container is None:
|
||||||
|
container = self._container
|
||||||
|
for provider_name, data in schema['providers'].items():
|
||||||
|
provider_type = _get_provider_cls(data['provider'])
|
||||||
|
args = []
|
||||||
|
|
||||||
|
provides = data.get('provides')
|
||||||
if provides:
|
if provides:
|
||||||
|
provides = _import_string(provides)
|
||||||
|
if provides:
|
||||||
|
args.append(provides)
|
||||||
|
|
||||||
|
if provider_type is providers.Container:
|
||||||
|
provides = containers.DynamicContainer
|
||||||
args.append(provides)
|
args.append(provides)
|
||||||
|
|
||||||
if provider_type is providers.Container:
|
provider = provider_type(*args)
|
||||||
provides = containers.DynamicContainer
|
container.set_provider(provider_name, provider)
|
||||||
args.append(provides)
|
|
||||||
|
|
||||||
provider = provider_type(*args)
|
if isinstance(provider, providers.Container):
|
||||||
container.set_provider(provider_name, provider)
|
self._create_providers(schema=data, container=provider)
|
||||||
|
|
||||||
if isinstance(provider, providers.Container):
|
def _setup_injections(self, schema: Schema, container: Optional[containers.Container] = None) -> None:
|
||||||
_create_providers(provider, data['providers'])
|
if container is None:
|
||||||
|
container = self._container
|
||||||
|
|
||||||
|
for provider_name, data in schema['providers'].items():
|
||||||
|
provider = getattr(container, provider_name)
|
||||||
|
args = []
|
||||||
|
kwargs = {}
|
||||||
|
|
||||||
def _setup_injections(
|
arg_injections = data.get('args')
|
||||||
container: containers.Container,
|
if arg_injections:
|
||||||
providers_data: Dict[str, Any],
|
for arg in arg_injections:
|
||||||
*,
|
injection = None
|
||||||
current_container: Optional[containers.Container] = None,
|
|
||||||
):
|
|
||||||
if not current_container:
|
|
||||||
current_container = container
|
|
||||||
|
|
||||||
for provider_name, data in providers_data.items():
|
if isinstance(arg, str):
|
||||||
provider = getattr(current_container, provider_name)
|
injection = self._resolve_provider(arg)
|
||||||
args = []
|
|
||||||
kwargs = {}
|
|
||||||
|
|
||||||
arg_injections = data.get('args')
|
# TODO: add inline injections
|
||||||
if arg_injections:
|
|
||||||
for arg in arg_injections:
|
|
||||||
injection = None
|
|
||||||
|
|
||||||
if isinstance(arg, str):
|
if not injection:
|
||||||
injection = _resolve_provider(container, arg)
|
injection = arg
|
||||||
|
|
||||||
# TODO: add inline injections
|
args.append(injection)
|
||||||
|
if args:
|
||||||
|
provider.add_args(*args)
|
||||||
|
|
||||||
if not injection:
|
kwarg_injections = data.get('kwargs')
|
||||||
injection = arg
|
if kwarg_injections:
|
||||||
|
for name, arg in kwarg_injections.items():
|
||||||
|
injection = None
|
||||||
|
|
||||||
args.append(injection)
|
if isinstance(arg, str):
|
||||||
if args:
|
injection = self._resolve_provider(arg)
|
||||||
provider.add_args(*args)
|
|
||||||
|
|
||||||
kwarg_injections = data.get('kwargs')
|
# TODO: refactoring
|
||||||
if kwarg_injections:
|
if isinstance(arg, dict):
|
||||||
for name, arg in kwarg_injections.items():
|
provider_args = []
|
||||||
injection = None
|
provider_type = _get_provider_cls(arg.get('provider'))
|
||||||
|
provides = arg.get('provides')
|
||||||
if isinstance(arg, str):
|
|
||||||
injection = _resolve_provider(container, arg)
|
|
||||||
|
|
||||||
# TODO: refactoring
|
|
||||||
if isinstance(arg, dict):
|
|
||||||
provider_args = []
|
|
||||||
provider_type = _get_provider_cls(arg.get('provider'))
|
|
||||||
provides = arg.get('provides')
|
|
||||||
if provides:
|
|
||||||
provides = _import_string(provides)
|
|
||||||
if provides:
|
if provides:
|
||||||
provider_args.append(provides)
|
provides = _import_string(provides)
|
||||||
for provider_arg in arg.get('args', []):
|
if provides:
|
||||||
provider_args.append(_resolve_provider(container, provider_arg))
|
provider_args.append(provides)
|
||||||
injection = provider_type(*provider_args)
|
for provider_arg in arg.get('args', []):
|
||||||
|
provider_args.append(self._resolve_provider(provider_arg))
|
||||||
|
injection = provider_type(*provider_args)
|
||||||
|
|
||||||
if not injection:
|
if not injection:
|
||||||
injection = arg
|
injection = arg
|
||||||
|
|
||||||
kwargs[name] = injection
|
kwargs[name] = injection
|
||||||
if kwargs:
|
if kwargs:
|
||||||
provider.add_kwargs(**kwargs)
|
provider.add_kwargs(**kwargs)
|
||||||
|
|
||||||
if isinstance(provider, providers.Container):
|
if isinstance(provider, providers.Container):
|
||||||
_setup_injections(container, data['providers'], current_container=provider)
|
self._setup_injections(schema=data, container=provider)
|
||||||
|
|
||||||
|
def _resolve_provider(self, name: str) -> Optional[providers.Provider]:
|
||||||
|
segments = name.split('.')
|
||||||
|
try:
|
||||||
|
provider = getattr(self._container, segments[0])
|
||||||
|
except AttributeError:
|
||||||
|
return None
|
||||||
|
|
||||||
|
for segment in segments[1:]:
|
||||||
|
if segment == 'as_int()':
|
||||||
|
provider = provider.as_int()
|
||||||
|
elif segment == 'as_float()':
|
||||||
|
provider = provider.as_float()
|
||||||
|
elif segment.startswith('is_'): # TODO
|
||||||
|
provider = provider.as_(str)
|
||||||
|
...
|
||||||
|
else:
|
||||||
|
try:
|
||||||
|
provider = getattr(provider, segment)
|
||||||
|
except AttributeError:
|
||||||
|
return None
|
||||||
|
return provider
|
||||||
|
|
||||||
|
|
||||||
def _resolve_provider(container: containers.Container, name: str) -> Optional[providers.Provider]:
|
def build_schema(schema: Schema) -> Dict[str, providers.Provider]:
|
||||||
segments = name.split('.')
|
"""Build provider schema."""
|
||||||
try:
|
schema_processor = SchemaProcessorV1(schema)
|
||||||
provider = getattr(container, segments[0])
|
schema_processor.process()
|
||||||
except AttributeError:
|
return schema_processor.get_providers()
|
||||||
return None
|
|
||||||
|
|
||||||
for segment in segments[1:]:
|
|
||||||
if segment == 'as_int()':
|
|
||||||
provider = provider.as_int()
|
|
||||||
elif segment == 'as_float()':
|
|
||||||
provider = provider.as_float()
|
|
||||||
elif segment.startswith('is_'): # TODO
|
|
||||||
provider = provider.as_(str)
|
|
||||||
...
|
|
||||||
else:
|
|
||||||
try:
|
|
||||||
provider = getattr(provider, segment)
|
|
||||||
except AttributeError:
|
|
||||||
return None
|
|
||||||
return provider
|
|
||||||
|
|
||||||
|
|
||||||
def _get_provider_cls(provider_cls_name: str) -> Type[providers.Provider]:
|
def _get_provider_cls(provider_cls_name: str) -> Type[providers.Provider]:
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue
Block a user