Allow skipping of unknown modules

This commit is contained in:
Aviv Palivoda committed 2019-07-04 23:58:01 +03:00
1 parent 72febb77d0
commit 2edba36de7
1 file changed
+7 -3
+7 -3
View File
@@ -60,6 +60,12 @@ def gen_methods(header, footer, output):
output.write("}\n\n")
output.write(footer.read())
def get_module_methods(name):
try:
return set(dir(__import__(name))) if name not in ("this", "antigravity") else None
except ModuleNotFoundError:
return None
def gen_modules(header, footer, output):
output.write(header.read())
@@ -70,9 +76,7 @@ def gen_modules(header, footer, output):
mod.name,
# check name b/c modules listed have side effects on import,
# e.g. printing something or opening a webpage
set(dir(__import__(mod.name)))
if mod.name not in ("this", "antigravity")
else None,
get_module_methods(mod.name)
),
pkgutil.iter_modules(),
)