use_lldb_suite_root.py

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()))

# Use environment variables to avoid plumbing flags, lit configs, etc.
if os.getenv("LLDB_TEST_USE_VENDOR_PACKAGES"):
    add_third_party_module_dirs(lldb_root)
add_lldbsuite_packages_dir(lldb_root)