from selenium import webdriver
import time
import pickle
import random
import save_cookies
from selenium.webdriver.common.by import By

login = input('是否需要重新授权登录（yes:no）：')
if login == 'yes':
    save_cookies.func()
url = input('请输入直播间链接：')
with open("抖音cookies文件.pickle",'rb') as file:
    cookiesList = pickle.load(file)
#忽略SSL证书
chrome_options = webdriver.ChromeOptions()
chrome_options.add_argument('--ignore-certificate-errors')
browser = webdriver.Chrome(options=chrome_options)

browser.get('https://www.douyin.com/')
# input("请输入任意键继续...")
for cookie in cookiesList:
    browser.add_cookie(cookie)
browser.get(url)
time.sleep(5)#等待几秒，防止网络原因退出

remarks = [
    'python测试弹幕发送，捕捉，机器人回复，当前速度3秒每条',
]
while True:
    textElement = browser.find_element(By.XPATH, '//*[@class="webcast-chatroom___textarea"]')
    textElement.clear()
    textElement.send_keys(random.choice(remarks)) # 输入新字符串
    sendElement = browser.find_element(By.XPATH, '//*[@class="webcast-chatroom___send-btn"][@type="button"]')
    time.sleep(3)#设置发送速度
    sendElement.click()

browser.quit()