diff --git a/gitmon/api.py b/gitmon/api.py index 1a9e7dd..f9abda8 100644 --- a/gitmon/api.py +++ b/gitmon/api.py @@ -6,13 +6,15 @@ from requests.packages.urllib3.util.retry import Retry class GitHubApi: - def __init__(self, proxy=None): + def __init__(self, headers=None, proxy=None): self.base_url = 'https://api.github.com' self.session = requests.Session() retries = Retry(total=3, backoff_factor=1) self.session.mount('https://', HTTPAdapter(max_retries=retries)) + if headers: + self.session.headers.update(headers) if proxy: self.session.proxies.update(proxy)