ayon_equalizer

class EqualizerAddon(AYONAddon, IHostAddon):
    """3DEqualizer Addon for AYON."""

    name = "equalizer"
    host_name = "equalizer"
    version = __version__
    heartbeat = 100

    def initialize(self, settings: dict[str, Any]) -> None:
        """Initialize Equalizer Addon."""
        self.heartbeat = settings["equalizer"]["heartbeat_interval"]
        self.enabled = True

    def add_implementation_envs(self, env: dict, _app: Any) -> None:  # noqa: ANN401
        """Add 3DEqualizer specific environment variables.

        3DEqualizer utilizes TDE4_ROOT for its root directory
        and PYTHON_CUSTOM_SCRIPTS_3DE4 as a colon-separated list of
        directories to look for additional python scripts.
        (Windows: list is separated by semicolons).

        Arguments:
            env (dict): Environment variables.
            _app (str): Application name.

        """
        startup_path = os.path.join(EQUALIZER_HOST_DIR, "startup")
        if "PYTHON_CUSTOM_SCRIPTS_3DE4" in env:
            startup_path = os.path.join(
                env["PYTHON_CUSTOM_SCRIPTS_3DE4"],
                startup_path)

        env["PYTHON_CUSTOM_SCRIPTS_3DE4"] = startup_path
        env["AYON_TDE4_HEARTBEAT_INTERVAL"] = str(self.heartbeat)

    def get_launch_hook_paths(self) -> list[str]:
        """Get paths to launch hooks."""
        return [os.path.join(EQUALIZER_HOST_DIR, "hooks")]

    def get_workfile_extensions(self) -> list[str]:
        """Get workfile extensions."""
        return [".3de"]