kastiglione created this revision.
kastiglione added a reviewer: JDevlieghere.
kastiglione requested review of this revision.
Herald added a project: LLDB.
Herald added a subscriber: lldb-commits.

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.


Repository:
  rG LLVM Github Monorepo

https://reviews.llvm.org/D119179

Files:
  lldb/bindings/python/createPythonInit.py


Index: lldb/bindings/python/createPythonInit.py
===================================================================
--- lldb/bindings/python/createPythonInit.py
+++ 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"))


Index: lldb/bindings/python/createPythonInit.py
===================================================================
--- lldb/bindings/python/createPythonInit.py
+++ 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"))
_______________________________________________
lldb-commits mailing list
lldb-commits@lists.llvm.org
https://lists.llvm.org/cgi-bin/mailman/listinfo/lldb-commits

Reply via email to