worker.py

"""
worker.py
Entry point for a Render Background Worker. Calls into main.call_api() on
a loop, with graceful shutdown on SIGTERM/SIGINT (Render sends SIGTERM when
stopping or redeploying a worker, so we want to exit cleanly rather than
get killed mid-request).
"""

import os
import signal
import sys
import time

from main import call_api

POLL_INTERVAL_SECONDS = int(os.environ.get("POLL_INTERVAL_SECONDS", "3600"))

_running = True


def _handle_shutdown(signum, frame):
    global _running
    print(f"Received signal {signum}, shutting down gracefully...", flush=True)
    _running = False


signal.signal(signal.SIGTERM, _handle_shutdown)
signal.signal(signal.SIGINT, _handle_shutdown)


def main_loop():
    print(f"Worker started. Calling API every {POLL_INTERVAL_SECONDS}s.", flush=True)
    while _running:
        try:
            call_api("Hello! Just testing the API connection.")
        except Exception as exc:
            print(f"Error calling API: {exc}", flush=True)

        # Sleep in short increments so a shutdown signal is handled promptly
        # instead of waiting out the full interval.
        slept = 0
        while slept < POLL_INTERVAL_SECONDS and _running:
            chunk = min(5, POLL_INTERVAL_SECONDS - slept)
            time.sleep(chunk)
            slept += chunk

    print("Worker stopped.", flush=True)
    sys.exit(0)


if __name__ == "__main__":
    main_loop()