immich/machine-learning/scripts/healthcheck.py

import os
import sys
from ipaddress import ip_address

import requests

port = os.getenv("IMMICH_PORT", 3003)
host = os.getenv("IMMICH_HOST", "0.0.0.0")


def is_ipv6(host: str) -> bool:
    try:
        return ip_address(host).version == 6
    except ValueError:
        return False


host = "localhost" if host == "0.0.0.0" else host
host = f"[{host}]" if is_ipv6(host) else host

try:
    response = requests.get(f"http://{host}:{port}/ping", timeout=2)
    if response.status_code == 200:
        sys.exit(0)
    sys.exit(1)
except requests.RequestException:
    sys.exit(1)