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

1. tk流程最后返回最新上传视频的video_id

Browse Source
This commit is contained in:
home-dev-pookz
2025-06-07 12:14:42 +08:00
parent bb9e8094dd
commit 95217cc9ed
1 changed files with 12 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
uploader/tk_uploader/main_chrome.py
+12 -2
View File
@@ -148,7 +148,7 @@ class TiktokVideo(object):
async def upload(self, playwright: Playwright) -> None: async def upload(self, playwright: Playwright) -> None:
browser = await playwright.chromium.launch(headless=False, executable_path=self.local_executable_path) browser = await playwright.chromium.launch(headless=False, executable_path=self.local_executable_path)
context = await browser.new_context(storage_state=f"{self.account_file}") context = await browser.new_context(storage_state=f"{self.account_file}")
context = await set_init_script(context) # context = await set_init_script(context)
page = await context.new_page() page = await context.new_page()
# change language to eng first # change language to eng first
@@ -186,6 +186,7 @@ class TiktokVideo(object):
await self.set_schedule_time(page, self.publish_date) await self.set_schedule_time(page, self.publish_date)
await self.click_publish(page) await self.click_publish(page)
tiktok_logger.success(f"video_id: {await self.get_last_video_id(page)}")
await context.storage_state(path=f"{self.account_file}") # save cookie await context.storage_state(path=f"{self.account_file}") # save cookie
tiktok_logger.info(' [-] update cookie！') tiktok_logger.info(' [-] update cookie！')
@@ -249,7 +250,7 @@ class TiktokVideo(object):
await page.locator('[data-e2e="nav-more-menu"]').click() await page.locator('[data-e2e="nav-more-menu"]').click()
await page.locator('[data-e2e="language-select"]').click() await page.locator('[data-e2e="language-select"]').click()
await page.locator('#creator-tools-selection-menu-header >> text=English').click() await page.locator('#creator-tools-selection-menu-header >> text=English (US)').click()
async def click_publish(self, page): async def click_publish(self, page):
success_flag_div = 'div.common-modal-confirm-modal' success_flag_div = 'div.common-modal-confirm-modal'
@@ -267,6 +268,15 @@ class TiktokVideo(object):
tiktok_logger.info(" [-] video publishing") tiktok_logger.info(" [-] video publishing")
await asyncio.sleep(0.5) await asyncio.sleep(0.5)
async def get_last_video_id(self, page):
await page.wait_for_selector('div[data-tt="components_PostTable_Container"]')
video_list_locator = self.locator_base.locator('div[data-tt="components_PostTable_Container"] div[data-tt="components_PostInfoCell_Container"] a')
if await video_list_locator.count():
first_video_obj = await video_list_locator.nth(0).get_attribute('href')
video_id = re.search(r'video/(\d+)', first_video_obj).group(1) if first_video_obj else None
return video_id
async def detect_upload_status(self, page): async def detect_upload_status(self, page):
while True: while True:
try: try: