[Lldb-commits] [lldb] bf02586 - [lldb] Call __lldb_init_module from __init__

[Dave Lee via lldb-commits]

Author: Dave Lee
Date: 2022-02-07T14:43:44-08:00
New Revision: bf02586c57ced32ca66fd98fa3208b632820b1c6

URL: https://github.com/llvm/llvm-project/commit/bf02586c57ced32ca66fd98fa3208b632820b1c6
DIFF: https://github.com/llvm/llvm-project/commit/bf02586c57ced32ca66fd98fa3208b632820b1c6.diff

LOG: [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

Added: 
    

Modified: 
    lldb/bindings/python/createPythonInit.py

Removed: 
    


################################################################################
diff  --git a/lldb/bindings/python/createPythonInit.py b/lldb/bindings/python/createPythonInit.py
index 3deb9159b7028..4ad7320675f45 100644
--- a/lldb/bindings/python/createPythonInit.py
+++ b/lldb/bindings/python/createPythonInit.py
@@ -10,6 +10,14 @@
 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"))