Run an End-of-Day Screen

"""End-of-day scan: run screen.json after the close and save the matches."""import csvimport datetime as dtimport jsonimport osimport sysimport time
import requests
API = "https://scanr.ai/api/v2"HEADERS = {"Authorization": f"Bearer {os.environ['SCANR_API_TOKEN']}"}

def call(method, path, **kwargs):    """One API call, retrying 429s and 5xx with backoff."""    for attempt in range(4):        response = requests.request(method, API + path, headers=HEADERS, timeout=120, **kwargs)        if response.status_code not in (429, 500, 502, 503, 504) or attempt == 3:            response.raise_for_status()            return response.json()        time.sleep(60 if response.status_code == 429 else 2**attempt)

def market_is_closed():    status = call("GET", "/misc/market-status")["status"]    return status.get("market") == "closed"

def latest_rs(tickers):    """Latest market RS (1-99) per ticker. Needs a Power User or Business token."""    since = int((time.time() - 5 * 86400) * 1000)  # the last few sessions, in ms    body = {"tickers": tickers, "from": since, "limit": 500}    latest = {}    while True:        page = call("POST", "/stocks/get-stock-indicators", json=body)        for row in page.get("results") or []:            seen = latest.get(row["ticker"])            if seen is None or row["date"] > seen["date"]:                latest[row["ticker"]] = row        if page.get("is_last_page", True):            break        body["cursor_date"] = page["next_cursor_date"]        body["cursor_ticker"] = page["next_cursor_ticker"]    return {ticker: row.get("rs_score") for ticker, row in latest.items()}

def main():    if not market_is_closed() and "--force" not in sys.argv:        sys.exit("Market is not closed yet; daily bars aren't settled. Use --force to run anyway.")
    with open("screen.json") as f:        payload = json.load(f)  # an exported or generated payload
    result = call("POST", "/scan", json={"instructions": payload})    tickers = result["tickers"] or []    print(f"{result['count']} matches")
    try:        rs = latest_rs(tickers) if tickers else {}    except requests.HTTPError as error:        if error.response.status_code != 403:            raise        rs = {}  # Starter plan: no RS column
    today = dt.date.today().isoformat()    with open(f"screen-{today}.csv", "w", newline="") as f:        writer = csv.writer(f)        writer.writerow(["date", "rank", "ticker", "rs_score"])        for rank, ticker in enumerate(tickers, start=1):            writer.writerow([today, rank, ticker, rs.get(ticker, "")])

if __name__ == "__main__":    main()