From 3af5f4105c08af82a163e01a8c6017f98e53cfed Mon Sep 17 00:00:00 2001 From: Lev Gorodetskiy Date: Mon, 19 Jul 2021 15:40:38 +0300 Subject: [PATCH] Fix package import verification --- src/dipdup/utils.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/src/dipdup/utils.py b/src/dipdup/utils.py index dd3c5506c..80c27fb0d 100644 --- a/src/dipdup/utils.py +++ b/src/dipdup/utils.py @@ -7,7 +7,6 @@ import types from contextlib import asynccontextmanager from logging import Logger -from os.path import dirname from typing import Any, AsyncIterator, Dict, Iterator, List, Optional, Tuple, Type import humps # type: ignore @@ -24,10 +23,9 @@ def import_submodules(package: str) -> Dict[str, types.ModuleType]: """Import all submodules of a module, recursively, including subpackages""" - module = importlib.import_module(package) results = {} - for _, name, is_pkg in pkgutil.walk_packages((dirname(module.__file__),)): - full_name = module.__name__ + '.' + name + for _, name, is_pkg in pkgutil.walk_packages((package,)): + full_name = package + '.' + name results[full_name] = importlib.import_module(full_name) if is_pkg: results.update(import_submodules(full_name))