ets-labs/python-dependency-injector
1
1
Fork 0
mirror of https://github.com/ets-labs/python-dependency-injector.git synced 2024-11-22 09:36:48 +03:00
Code Issues Packages Projects Releases Wiki Activity

Update wiring to support modules provided as packages

Browse Source
This commit is contained in:
Roman Mogylatov 2021-08-06 15:56:24 -04:00
parent b4ddf61939
commit 7bdcc33eda
2 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/dependency_injector/wiring.py
View File

@ -552,6 +552,8 @@ def _unbind_injections(fn: Callable[..., Any]) -> None:
def _fetch_modules(package): def _fetch_modules(package):
modules = [package] modules = [package]
if not hasattr(package, '__path__') or not hasattr(package, '__name__'):
return modules
for module_info in pkgutil.walk_packages( for module_info in pkgutil.walk_packages(
path=package.__path__, path=package.__path__,
prefix=package.__name__ + '.', prefix=package.__name__ + '.',

12
tests/unit/wiring/test_wiring_py36.py
View File

@ -310,6 +310,18 @@ class WiringTest(unittest.TestCase):
self.assertIsInstance(service, Service) self.assertIsInstance(service, Service)
class ModuleAsPackagingTest(unittest.TestCase):
def setUp(self):
self.container = Container(config={'a': {'b': {'c': 10}}})
self.addCleanup(self.container.unwire)
def test_module_as_package_wiring(self):
# See: https://github.com/ets-labs/python-dependency-injector/issues/481
self.container.wire(packages=[module])
self.assertIsInstance(module.service, Service)
class WiringAndQueue(unittest.TestCase): class WiringAndQueue(unittest.TestCase):
def test_wire_queue(self) -> None: def test_wire_queue(self) -> None: