scripts/healthcheck.py

import os

HEALTH_MODELS = os.getenv("HEALTH_MODELS", "")


def run_healthcheck():
    import os
    import requests
    import sys
    import structlog
    from samgis_core.utilities.session_logger import setup_logging

    log_level = os.getenv("LOG_LEVEL", "INFO")
    url1 = os.getenv("HEALTHCHECK_URL1", "http://localhost:7860/health")
    url2 = os.getenv("HEALTHCHECK_URL2", "http://localhost:7860/health_models")
    setup_logging(log_level=log_level)
    app_logger = structlog.stdlib.get_logger()

    r1 = requests.get(url1)
    app_logger.info(r1.status_code)
    msg = f"status health:{r1.status_code}!"
    if HEALTH_MODELS:
        r2 = requests.get(url2)
        msg += f" status health_models:{r2.status_code}!"
        app_logger.info(msg)
        sys.exit(0) if r1.status_code == 200 and r2.status_code == 200 else sys.exit(1)
    app_logger.info(msg)
    sys.exit(0) if r1.status_code == 200 else sys.exit(1)



if __name__ == "__main__":
    run_healthcheck()