421 lines
17 KiB
Python
421 lines
17 KiB
Python
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 json
|
||
import base64
|
||
import time
|
||
|
||
class DouyinPublisher():
|
||
def __init__(self, driver, cookies_path):
|
||
self.driver = driver
|
||
self.cookies_path = cookies_path
|
||
self.cookies = json.loads(open(self.cookies_path, "r").read())
|
||
|
||
# 先访问抖音网站
|
||
self.driver.get("https://creator.douyin.com/creator-micro/content/upload?enter_from=dou_web")
|
||
time.sleep(3) # 等待页面加载
|
||
|
||
# 然后加载cookies
|
||
self.load_cookies()
|
||
|
||
self.WEB_URL = "https://creator.douyin.com/creator-micro/content/upload?enter_from=dou_web"
|
||
self.driver.get(self.WEB_URL)
|
||
|
||
def load_cookies(self):
|
||
try:
|
||
for cookie in self.cookies:
|
||
# 移除可能导致问题的属性
|
||
for attr in ['sameSite', 'expiry']:
|
||
if attr in cookie:
|
||
del cookie[attr]
|
||
try:
|
||
self.driver.add_cookie(cookie)
|
||
except Exception as e:
|
||
print(f"添加cookie失败: {e}")
|
||
|
||
print("cookies加载完成")
|
||
# 刷新页面使cookies生效
|
||
self.driver.refresh()
|
||
time.sleep(3) # 等待刷新完成
|
||
|
||
except Exception as e:
|
||
print(f"添加cookies过程中发生错误: {e}")
|
||
return True
|
||
|
||
def open_web(self):
|
||
self.driver.get(self.WEB_URL)
|
||
# 等待页面加载完成
|
||
time.sleep(5)
|
||
return True
|
||
|
||
def select_upload_type(self, upload_type):
|
||
# 注意：以下XPath选择器需要根据实际网页结构调整
|
||
if upload_type == "image":
|
||
# 图文
|
||
try:
|
||
self.driver.find_element(By.XPATH, "//div[contains(text(), '图文')]").click()
|
||
except:
|
||
print("无法找到图文上传按钮，请检查XPath")
|
||
elif upload_type == "video":
|
||
# 视频
|
||
try:
|
||
self.driver.find_element(By.XPATH, "//div[contains(text(), '视频')]").click()
|
||
except:
|
||
print("无法找到视频上传按钮，请检查XPath")
|
||
return True
|
||
|
||
def upload_image(self, image_path):
|
||
# 注意：以下XPath选择器需要根据实际网页结构调整
|
||
try:
|
||
# 找到文件上传输入框
|
||
file_input = self.driver.find_element(By.XPATH, "//input[@type='file' and @accept='.jpg,.jpeg,.png']")
|
||
file_input.send_keys(image_path)
|
||
except Exception as e:
|
||
print(f"上传图片失败: {e}")
|
||
|
||
# 等待上传完成
|
||
time.sleep(5)
|
||
return True
|
||
|
||
def upload_video(self, video_path):
|
||
# 注意：以下XPath选择器需要根据实际网页结构调整
|
||
try:
|
||
# 找到文件上传输入框
|
||
# 首先尝试点击上传视频按钮（如果需要）
|
||
try:
|
||
# 尝试通过文本内容定位上传视频按钮
|
||
upload_btn = self.driver.find_element(By.XPATH, "//span[contains(text(), '上传视频')]")
|
||
upload_btn.click()
|
||
time.sleep(1) # 等待文件选择框出现
|
||
except Exception as e:
|
||
print(f"点击上传视频按钮失败，尝试直接查找文件输入框: {e}")
|
||
|
||
# 然后找到文件输入框并上传文件
|
||
file_input = self.driver.find_element(By.XPATH, "//input[@type='file' and @accept='.mp4,.mov,.wmv,.avi,.m4v']")
|
||
file_input.send_keys(video_path)
|
||
except Exception as e:
|
||
print(f"上传视频失败: {e}")
|
||
|
||
# 等待上传完成
|
||
time.sleep(10) # 视频上传可能需要更长时间
|
||
return True
|
||
|
||
def publish(self, title, content):
|
||
# 注意：以下XPath选择器需要根据实际网页结构调整
|
||
try:
|
||
# 输入标题
|
||
title_input = self.driver.find_element(By.XPATH, "//input[@placeholder='请输入标题']")
|
||
title_input.clear()
|
||
title_input.send_keys(title)
|
||
|
||
# 输入内容/描述
|
||
content_input = self.driver.find_element(By.XPATH, "//div[contains(@class, 'editor')]")
|
||
content_input.clear()
|
||
content_input.send_keys(content)
|
||
except Exception as e:
|
||
print(f"输入标题和内容失败: {e}")
|
||
|
||
return True
|
||
|
||
def click_publish(self):
|
||
# 注意：以下XPath选择器需要根据实际网页结构调整
|
||
try:
|
||
# 找到并点击发布按钮
|
||
publish_button = self.driver.find_element(By.XPATH, "//button[contains(text(), '发布')]")
|
||
publish_button.click()
|
||
|
||
# 等待发布完成
|
||
time.sleep(5)
|
||
|
||
# 有时候可能会有确认弹窗
|
||
try:
|
||
confirm_button = self.driver.find_element(By.XPATH, "//button[contains(text(), '确定')]")
|
||
confirm_button.click()
|
||
except:
|
||
pass # 如果没有确认弹窗，则忽略
|
||
except Exception as e:
|
||
print(f"点击发布按钮失败: {e}")
|
||
|
||
return True
|
||
|
||
def publish_article(self, title, content, image_path):
|
||
self.select_upload_type("image")
|
||
self.upload_image(image_path)
|
||
self.publish(title, content)
|
||
self.click_publish()
|
||
return True
|
||
|
||
def publish_video(self, title, content, video_path):
|
||
self.select_upload_type("video")
|
||
self.upload_video(video_path)
|
||
self.publish(title, content)
|
||
self.click_publish()
|
||
return True
|
||
|
||
def work(self, type, title, content, image_path=None, video_path=None):
|
||
try:
|
||
self.open_web()
|
||
self.driver.maximize_window()
|
||
self.driver.implicitly_wait(10)
|
||
|
||
if type == "article" and image_path:
|
||
self.publish_article(title, content, image_path)
|
||
elif type == "video" and video_path:
|
||
self.publish_video(title, content, video_path)
|
||
else:
|
||
raise ValueError("Invalid type or missing file path")
|
||
|
||
print("发布完成!")
|
||
|
||
except Exception as e:
|
||
print(f"发布失败: {e}")
|
||
finally:
|
||
# 可以选择不立即退出，以便查看结果
|
||
# self.driver.quit()
|
||
pass
|
||
|
||
return True
|
||
|
||
# 新增方法：封装团购选择流程
|
||
def select_group_buy(self, product_name, product_info):
|
||
try:
|
||
# 下拉框选择
|
||
tab_selector = self.driver.find_element(By.XPATH, "//div[@class='semi-select select-lJTtRL semi-select-single']")
|
||
self.driver.execute_script("arguments[0].click();", tab_selector)
|
||
time.sleep(2)
|
||
|
||
# 选择团购
|
||
tab_selector_option = self.driver.find_element(By.XPATH, "//div[@data-code='13010' and @class='select-dropdown-option-video']")
|
||
self.driver.execute_script("arguments[0].click();", tab_selector_option)
|
||
time.sleep(2)
|
||
|
||
# 点击产品选择下拉框
|
||
product_selector = self.driver.find_element(By.XPATH, "//div[@class='semi-select select-Qm4u8S semi-select-single semi-select-filterable']")
|
||
self.driver.execute_script("arguments[0].click();", product_selector)
|
||
time.sleep(2)
|
||
|
||
# 输入产品名称
|
||
product_input = self.driver.find_element(By.XPATH, "//input[@class='semi-input semi-input-default' and @placeholder='']")
|
||
product_input.send_keys(product_name)
|
||
time.sleep(5)
|
||
|
||
# 选择产品
|
||
max_retries = 10
|
||
for attempt in range(max_retries):
|
||
clicked = self.driver.execute_script("""
|
||
let elements = document.querySelectorAll('.semi-select-option');
|
||
for (let el of elements) {
|
||
if (el.textContent.includes(arguments[0])) {
|
||
el.click();
|
||
return true;
|
||
}
|
||
}
|
||
return false;
|
||
""", product_name)
|
||
|
||
if clicked:
|
||
print(f"成功点击产品选项，尝试次数: {attempt+1}")
|
||
break
|
||
else:
|
||
print(f"等待产品选项出现，尝试次数: {attempt+1}/{max_retries}")
|
||
time.sleep(1)
|
||
|
||
time.sleep(2)
|
||
|
||
# 输入产品信息
|
||
product_info_input = self.driver.find_element(By.XPATH, "//input[@class='semi-input semi-input-default' and @placeholder='如：海底捞超值双人套餐']")
|
||
product_info_input.send_keys(product_info)
|
||
time.sleep(2)
|
||
|
||
# 点击确认按钮
|
||
confirm_button = self.driver.find_element(By.XPATH, "//div[@class='footer-button-DL8zDh']")
|
||
self.driver.execute_script("arguments[0].click();", confirm_button)
|
||
time.sleep(2)
|
||
|
||
return True
|
||
except Exception as e:
|
||
print(f"选择团购失败: {e}")
|
||
return False
|
||
|
||
# 新增方法：设置定时发布
|
||
def set_schedule_time(self, time_str):
|
||
try:
|
||
# 选择定时发布
|
||
schedule_label = self.driver.find_element(By.XPATH, "//label[contains(@class, 'radio-d4zkru')]//span[contains(text(), '定时发布')]/ancestor::label")
|
||
self.driver.execute_script("arguments[0].click();", schedule_label)
|
||
time.sleep(2)
|
||
|
||
# 输入时间
|
||
time_inputor = self.driver.find_element(By.XPATH, "//input[@class='semi-input semi-input-default' and @type='text' and @placeholder='日期和时间']")
|
||
time_inputor.send_keys(time_str)
|
||
time.sleep(2)
|
||
|
||
return True
|
||
except Exception as e:
|
||
print(f"设置定时发布失败: {e}")
|
||
return False
|
||
|
||
# 新增方法：最后确认接口
|
||
def final_confirmation(self, prompt="按任意键继续..."):
|
||
input(prompt)
|
||
return True
|
||
|
||
if __name__ == "__main__":
|
||
# 使用Safari浏览器
|
||
driver = webdriver.Safari()
|
||
|
||
# 也可以根据需要选择其他浏览器
|
||
# driver = webdriver.Chrome()
|
||
# driver = webdriver.Firefox()
|
||
|
||
# 设置浏览器全屏
|
||
driver.maximize_window()
|
||
|
||
# 设置页面加载超时时间
|
||
driver.set_page_load_timeout(30)
|
||
|
||
# cookies路径，需要先获取登录后的cookies
|
||
cookies_path = "cookies.json"
|
||
|
||
# 创建抖音发布器实例
|
||
publisher = DouyinPublisher(driver, cookies_path)
|
||
time.sleep(10)
|
||
# <button class="semi-button semi-button-primary container-drag-btn-k6XmB4 semi-button-with-icon" type="button"><span class="semi-button-content"><svg xmlns="http://www.w3.org/2000/svg" width="20" height="20" fill="none" viewBox="0 0 20 20"><rect width="16.333" height="13" x="1.833" y="3.5" stroke="#fff" stroke-width="2" rx="2"></rect><path stroke="#fff" stroke-linecap="round" stroke-width="2" d="M10 7.5v5M12.5 10h-5"></path></svg><span class="semi-button-content-right">上传视频</span></span></button>
|
||
|
||
# 上传视频按钮
|
||
video_path = "/Users/yarrow/autoPublisher/8970_1747905081.mp4"
|
||
file_input = driver.find_element(By.XPATH, "//input[@style='display: none;']")
|
||
file_input.send_keys(video_path)
|
||
time.sleep(10)
|
||
print("视频上传完成")
|
||
# input("按任意键继续...")
|
||
|
||
title_input = driver.find_element(By.XPATH, "//input[@placeholder='填写作品标题，为作品获得更多流量']")
|
||
title_input.send_keys("测试标题")
|
||
|
||
content_input = driver.find_element(By.XPATH, "//div[@data-placeholder='添加作品简介']")
|
||
content_input.send_keys("测试内容")
|
||
print("内容输入完成")
|
||
# input("按任意键继续...")
|
||
|
||
# 下拉框选择
|
||
tab_selector = driver.find_element(By.XPATH, "//div[@class='semi-select select-lJTtRL semi-select-single']")
|
||
# 方法1：使用JavaScript执行点击
|
||
driver.execute_script("arguments[0].click();", tab_selector)
|
||
|
||
# 如果上面方法不生效，可以尝试以下备选方法：
|
||
# 方法2：使用Actions类进行点击
|
||
# from selenium.webdriver.common.action_chains import ActionChains
|
||
# ActionChains(driver).move_to_element(tab_selector).click().perform()
|
||
|
||
# 方法3：使用显式等待确保元素可点击
|
||
# WebDriverWait(driver, 10).until(EC.element_to_be_clickable((By.XPATH, "//div[@class='semi-select select-lJTtRL semi-select-single']"))).click()
|
||
|
||
time.sleep(2)
|
||
print("下拉框选择完成")
|
||
# input("按任意键继续...")
|
||
tab_selector_option = driver.find_element(By.XPATH, "//div[@class='semi-select-option-list semi-select-option-list-chosen']")
|
||
|
||
## 选择团购
|
||
tab_selector_option = driver.find_element(By.XPATH, "//div[@data-code='13010' and @class='select-dropdown-option-video']")
|
||
driver.execute_script("arguments[0].click();", tab_selector_option)
|
||
time.sleep(2)
|
||
print("团购选择完成")
|
||
# input("按任意键继续...")
|
||
|
||
## 点击新的下拉框
|
||
product_selector = driver.find_element(By.XPATH, "//div[@class='semi-select select-Qm4u8S semi-select-single semi-select-filterable']")
|
||
driver.execute_script("arguments[0].click();", product_selector)
|
||
time.sleep(2)
|
||
print("新的下拉框选择完成")
|
||
# input("按任意键继续...")
|
||
|
||
product_input = driver.find_element(By.XPATH, "//input[@class='semi-input semi-input-default' and @placeholder='']")
|
||
product_input.send_keys("古龙峡漂流|国际(勇猛)赛道全程漂流票 成人票")
|
||
print("产品输入完成")
|
||
# input("按任意键继续...")
|
||
time.sleep(5)
|
||
|
||
## 下拉框选择这里有问题 找不到下拉框
|
||
# 添加等待和重试机制
|
||
max_retries = 10
|
||
for attempt in range(max_retries):
|
||
# 使用JavaScript直接查找并点击元素，与控制台操作相同
|
||
clicked = driver.execute_script("""
|
||
// 尝试精确匹配类名
|
||
let elements = document.getElementsByClassName('semi-select-option semi-select-option-focused option-v2-U5PXMk');
|
||
if (elements.length > 0) {
|
||
elements[0].click();
|
||
return true;
|
||
}
|
||
|
||
// 备选方案1：使用更宽泛的类名查找
|
||
elements = document.querySelectorAll('.semi-select-option');
|
||
for (let el of elements) {
|
||
if (el.textContent.includes('古龙峡漂流')) {
|
||
el.click();
|
||
return true;
|
||
}
|
||
}
|
||
|
||
// 备选方案2：使用aria-role查找
|
||
elements = document.querySelectorAll('[role="option"]');
|
||
for (let el of elements) {
|
||
if (el.textContent.includes('古龙峡漂流')) {
|
||
el.click();
|
||
return true;
|
||
}
|
||
}
|
||
|
||
return false;
|
||
""")
|
||
|
||
if clicked:
|
||
print(f"成功点击产品选项，尝试次数: {attempt+1}")
|
||
break
|
||
else:
|
||
print(f"等待产品选项出现，尝试次数: {attempt+1}/{max_retries}")
|
||
time.sleep(1) # 等待1秒后重试
|
||
|
||
# 等待一会，让点击生效
|
||
time.sleep(2)
|
||
print("产品选择完成")
|
||
# input("按任意键继续...")
|
||
|
||
product_info_input = driver.find_element(By.XPATH, "//input[@class='semi-input semi-input-default' and @placeholder='如：海底捞超值双人套餐']")
|
||
product_info_input.send_keys("古龙峡漂流成人票")
|
||
print("产品信息输入完成")
|
||
# input("按任意键继续...")
|
||
|
||
confirm_button = driver.find_element(By.XPATH, "//div[@class='footer-button-DL8zDh']")
|
||
driver.execute_script("arguments[0].click();", confirm_button)
|
||
time.sleep(2)
|
||
print("确认完成")
|
||
# input("按任意键继续...")
|
||
|
||
# 例如选择"定时发布"选项
|
||
schedule_label = driver.find_element(By.XPATH, "//label[contains(@class, 'radio-d4zkru')]//span[contains(text(), '定时发布')]/ancestor::label")
|
||
driver.execute_script("arguments[0].click();", schedule_label)
|
||
# driver.execute_script("arguments[0].click();", time_selector_label)
|
||
time.sleep(2)
|
||
print("定时发布切换完成")
|
||
time_inputor = driver.find_element(By.XPATH, "//input[@class='semi-input semi-input-default' and @type='text' and @placeholder='日期和时间']")
|
||
time_inputor.send_keys("2025-05-27 10:00")
|
||
|
||
time.sleep(10)
|
||
print("定时发布选择完成")
|
||
input("按任意键继续...")
|
||
|
||
|
||
# 发布图文
|
||
# publisher.work(type="article", title="测试标题", content="测试内容 #测试话题", image_path="/path/to/your/image.jpg")
|
||
|
||
# 发布视频
|
||
# publisher.work(type="video", title="测试视频", content="测试视频内容 #测试话题", video_path="/Users/mac/Desktop/test.mp4")
|
||
|
||
# 发布后保持浏览器窗口开启，方便检查结果
|
||
input("按任意键退出...")
|
||
driver.quit() |