File size: 3,174 Bytes
d10e42a | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 | import asyncio
import json
import os
from rich.console import Console
from core.browser import get_browser
xpaths = {
"unique_id": """xpath=//*[contains(@id, "garfish_app_for_douyin_creator_pc_home")]/div/div[2]/div/div[2]/div[1]/div[2]/div[1]/div[3]""",
"name": """xpath=//*[contains(@id, "garfish_app_for_douyin_creator_pc_home")]/div/div[2]/div/div[2]/div[1]/div[2]/div[1]/div[1]/div[1]""",
}
console = Console()
async def userLogin():
playwright, browser = await get_browser(GUI=True)
try:
context = await browser.new_context()
page = await context.new_page()
# 打开目标页面
await page.goto("https://creator.douyin.com/")
print("请手动登录抖音创作者中心")
# 等待页面跳转或特定元素出现
# 等待 XPath 元素加载完成
await page.wait_for_selector(
'xpath=//*[contains(@id, "garfish_app_for_douyin_creator_pc_home")]/div/div[2]/div/div[2]/div[1]',
timeout=300000, # 设置输入超时时间为 5 分钟
)
# 等待 unique_id 元素加载完成
unique_id_element = await page.wait_for_selector(xpaths["unique_id"])
unique_id = (await unique_id_element.inner_text())[
4:
] # 去掉前四个字符 "抖音号："
print("Unique ID:", unique_id)
# 等待 name 元素加载完成
name_element = await page.wait_for_selector(xpaths["name"])
username = await name_element.inner_text()
print("Name:", username)
# 获取所有 Cookie
cookies = await context.cookies()
print("Cookies:", f"找到 {len(cookies)} 条 Cookie")
if os.path.exists("usersData.json"):
with open("usersData.json", "r", encoding="utf-8") as f:
userdata = json.load(f)
else:
userdata = []
targets = input(
"点击互动管理->私信管理->朋友私信，查看并输入目标好友对应昵称（空格分割）"
)
for user in userdata:
if user["unique_id"] == unique_id:
print(f"用户 {unique_id} 已存在，更新信息。")
user["cookies"] = cookies
user["username"] = username
user["targets"] = [target.strip() for target in targets.split(" ")]
break
else:
print(f"添加新用户 {unique_id} 。")
userdata.append(
{
"unique_id": unique_id,
"username": username,
"cookies": cookies,
"targets": [target.strip() for target in targets.split(" ")],
}
)
with open("usersData.json", "w", encoding="utf-8") as f:
json.dump(userdata, f, ensure_ascii=False, indent=4)
console.print(f"[bold green]登录完成！已添加用户 {username}[/bold green]")
finally:
await playwright.stop()
await browser.close()
if __name__ == "__main__":
asyncio.run(userLogin())
|