def _get_pkg_modules(self, pkg_name, get_sub_folders=True):
"""get all package modules recursively
:param pkg_name: (str) path to search for python modules
:param get_sub_folders: (bool) search sub-folders even if they are
not python packages
"""
pkg_glob = os.path.join(pkg_name, "*.py")
this_modules = glob.glob(pkg_glob)
for dirname, dirnames, filenames in os.walk(pkg_name):
for subdirname in dirnames:
sub_path = os.path.join(dirname, subdirname)
if get_sub_folders or PyModule.is_pkg(sub_path):
this_modules.extend(self._get_pkg_modules(sub_path))
return this_modules
def test_is_pkg(self):
assert True == PyModule.is_pkg(FOO.pkg)
assert False == PyModule.is_pkg(FOO.init)
assert False == PyModule.is_pkg(FOO.a)
assert True == PyModule.is_pkg(SUB.pkg)
assert False == PyModule.is_pkg(SUB.a)