awaioi/EveryPublish
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

1. use loguru for clean console message

Browse Source
This commit is contained in:
pookz@stme
2024-06-14 16:19:10 +08:00
parent 096f5f87c2
commit 38bcc315bb
7 changed files with 108 additions and 67 deletions
Download Patch File Download Diff File Expand all files Collapse all files
douyin_uploader/main.py
+17 -19
View File
@@ -7,6 +7,7 @@ import asyncio
from conf import LOCAL_CHROME_PATH
from utils.base_social_media import set_init_script
from utils.log import douyin_logger
async def cookie_auth(account_file):
@@ -20,10 +21,10 @@ async def cookie_auth(account_file):
await page.goto("https://creator.douyin.com/creator-micro/content/upload")
try:
await page.wait_for_selector("div.boards-more h3:text('抖音排行榜')", timeout=5000) # 等待5秒
print("[+] 等待5秒 cookie 失效")
douyin_logger.error(f'等待5秒 cookie 失效')
return False
except:
print("[+] cookie 有效")
douyin_logger.success(f'cookie 有效')
return True
@@ -32,7 +33,7 @@ async def douyin_setup(account_file, handle=False):
if not handle:
# Todo alert message
return False
print('[+] cookie文件不存在或已失效，即将自动打开浏览器，请扫码登录，登陆后会自动生成cookie文件')
douyin_logger.info('[+] cookie文件不存在或已失效，即将自动打开浏览器，请扫码登录，登陆后会自动生成cookie文件')
await douyin_cookie_gen(account_file)
return True
@@ -82,7 +83,7 @@ class DouYinVideo(object):
await asyncio.sleep(1)
async def handle_upload_error(self, page):
print("视频出错了，重新上传中")
douyin_logger.info('视频出错了，重新上传中')
await page.locator('div.progress-div [class^="upload-btn-input"]').set_input_files(self.file_path)
async def upload(self, playwright: Playwright) -> None:
@@ -99,9 +100,9 @@ class DouYinVideo(object):
page = await context.new_page()
# 访问指定的 URL
await page.goto("https://creator.douyin.com/creator-micro/content/upload")
print('[+]正在上传-------{}.mp4'.format(self.title))
douyin_logger.info(f'[+]正在上传-------{self.title}.mp4')
# 等待页面跳转到指定的 URL，没进入，则自动等待到超时
print('[-] 正在打开主页...')
douyin_logger.info(f'[-] 正在打开主页...')
await page.wait_for_url("https://creator.douyin.com/creator-micro/content/upload")
# 点击 "上传视频" 按钮
await page.locator(".upload-btn--9eZLd").set_input_files(self.file_path)
@@ -114,32 +115,30 @@ class DouYinVideo(object):
"https://creator.douyin.com/creator-micro/content/publish?enter_from=publish_page")
break
except:
print(" [-] 正在等待进入视频发布页面...")
douyin_logger.info(f' [-] 正在等待进入视频发布页面...')
await asyncio.sleep(0.1)
# 填充标题和话题
# 检查是否存在包含输入框的元素
# 这里为了避免页面变化，故使用相对位置定位：作品标题父级右侧第一个元素的input子元素
await asyncio.sleep(1)
print(" [-] 正在填充标题和话题...")
douyin_logger.info(f' [-] 正在填充标题和话题...')
title_container = page.get_by_text('作品标题').locator("..").locator("xpath=following-sibling::div[1]").locator("input")
if await title_container.count():
await title_container.fill(self.title[:30])
else:
titlecontainer = page.locator(".notranslate")
await titlecontainer.click()
print("clear existing title")
await page.keyboard.press("Backspace")
await page.keyboard.press("Control+KeyA")
await page.keyboard.press("Delete")
print("filling new title")
await page.keyboard.type(self.title)
await page.keyboard.press("Enter")
css_selector = ".zone-container"
for index, tag in enumerate(self.tags, start=1):
print("正在添加第%s个话题" % index)
await page.type(css_selector, "#" + tag)
await page.press(css_selector, "Space")
douyin_logger.info(f'总共添加{len(self.tags)}个话题')
while True:
# 判断重新上传按钮是否存在，如果不存在，代表视频正在上传，则等待
@@ -147,23 +146,22 @@ class DouYinVideo(object):
# 新版：定位重新上传
number = await page.locator('div label+div:has-text("重新上传")').count()
if number > 0:
print(" [-]视频上传完毕")
douyin_logger.success(" [-]视频上传完毕")
break
else:
print(" [-] 正在上传视频中...")
douyin_logger.info(" [-] 正在上传视频中...")
await asyncio.sleep(2)
if await page.locator('div.progress-div > div:has-text("上传失败")').count():
print(" [-] 发现上传出错了...")
douyin_logger.error(" [-] 发现上传出错了... 准备重试")
await self.handle_upload_error(page)
except:
print(" [-] 正在上传视频中...")
douyin_logger.info(" [-] 正在上传视频中...")
await asyncio.sleep(2)
# 更换可见元素
await page.locator('div.semi-select span:has-text("输入地理位置")').click()
await asyncio.sleep(1)
print("clear existing location")
await page.keyboard.press("Backspace")
await page.keyboard.press("Control+KeyA")
await page.keyboard.press("Delete")
@@ -192,15 +190,15 @@ class DouYinVideo(object):
await publish_button.click()
await page.wait_for_url("https://creator.douyin.com/creator-micro/content/manage",
timeout=1500) # 如果自动跳转到作品页面，则代表发布成功
print(" [-]视频发布成功")
douyin_logger.success(" [-]视频发布成功")
break
except:
print(" [-] 视频正在发布中...")
douyin_logger.info(" [-] 视频正在发布中...")
await page.screenshot(full_page=True)
await asyncio.sleep(0.5)
await context.storage_state(path=self.account_file) # 保存cookie
print(' [-]cookie更新完毕！')
douyin_logger.success(' [-]cookie更新完毕！')
await asyncio.sleep(2) # 这里延迟是为了方便眼睛直观的观看
# 关闭浏览器上下文和浏览器实例
await context.close()