68 lines
1.8 KiB
Python
68 lines
1.8 KiB
Python
import time
|
|
|
|
from flask import Flask, request
|
|
from gevent import monkey
|
|
from playwright.sync_api import sync_playwright
|
|
from dotenv import load_dotenv
|
|
# 加载环境变量
|
|
load_dotenv()
|
|
monkey.patch_all()
|
|
|
|
app = Flask(__name__)
|
|
|
|
A1 = ""
|
|
|
|
|
|
def get_context_page(instance, stealth_js_path):
|
|
chromium = instance.chromium
|
|
browser = chromium.launch(headless=True)
|
|
context = browser.new_context()
|
|
context.add_init_script(path=stealth_js_path)
|
|
page = context.new_page()
|
|
return context, page
|
|
|
|
|
|
# 如下更改为 stealth.min.js 文件路径地址
|
|
stealth_js_path = "stealth.min.js"
|
|
print("正在启动 playwright")
|
|
playwright = sync_playwright().start()
|
|
browser_context, context_page = get_context_page(playwright, stealth_js_path)
|
|
context_page.goto("https://www.xiaohongshu.com")
|
|
print("正在跳转至小红书首页")
|
|
time.sleep(5)
|
|
context_page.reload()
|
|
time.sleep(1)
|
|
cookies = browser_context.cookies()
|
|
for cookie in cookies:
|
|
if cookie["name"] == "a1":
|
|
A1 = cookie["value"]
|
|
print("当前浏览器 cookie 中 a1 值为：" + cookie["value"] + "，请将需要使用的 a1 设置成一样方可签名成功")
|
|
print("跳转小红书首页成功，等待调用")
|
|
|
|
|
|
def sign(uri, data, a1, web_session):
|
|
encrypt_params = context_page.evaluate("([url, data]) => window._webmsxyw(url, data)", [uri, data])
|
|
return {
|
|
"x-s": encrypt_params["X-s"],
|
|
"x-t": str(encrypt_params["X-t"])
|
|
}
|
|
|
|
|
|
@app.route("/sign", methods=["POST"])
|
|
def hello_world():
|
|
json = request.json
|
|
uri = json["uri"]
|
|
data = json["data"]
|
|
a1 = json["a1"]
|
|
web_session = json["web_session"]
|
|
return sign(uri, data, a1, web_session)
|
|
|
|
|
|
@app.route("/a1", methods=["GET"])
|
|
def get_a1():
|
|
return {'a1': A1}
|
|
|
|
|
|
if __name__ == '__main__':
|
|
port = os.getenv('XHS_API_PORT', 5005)
|
|
app.run(host="0.0.0.0", port=port) |