69 lines
2.5 KiB
Python
69 lines
2.5 KiB
Python
import requests
|
||
import selenium
|
||
from selenium import webdriver
|
||
from selenium.webdriver.common.by import By
|
||
from selenium.webdriver.support.ui import WebDriverWait
|
||
from selenium.webdriver.support import expected_conditions as EC
|
||
import time
|
||
import json
|
||
import base64
|
||
|
||
if __name__ == "__main__":
|
||
url = "https://creator.xiaohongshu.com/login"
|
||
|
||
driver = webdriver.Safari()
|
||
driver.maximize_window()
|
||
driver.get(url)
|
||
time.sleep(5)
|
||
WebDriverWait(driver, 10).until(EC.presence_of_element_located((By.XPATH, "//img[@class='css-wemwzq']")))
|
||
driver.find_element(By.XPATH, "//img[@class='css-wemwzq']").click()
|
||
# time.sleep(10)
|
||
|
||
# save img
|
||
## 等待跳转
|
||
WebDriverWait(driver, 10).until(EC.presence_of_element_located((By.CLASS_NAME, "css-1lhmg90")))
|
||
# 获取图片
|
||
time.sleep(3)
|
||
img_element = driver.find_element(By.CLASS_NAME, "css-1lhmg90")
|
||
img_src = img_element.get_attribute("src")
|
||
print(f"img_src: {img_src}")
|
||
img_data = base64.b64decode(img_src.split(",")[1])
|
||
print(img_data)
|
||
with open("qrcode.png", "wb") as f:
|
||
f.write(img_data)
|
||
|
||
# 打印提示信息，请用户扫描二维码
|
||
print("二维码已保存为 qrcode.png，请使用小红书 App 扫描登录")
|
||
print("等待登录中...")
|
||
|
||
# 保存当前 URL 用于检测页面跳转
|
||
current_url = driver.current_url
|
||
|
||
# 等待页面跳转（登录成功）
|
||
# 方法1：检测 URL 变化
|
||
try:
|
||
# 使用一个较长的超时时间，给用户足够时间扫码
|
||
WebDriverWait(driver, 120).until(lambda d: d.current_url != current_url)
|
||
print("检测到页面跳转，登录成功！")
|
||
except:
|
||
# 如果 URL 没有变化，尝试方法2
|
||
print("URL 没有变化，尝试检测登录状态元素...")
|
||
|
||
# 方法2：检测登录后特定元素的出现
|
||
# try:
|
||
# # 例如，检测用户头像或其他登录后才会出现的元素
|
||
# # 这里以用户头像为例，需要根据实际网页元素调整
|
||
# WebDriverWait(driver, 30).until(EC.presence_of_element_located((By.CLASS_NAME, "user-avatar")))
|
||
# print("检测到用户登录元素，登录成功！")
|
||
# except:
|
||
# print("登录超时或失败，请重试")
|
||
|
||
# 登录成功后，保存 cookies
|
||
cookies = driver.get_cookies()
|
||
with open("cookies_xhs.json", "w") as f:
|
||
json.dump(cookies, f, indent=4)
|
||
print(f"已保存 cookies 到 cookies_xhs.json 文件")
|
||
|
||
# 关闭浏览器
|
||
driver.quit()
|