Status and polling

import time

import requests

API = "https://api.aigenda.tech"


def wait_for_result(token: str, space_id: int, audio_id: int, timeout: int = 3600) -> dict:
    """Waits for the processing result. Returns the result/ response body or raises an exception."""
    headers = {"Authorization": f"Bearer {token}"}
    url = f"{API}/api/spaces/{space_id}/audio/{audio_id}/result/"
    started = time.monotonic()
    not_found_since = None
    failing_since = None

    while time.monotonic() - started < timeout:
        r = requests.get(url, headers=headers, timeout=30)
        now = time.monotonic()

        if r.status_code == 200:
            not_found_since = failing_since = None
            body = r.json()
            if body.get("classified"):
                if body.get("success"):
                    return body
                raise RuntimeError("Processing failed, the minutes will be returned automatically")
        elif r.status_code == 404:
            # right after classify the status may not update instantly
            not_found_since = not_found_since or now
            if now - not_found_since > 60:
                raise RuntimeError("Processing not started: call classify")
        elif r.status_code >= 500:
            failing_since = failing_since or now
            if now - failing_since > 180:
                raise RuntimeError("Processing failed, the minutes will be returned automatically")

        time.sleep(15)

    raise TimeoutError("The result was not ready within the allotted time")