Instagram API in Python — Runnable requests Code for Every Endpoint

Python — error handling

import os
import time
import requests

BASE = "https://api.socialscrape.dev/v1"
KEY = os.environ["INSCRAPE_KEY"]


class InScrapeError(RuntimeError):
    def __init__(self, status, code, message):
        super().__init__(f"{status} {code}: {message}")
        self.status = status
        self.code = code


def call(path, **params):
    # raise_for_status() throws away the body, and the body is where the API
    # tells you the error code and that credits_charged is 0. Read it first.
    res = requests.get(f"{BASE}{path}", params=params, headers={"x-api-key": KEY}, timeout=30)
    body = res.json()

    if res.status_code == 200:
        return body

    err = body.get("error", {})
    raise InScrapeError(res.status_code, err.get("code", "UNKNOWN"), err.get("message", ""))


def call_with_retry(path, attempts=3, **params):
    for attempt in range(attempts):
        try:
            return call(path, **params)
        except InScrapeError as exc:
            # Only 5xx is worth retrying. 4xx will fail the same way forever,
            # and only a 200 is ever charged either way.
            if exc.status < 500 or attempt == attempts - 1:
                raise
            time.sleep(2 ** attempt)


try:
    body = call_with_retry("/instagram/profile", handle="natgeo")
except InScrapeError as exc:
    if exc.code == "INSUFFICIENT_CREDITS":
        print("out of credits, top up before retrying")
    elif exc.code == "ENDPOINT_NOT_FOUND":
        print("API route not found")
    elif exc.code == "INVALID_API_KEY":
        print("check INSCRAPE_KEY")
    else:
        raise
else:
    if body.get("data", {}).get("profile") is None:
        print(body["data"]["profile"], "credits charged:", body["credits_charged"])
    else:
        print(body["data"]["profile"]["follower_count"], "credits left:", body["credits_remaining"])