From bf02586c57ced32ca66fd98fa3208b632820b1c6 Mon Sep 17 00:00:00 2001 From: Dave Lee Date: Mon, 7 Feb 2022 12:24:51 -0800 Subject: [lldb] Call __lldb_init_module from __init__ Update `__init__.py` generation to implement `__lldb_init_module`, which calls `__lldb_init_module` on submodules that define it. This allows the use case where a user runs `command script import lldb.macosx`. With this change, the `__lldb_init_module` function in `crashlog.py` and `heap.py` will be run, which is where command registration is occurring. Differential Revision: https://reviews.llvm.org/D119179 --- lldb/bindings/python/createPythonInit.py | 8 ++++++++ 1 file changed, 8 insertions(+) (limited to 'lldb/bindings/python/createPythonInit.py') diff --git a/lldb/bindings/python/createPythonInit.py b/lldb/bindings/python/createPythonInit.py index 3deb9159b702..4ad7320675f4 100644 --- a/lldb/bindings/python/createPythonInit.py +++ b/lldb/bindings/python/createPythonInit.py @@ -10,6 +10,14 @@ importNames = ', '.join('"{}"'.format(getFileName(f)) for f in pkgFiles) script = """__all__ = [{import_names}] for x in __all__: __import__('lldb.{pkg_name}.' + x) + +def __lldb_init_module(debugger, internal_dict): + import lldb + for x in __all__: + submodule = getattr(lldb.{pkg_name}, x) + lldb_init = getattr(submodule, '__lldb_init_module', None) + if lldb_init: + lldb_init(debugger, internal_dict) """.format(import_names=importNames, pkg_name=pkgRelDir.replace("/", ".")) pkgIniFile = os.path.normpath(os.path.join(pkgRelDir, "__init__.py")) -- cgit v1.2.3