urllib3 and requests retry for REST and GraphQL GitHub API

github_retry.py

from urllib3 import BaseHTTPResponse, Retry
import time

class GitHubRetry(Retry):
    def get_retry_after(self, response: BaseHTTPResponse) -> float | None:
        if 'X-RateLimit-Remaining' in response.headers and 'X-RateLimit-Reset' in response.headers:
            remaining_requests = int(response.headers['X-RateLimit-Remaining'])
            if remaining_requests > 0:
                return None
            else:
                reset_time = int(response.headers['X-RateLimit-Reset'])
                current_time = int(time.time())
                return float(reset_time - current_time)
        return None