diff --git a/server.py b/server.py index e4a3020..0d594d1 100644 --- a/server.py +++ b/server.py @@ -11,7 +11,7 @@ monkey.patch_all() app = Flask(__name__) -A1 = "" +global_a1 = "" # 确保在模块级别声明全局变量 browser_context = None context_page = None @@ -39,13 +39,14 @@ time.sleep(1) cookies = browser_context.cookies() for cookie in cookies: if cookie["name"] == "a1": - A1 = cookie["value"] + global_a1 = cookie["value"] print("当前浏览器 cookie 中 a1 值为:" + cookie["value"] + ",请将需要使用的 a1 设置成一样方可签名成功") print("跳转小红书首页成功,等待调用") def sign(uri, data, a1, web_session): global browser_context, context_page # 声明全局变量 + global global_a1 try: # 确保页面仍然有效,如果页面已关闭则重新初始化 if context_page is None or context_page.is_closed(): @@ -54,8 +55,15 @@ def sign(uri, data, a1, web_session): time.sleep(5) context_page.reload() 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]) return { "x-s": encrypt_params["X-s"],