Slackに通知を送らせよう

import argparse
import os
import time

import requests
from slack_sdk import WebClient
from slack_sdk.errors import SlackApiError


def check_connection(url):
    try:
        response = requests.get(url, timeout=10)
        if response.status_code == 200:
            return True
    except:
        pass
    return False


def send_slack_notification(message, slack_token, user_id):
    client = WebClient(token=slack_token)
    try:
        response = client.chat_postMessage(
            channel=user_id, text=f"<@{user_id}> {message}"
        )
        print(f"Slack notification sent: {response['ts']}")
    except SlackApiError as e:
        print(f"Error sending Slack notification: {e}")


def send_notification(message, slack_token, user_id, debug=False):
    if debug:
        print(f"Debug mode: <@{user_id}> {message}")
    else:
        send_slack_notification(message, slack_token, user_id)


def main(args):
    url = "https://nlp.stanford.edu/"
    while True:
        if check_connection(url):
            send_notification(
                "nlp.stanford.edu is back online!",
                args.slack_token,
                args.slack_user_id,
                args.debug,
            )
            break
        else:
            # send_notification("nlp.stanford.edu is still offline. Checking again in 10 minutes...", args.slack_token, args.slack_user_id, args.debug)
            time.sleep(600)  # Wait for 10 minutes (600 seconds)


if __name__ == "__main__":
    parser = argparse.ArgumentParser(description="Check connection to nlp.stanford.edu")
    parser.add_argument("--debug", action="store_true", help="Enable debug mode")
    parser.add_argument("--slack-token", required=True, help="Slack API token")
    parser.add_argument(
        "--slack-user-id", required=True, help="Slack user ID to send notifications"
    )
    args = parser.parse_args()
    main(args)