Add header support

2023-11-10 19:42:16 -05:00 · 2023-11-10 19:42:16 -05:00 · 8a9200e627
parent d00b009706
commit 8a9200e627
1 changed files with 3 additions and 1 deletions
--- 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)