diff --git a/DrissionPage/session_page.py b/DrissionPage/session_page.py index 577fe23..1b8c92d 100644 --- a/DrissionPage/session_page.py +++ b/DrissionPage/session_page.py @@ -5,6 +5,7 @@ @File : session_page.py """ import os +import re from pathlib import Path from random import random from time import time @@ -196,7 +197,7 @@ class SessionPage(object): # 设置referer和host值 if self._url: if 'headers' in set(x.lower() for x in kwargs): - keys=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 if 'host' not in keys: @@ -216,13 +217,14 @@ class SessionPage(object): return_value = False else: headers = dict(r.headers) - if 'Content-Type' not in headers: - charset = 'utf-8' - else: - if 'charset' not in headers['Content-Type']: + if 'Content-Type' not in headers or 'charset' not in headers['Content-Type']: + re_result = re.search(r'