Schedule a monthly export

"""Download last month's SimplyParse data for one parser as an .xlsx file."""

import os
from datetime import date, timedelta

import requests

PARSER_ID = os.environ["SIMPLYPARSE_PARSER_ID"]  # from app.simplyparse.com/parsers/<id>
TOKEN = os.environ["SIMPLYPARSE_API_TOKEN"]
NAME = os.environ.get("EXPORT_NAME", "export")


def last_month() -> tuple[date, date]:
    first_of_this_month = date.today().replace(day=1)
    last_day = first_of_this_month - timedelta(days=1)
    return last_day.replace(day=1), last_day


def main() -> None:
    start, end = last_month()
    response = requests.get(
        f"https://api.simplyparse.com/dapi/v1/parser/{PARSER_ID}/export/",
        headers={"Authorization": f"Token {TOKEN}"},
        params={"processed_from": start.isoformat(), "processed_to": end.isoformat()},
        timeout=300,
    )
    if response.status_code != 200:
        raise SystemExit(f"Export failed ({response.status_code}): {response.text[:300]}")

    path = f"{NAME}-{start:%Y-%m}.xlsx"
    with open(path, "wb") as f:
        f.write(response.content)
    print(f"Saved {path} ({len(response.content):,} bytes) for {start} to {end}")


if __name__ == "__main__":
    main()