xref: /llvm-project/lldb/use_lldb_suite_root.py (revision b9c1b51e45b845debb76d8658edabca70ca56079)

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)