[lldb] Call __lldb_init_module from __init__
authorDave Lee <davelee.com@gmail.com>
Mon, 7 Feb 2022 20:24:51 +0000 (12:24 -0800)
committerDave Lee <davelee.com@gmail.com>
Mon, 7 Feb 2022 22:43:44 +0000 (14:43 -0800)
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

index 3deb915..4ad7320 100644 (file)
@@ -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"))