Notification Script

import requests
import time

# Replace these values with your Jellyfin server details
JELLYFIN_SERVER_URL = "http://172.22.20.1:8096"
JELLYFIN_API_KEY = "5f1013b18bba4cada3c6741be6c49596"

# Function to get session IDs
def get_session_ids():
    headers = {"Authorization": f"MediaBrowser Token={JELLYFIN_API_KEY}"}
    params = {"activeWithinSeconds": 60}  # Adjust the time as needed

    response = requests.get(f"{JELLYFIN_SERVER_URL}/Sessions", headers=headers, params=params)

    if response.status_code == 200:
        sessions = response.json()
        session_ids = [session["Id"] for session in sessions]
        return session_ids
    else:
        print(f"Failed to get session IDs. Status code: {response.status_code}")
        return []

# Function to send a message to a specific session
def send_message(session_id, header, text, timeout_ms=10000):
    headers = {"Authorization": f"MediaBrowser Token={JELLYFIN_API_KEY}"}
    url = f"{JELLYFIN_SERVER_URL}/Sessions/{session_id}/Message"

    message_data = {
        "Header": header,
        "Text": text,
        "TimeoutMs": timeout_ms
    }

    response = requests.post(url, headers=headers, json=message_data)

    if response.status_code == 204:
        print(f"Message sent to session {session_id}")
    else:
        print(f"Failed to send message to session {session_id}. Status code: {response.status_code}")

# Function to send maintenance countdown messages
def send_maintenance_countdown(initial_countdown=60):
    header = "Maintenance Alert"

    # Announce initial countdown
    text = f"Server maintenance will start in {initial_countdown} minutes"
    send_message_to_all(header, text)
    time.sleep(600)  # Wait for 10 minutes

    # Announce remaining countdown
    for minutes in range(initial_countdown - 10, 0, -10):
        text = f"Server maintenance will start in {minutes} minute{'s' if minutes > 1 else ''}"
        send_message_to_all(header, text)
        if minutes > 5:
            time.sleep(600)  # Wait for 10 minutes
        else:
            time.sleep(60 * minutes)  # Wait for remaining minutes

# Function to send a message to all online users
def send_message_to_all(header, text):
    session_ids = get_session_ids()

    if session_ids:
        for session_id in session_ids:
            send_message(session_id, header, text)

# Main script
def main():
    # Example: Send immediate messages
    header_immediate = "Immediate Alert"
    text_immediate = "This is an immediate alert!"
    send_message_to_all(header_immediate, text_immediate)

    # Example: Send maintenance countdown messages with initial countdown of 60 minutes
    #send_maintenance_countdown(initial_countdown=60)

if __name__ == "__main__":
    main()