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

_make_response()增加自动设置host值功能

Browse Source
This commit is contained in:
g1879 2020-05-26 16:00:21 +08:00
parent 106e6a4ce5
commit 6d799fd807
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
DrissionPage/session_page.py
View File

@ -10,6 +10,7 @@ from random import random
from time import time from time import time
from typing import Union, List from typing import Union, List
from urllib import parse from urllib import parse
from urllib.parse import urlparse
from requests_html import HTMLSession, HTMLResponse from requests_html import HTMLSession, HTMLResponse
@ -180,14 +181,18 @@ class SessionPage(object):
if mode not in ['get', 'post']: if mode not in ['get', 'post']:
raise ValueError("mode must be 'get' or 'post'.") raise ValueError("mode must be 'get' or 'post'.")
# 设置referer值 # 设置referer和host值
if self._url: if self._url:
if 'headers' in set(x.lower() for x in kwargs): if 'headers' in set(x.lower() for x in kwargs):
if 'referer' not in set(x.lower() for x in kwargs['headers']): keys=set(x.lower() for x in kwargs['headers'])
if 'referer' not in keys:
kwargs['headers']['Referer'] = self._url kwargs['headers']['Referer'] = self._url
if 'host' not in keys:
kwargs['headers']['Host'] = urlparse(self._url).hostname
else: else:
kwargs['headers'] = self.session.headers kwargs['headers'] = self.session.headers
kwargs['headers']['Referer'] = self._url kwargs['headers']['Referer'] = self._url
kwargs['headers']['Host'] = urlparse(self._url).hostname
try: try:
r = None r = None