#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""查 docker-worker-1 最近的 ops_trace / tool_trace 相关日志（用于判断 TOOL_TRACE 任务是否执行）。"""

from __future__ import annotations

import sys

import _bootstrap  # noqa: F401
import _config as C
from dify_api import Dify


def main() -> int:
    d = Dify(C.DIFY_EMAIL, C.DIFY_PASSWORD)
    try:
        d.login()
        out, err = d.sh(
            "docker logs --since 20m docker-worker-1 2>&1 | grep -iE 'ops_trace|TOOL_TRACE|tool_trace|getServerStatus' | tail -120",
            t=240,
        )
        print(out or err or "(no matching logs)")
        return 0
    finally:
        d.close()


if __name__ == "__main__":
    sys.exit(main())