Job Run Status Monitor

from airflow.providers.dbt.cloud.hooks.dbt import DbtCloudHook
from typing import Any

class JobRunMonitor:
    """Monitor and query dbt Cloud job run statuses."""

    def __init__(self, dbt_cloud_conn_id: str = "dbt_cloud_default"):
        """
        Initialize the job run monitor.

        Args:
            dbt_cloud_conn_id: Airflow connection ID for dbt Cloud
        """
        pass

    def get_run_status(self, run_id: int, account_id: int | None = None) -> int:
        """
        Get the current status of a job run.

        Args:
            run_id: The ID of the job run to check
            account_id: Optional account ID (defaults to connection default)

        Returns:
            int: Status code representing the current state

        Raises:
            Exception: If the run cannot be found or an error occurs
        """
        pass

    def list_runs_for_job(
        self,
        job_id: int,
        account_id: int | None = None
    ) -> list[dict[str, Any]]:
        """
        List all runs for a specific job.

        Args:
            job_id: The ID of the job to query runs for
            account_id: Optional account ID (defaults to connection default)

        Returns:
            list[dict]: List of job run information dictionaries
        """
        pass

    def get_run_details(
        self,
        run_id: int,
        account_id: int | None = None,
        include_related: list[str] | None = None
    ) -> dict[str, Any]:
        """
        Get detailed information about a specific job run.

        Args:
            run_id: The ID of the job run to retrieve
            account_id: Optional account ID (defaults to connection default)
            include_related: List of related resources to include

        Returns:
            dict: Detailed job run information including metadata
        """
        pass