diff --git a/gitmon/api.py b/gitmon/api.py index 41d3222..b740db7 100644 --- a/gitmon/api.py +++ b/gitmon/api.py @@ -8,13 +8,15 @@ from requests.packages.urllib3.util.retry import Retry class GitHubApi: - def __init__(self, headers=None, proxy=None): + def __init__(self, token=None, 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 token: + self.session.headers.update({'Authorization': f'Bearer {token}'}) if headers: self.session.headers.update(headers) if proxy: