# coding=utf-8
"""
@Time   :2026/6/24
@Author :tao.chen
"""
from common.logging import logger
from spark_executor.core.job_store import JobStore
from spark_executor.core.yarn_client import YarnClientConfig, get_application_logs
from spark_executor.tools.connections import store as conn_store

store = JobStore()


def get_job_logs(job_id: str, tail_chars: int = 5000) -> str:
    logger.debug(f"get_job_logs enter job_id={job_id} tail_chars={tail_chars}")
    job = store.get(job_id)
    if job is None:
        raise KeyError(f"Unknown job_id: {job_id}")
    conn = conn_store.get(job.connection)
    if conn is None:
        raise KeyError(f"Connection not found: {job.connection}")
    config = YarnClientConfig.from_connection(conn)
    full = get_application_logs(job.application_id, config)
    tailed = full[-tail_chars:] if len(full) > tail_chars else full
    logger.info(
        f"get_job_logs ok job_id={job_id} application_id={job.application_id} "
        f"full_chars={len(full)} returned_chars={len(tailed)}"
    )
    return tailed