Include headers in GitHubError, refs #79

dogsheep · Apr 5, 2023 · 6eb97a2 · 6eb97a2
1 parent 2e84e26
commit 6eb97a2
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/github_to_sqlite/utils.py b/github_to_sqlite/utils.py
@@ -74,16 +74,17 @@
 
 
 class GitHubError(Exception):
-    def __init__(self, message, status_code):
+    def __init__(self, message, status_code, headers=None):
         self.message = message
         self.status_code = status_code
+        self.headers = headers
 
     @classmethod
     def from_response(cls, response):
         message = response.json()["message"]
         if "git repository is empty" in message.lower():
             cls = GitHubRepositoryEmpty
-        return cls(message, response.status_code)
+        return cls(message, response.status_code, response.headers)
 
 
 class GitHubRepositoryEmpty(GitHubError):