bacow/AV_Data_Capture
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

check add cookies

Browse Source
This commit is contained in:
lededev
2021-09-23 03:56:55 +08:00
parent 446eb166a3
commit d6677b717d
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
ADC_function.py
View File

@@ -86,17 +86,20 @@ def post_html(url: str, query: dict, headers: dict = None) -> requests.Response:
def get_html_by_form(url, form_name: str = None, fields: dict = None, cookies: dict = None, ua: str = None, return_type: str = None): def get_html_by_form(url, form_name: str = None, fields: dict = None, cookies: dict = None, ua: str = None, return_type: str = None):
browser = mechanicalsoup.StatefulBrowser(user_agent=G_USER_AGENT if ua is None else ua) browser = mechanicalsoup.StatefulBrowser(user_agent=G_USER_AGENT if ua is None else ua)
if isinstance(cookies, dict):
requests.utils.add_dict_to_cookiejar(browser.session.cookies, cookies)
configProxy = config.Config().proxy() configProxy = config.Config().proxy()
if configProxy.enable: if configProxy.enable:
browser.session.proxies = configProxy.proxies() browser.session.proxies = configProxy.proxies()
result = browser.open(url) result = browser.open(url)
if not result.ok:
return ''
form = browser.select_form() if form_name is None else browser.select_form(form_name) form = browser.select_form() if form_name is None else browser.select_form(form_name)
if isinstance(fields, dict): if isinstance(fields, dict):
for k, v in fields.items(): for k, v in fields.items():
browser[k] = v browser[k] = v
response = browser.submit_selected() response = browser.submit_selected()
response.encoding = "utf-8" response.encoding = "utf-8"
if return_type == "object": if return_type == "object":
return response return response
elif return_type == "content": elif return_type == "content":