from selenium import webdriver
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.common.by import By
import time
def send_douyin_message(username, message):
# 设置 WebDriver
driver = webdriver.Chrome()
driver.get('https://www.douyin.com')
# 等待页面加载
time.sleep(5)
# 登录(假设你已经手动登录或有合适的自动化登录方式)
# 找到并点击“消息”图标
try:
message_icon = driver.find_element(By.CSS_SELECTOR, '#message-icon')
message_icon.click()
except Exception as e:
print(f"Error clicking message icon: {e}")
# 等待消息页面加载
time.sleep(5)
# 点击“新消息”
try:
new_message_button = driver.find_element(By.CSS_SELECTOR, '#new-message-button')
new_message_button.click()
except Exception as e:
print(f"Error clicking new message button: {e}")
# 输入接收者用户名
try:
recipient_input = driver.find_element(By.CSS_SELECTOR, '#recipient-input')
recipient_input.send_keys(username)
recipient_input.send_keys(Keys.RETURN)
except Exception as e:
print(f"Error entering recipient username: {e}")
# 输入消息内容
try:
message_input = driver.find_element(By.CSS_SELECTOR, '#message-input')
message_input.send_keys(message)
message_input.send_keys(Keys.RETURN)
except Exception as e:
print(f"Error entering message: {e}")
# 等待一会儿
time.sleep(5)
# 关闭浏览器
driver.quit()
# 使用示例
send_douyin_message('1103', '你好,这是改进后的自动发送消息。')