Install NXOpen

install_nxopen.py

import os
import pathlib
import site


def main():
    # Hack-ish way to get site-packages
    site_packages_path = pathlib.Path(
        [x for x in site.getsitepackages() if "site-packages" in x][0]
    )

    pth_file_path = site_packages_path / "nxopen.pth"

    ugii_base_dir = os.getenv("UGII_BASE_DIR", None)
    if not ugii_base_dir:
        raise Exception("Cannot find NX installation (no UGII_BASE_DIR variable).")

    print(f"Found NX installation at {ugii_base_dir}")

    nxopen_path = pathlib.Path(ugii_base_dir) / "NXBIN" / "python"

    pth_file_path.write_text(str(nxopen_path.resolve()))

    print(f"Installed NXOpen through {pth_file_path}")


if __name__ == "__main__":
    main()