pre_maya_launch

class SetupAssetResolver(PreLaunchHook):
    """Configure the AYON USD startup script for Maya."""

    # Must run after InitializeAssetResolver.
    order = 20

    app_groups = {"maya"}
    launch_types = {
        LaunchTypes.local,
    }

    def execute(self):
        """Add the AYON USD Maya startup directory to PYTHONPATH."""
        if not self.data.get("ayon_usd_resolver_initialized"):
            self.log.debug(
                "Skipping AYON USD Maya startup initialization because USD resolver is disabled."
            )
            return

        user_setup_path = os.path.join(
            MAYA_STARTUP_DIR,
            "userSetup.py",
        )

        if not os.path.isfile(user_setup_path):
            raise RuntimeError(
                "AYON USD Maya userSetup.py was not found: "
                f"{user_setup_path}"
            )

        env = self.launch_context.env
        current_pythonpath = env.get("PYTHONPATH", "")
        paths = current_pythonpath.split(os.pathsep) if current_pythonpath else []
        if MAYA_STARTUP_DIR not in paths:
            paths.append(MAYA_STARTUP_DIR)

        env["PYTHONPATH"] = os.pathsep.join(paths)

        self.log.info(
            "Added AYON USD Maya startup directory: %s",
            MAYA_STARTUP_DIR,
        )