小技巧 | 通过Python向飞书发送训练日志

import osimport requestsfrom dotenv import load_dotenvload_dotenv()def send_slack(message: str) -> None:    webhook_url = os.environ.get("SLACK_WEBHOOK_URL")    if not webhook_url:        return    response = requests.post(        webhook_url,        json={"text": message},        timeout=10,    )    response.raise_for_status()def send_feishu(message: str) -> None:    webhook_url = os.environ.get("FEISHU_WEBHOOK_URL")    print(webhook_url)    if not webhook_url:        raise ValueError("webhook not found")    response = requests.post(        webhook_url,        json={            "msg_type": "text",            "content": {                "text": message,            },        },        timeout=10,    )    response.raise_for_status()def send_notification(message: str) -> None:    """同时向已配置的平台发送消息；单个平台失败不影响另一个。"""    for sender in (send_slack, send_feishu):        try:            sender(message)        except Exception as exc:            print(f"Notification failed: {exc}")if __name__ == "__main__":    try:        send_feishu("✅ Python Webhook 最小测试成功")        print("请求已发送，请检查 Slack 或飞书")    except Exception as exc:        print(f"发送失败：{type(exc).__name__}: {exc}")        raise

if __name__ == "__main__":  main()

args = parser.parse_args()task_name = f"\"{args.experiment_name}_{args.dataset}_{current_time}\""send_feishu(f"Task: {task_name} starts")try:    main(args)except Exception as error:    error_message = (f"Task: {task_name} failed\nError: {type(error).__name__}: {error}")    send_feishu(error_message)    # 在终端输出完整报错栈    traceback.print_exc()    # 继续抛出异常，使脚本以非零状态码退出    raiseelse:    send_feishu(f"Task: {task_name} completed")