blackteay/DrissionPage
1
0
Fork 0
mirror of https://gitee.com/g1879/DrissionPage.git synced 2024-12-10 04:00:23 +08:00
Code Issues Projects Releases Wiki Activity

extra_info中的headers补充到headers中

Browse Source
This commit is contained in:
g1879 2024-09-29 17:53:38 +08:00
parent 7d936defd0
commit 430b304a22
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
DrissionPage/_units/listener.py
View File

@ -451,6 +451,11 @@ class Request(object):
def headers(self): def headers(self):
if self._headers is None: if self._headers is None:
self._headers = CaseInsensitiveDict(self._request['headers']) self._headers = CaseInsensitiveDict(self._request['headers'])
if self.extra_info.headers:
h = CaseInsensitiveDict(self.extra_info.headers)
for k, v in h.items():
if k not in self._headers:
self._headers[k] = v
return self._headers return self._headers
@property @property
@ -498,6 +503,11 @@ class Response(object):
def headers(self): def headers(self):
if self._headers is None: if self._headers is None:
self._headers = CaseInsensitiveDict(self._response['headers']) self._headers = CaseInsensitiveDict(self._response['headers'])
if self.extra_info.headers:
h = CaseInsensitiveDict(self.extra_info.headers)
for k, v in h.items():
if k not in self._headers:
self._headers[k] = v
return self._headers return self._headers
@property @property