1 files changed, 22 insertions, 0 deletions
diff --git a/use_lldb_suite_root.py b/use_lldb_suite_root.py
new file mode 100644
index 0000000000000..5492d4081ac57
--- /dev/null
+++ b/use_lldb_suite_root.py
@@ -0,0 +1,22 @@
+import inspect
+import os
+import sys
+
+def add_third_party_module_dirs(lldb_root):
+    third_party_modules_dir = os.path.join(lldb_root, "third_party", "Python", "module")
+    if not os.path.isdir(third_party_modules_dir):
+        return
+
+    module_dirs = os.listdir(third_party_modules_dir)
+    for module_dir in module_dirs:
+        module_dir = os.path.join(third_party_modules_dir, module_dir)
+        sys.path.insert(0, module_dir)
+
+def add_lldbsuite_packages_dir(lldb_root):
+    packages_dir = os.path.join(lldb_root, "packages", "Python")
+    sys.path.insert(0, packages_dir)
+
+lldb_root = os.path.dirname(inspect.getfile(inspect.currentframe()))
+
+add_third_party_module_dirs(lldb_root)
+add_lldbsuite_packages_dir(lldb_root)