ets-labs/python-dependency-injector
1
1
Fork 0
mirror of https://github.com/ets-labs/python-dependency-injector.git synced 2025-07-04 12:23:14 +03:00
Code Issues Packages Projects Releases Wiki Activity

Rename properties of PatchedRegistry

Browse Source
This commit is contained in:
Roman Mogylatov 2022-07-23 23:03:35 -04:00
parent 2d52b5c740
commit 99c18cd6c2
1 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
src/dependency_injector/wiring.py
View File

@ -92,38 +92,38 @@ Container = Any
class PatchedRegistry: class PatchedRegistry:
def __init__(self): def __init__(self):
self._patched_callables: Dict[Callable[..., Any], "PatchedCallable"] = {} self._callables: Dict[Callable[..., Any], "PatchedCallable"] = {}
self._patched_attributes: Set[PatchedAttribute] = set() self._attributes: Set[PatchedAttribute] = set()
def register_callable(self, patched: "PatchedCallable") -> None: def register_callable(self, patched: "PatchedCallable") -> None:
self._patched_callables[patched.patched] = patched self._callables[patched.patched] = patched
def get_callables_from_module(self, module: ModuleType) -> Iterator[Callable[..., Any]]: def get_callables_from_module(self, module: ModuleType) -> Iterator[Callable[..., Any]]:
for patched_callable in self._patched_callables.values(): for patched_callable in self._callables.values():
if not patched_callable.is_in_module(module): if not patched_callable.is_in_module(module):
continue continue
yield patched_callable.patched yield patched_callable.patched
def get_callable(self, fn: Callable[..., Any]) -> "PatchedCallable": def get_callable(self, fn: Callable[..., Any]) -> "PatchedCallable":
return self._patched_callables.get(fn) return self._callables.get(fn)
def has_callable(self, fn: Callable[..., Any]) -> bool: def has_callable(self, fn: Callable[..., Any]) -> bool:
return fn in self._patched_callables return fn in self._callables
def register_attribute(self, patched: "PatchedAttribute"): def register_attribute(self, patched: "PatchedAttribute"):
self._patched_attributes.add(patched) self._attributes.add(patched)
def get_attributes_from_module(self, module: ModuleType) -> Iterator["PatchedAttribute"]: def get_attributes_from_module(self, module: ModuleType) -> Iterator["PatchedAttribute"]:
for attribute in self._patched_attributes: for attribute in self._attributes:
if not attribute.is_in_module(module): if not attribute.is_in_module(module):
continue continue
yield attribute yield attribute
def clear_module_attributes(self, module: ModuleType): def clear_module_attributes(self, module: ModuleType):
for attribute in self._patched_attributes.copy(): for attribute in self._attributes.copy():
if not attribute.is_in_module(module): if not attribute.is_in_module(module):
continue continue
self._patched_attributes.remove(attribute) self._attributes.remove(attribute)
class PatchedCallable: class PatchedCallable: