update reset a1 module

This commit is contained in:
熊潇 2025-05-01 20:27:36 +08:00
parent 837ae3e01b
commit c2d83a4927

View File

@ -11,7 +11,7 @@ monkey.patch_all()
app = Flask(__name__) app = Flask(__name__)
A1 = "" global_a1 = ""
# 确保在模块级别声明全局变量 # 确保在模块级别声明全局变量
browser_context = None browser_context = None
context_page = None context_page = None
@ -39,13 +39,14 @@ time.sleep(1)
cookies = browser_context.cookies() cookies = browser_context.cookies()
for cookie in cookies: for cookie in cookies:
if cookie["name"] == "a1": if cookie["name"] == "a1":
A1 = cookie["value"] global_a1 = cookie["value"]
print("当前浏览器 cookie 中 a1 值为:" + cookie["value"] + ",请将需要使用的 a1 设置成一样方可签名成功") print("当前浏览器 cookie 中 a1 值为:" + cookie["value"] + ",请将需要使用的 a1 设置成一样方可签名成功")
print("跳转小红书首页成功,等待调用") print("跳转小红书首页成功,等待调用")
def sign(uri, data, a1, web_session): def sign(uri, data, a1, web_session):
global browser_context, context_page # 声明全局变量 global browser_context, context_page # 声明全局变量
global global_a1
try: try:
# 确保页面仍然有效,如果页面已关闭则重新初始化 # 确保页面仍然有效,如果页面已关闭则重新初始化
if context_page is None or context_page.is_closed(): if context_page is None or context_page.is_closed():
@ -54,8 +55,15 @@ def sign(uri, data, a1, web_session):
time.sleep(5) time.sleep(5)
context_page.reload() context_page.reload()
time.sleep(1) time.sleep(1)
# 执行 JavaScript 函数 if a1 != global_a1:
browser_context.add_cookies([
{'name': 'a1', 'value': a1, 'domain': ".xiaohongshu.com", 'path': "/"}
])
context_page.reload()
time.sleep(1)
global_a1 = a1
# 执行 JavaScript 函数
encrypt_params = context_page.evaluate("([url, data]) => window._webmsxyw(url, data)", [uri, data]) encrypt_params = context_page.evaluate("([url, data]) => window._webmsxyw(url, data)", [uri, data])
return { return {
"x-s": encrypt_params["X-s"], "x-s": encrypt_params["X-s"],