Skip to content

Commit c126ba1

Browse files
Merge pull request #37 from ytinyui/main
Add response and request attributes to HTTPError
2 parents 6375fbf + 7fb2f1d commit c126ba1

File tree

2 files changed

+9
-2
lines changed

2 files changed

+9
-2
lines changed

tls_requests/exceptions.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,8 +21,14 @@
2121
class HTTPError(Exception):
2222
"""HTTP Error"""
2323

24-
def __init__(self, message: str) -> None:
24+
def __init__(self, message: str, **kwargs) -> None:
2525
self.message = message
26+
response = kwargs.pop("response", None)
27+
self.response = response
28+
self.request = kwargs.pop("request", None)
29+
if response is not None and not self.request and hasattr(response, "request"):
30+
self.request = self.response.request
31+
super().__init__(message, **kwargs)
2632

2733

2834
class ProtocolError(HTTPError):

tls_requests/models/response.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -197,7 +197,8 @@ def raise_for_status(self) -> "Response":
197197
else StatusCodes.get_reason(self.status_code)
198198
),
199199
self.url,
200-
)
200+
),
201+
response=self,
201202
)
202203

203204
return self

0 commit comments

Comments
 (0)