kenzuya/unshackle
1
0
Fork 2
mirror of https://github.com/unshackle-dl/unshackle.git synced 2026-05-16 21:59:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(session): Optimize header handling in session requests

Browse Source 
Removed redundant conversion of headers to dict for requests.
This commit is contained in:
CodeName393
2026-04-25 14:53:07 +09:00
committed by GitHub
parent 9dbdf9804f
commit bddb305c5d
1 changed files with 4 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
unshackle/core/session.py
View File

@@ -584,11 +584,12 @@ class RnetSession:
if rnet_method is None: if rnet_method is None:
raise ValueError(f"Unsupported HTTP method: {method}") raise ValueError(f"Unsupported HTTP method: {method}")
# Convert headers to standard dict once to resolve PyO3 CaseInsensitiveDict rejection.
if kwargs.get("headers") is not None:
kwargs["headers"] = dict(kwargs["headers"])
# Skip retry for non-allowed methods # Skip retry for non-allowed methods
if method_upper not in self.allowed_methods: if method_upper not in self.allowed_methods:
if "headers" in kwargs and kwargs["headers"] is not None:
kwargs["headers"] = dict(kwargs["headers"])
raw_resp = client.request(rnet_method, url, **kwargs) raw_resp = client.request(rnet_method, url, **kwargs)
return RnetResponse(raw_resp) return RnetResponse(raw_resp)
@@ -597,9 +598,6 @@ class RnetSession:
for attempt in range(self.max_retries + 1): for attempt in range(self.max_retries + 1):
try: try:
if "headers" in kwargs and kwargs["headers"] is not None:
kwargs["headers"] = dict(kwargs["headers"])
raw_resp = client.request(rnet_method, url, **kwargs) raw_resp = client.request(rnet_method, url, **kwargs)
response = RnetResponse(raw_resp) response = RnetResponse(raw_resp)
if response.status_code not in self.status_forcelist: if response.status_code not in self.status_forcelist: