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

fix tk change

Browse Source
This commit is contained in:
pookz@stme
2024-05-20 10:45:16 +08:00
parent ab388b5b2d
commit 72a54ebf33
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tk_uploader/main.py
+6 -5
View File
@@ -78,7 +78,7 @@ class TiktokVideo(object):
await page.frame_locator(Tk_Locator.tk_iframe).locator('div.scheduled-container input').click() await page.frame_locator(Tk_Locator.tk_iframe).locator('div.scheduled-container input').click()
scheduled_picker = page.frame_locator(Tk_Locator.tk_iframe).locator('div.scheduled-picker') scheduled_picker = page.frame_locator(Tk_Locator.tk_iframe).locator('div.scheduled-picker')
await scheduled_picker.locator('div.date-picker-input').click() await scheduled_picker.locator('div.TUXInputBox').nth(0).click()
calendar_month = await page.frame_locator(Tk_Locator.tk_iframe).locator('div.calendar-wrapper span.month-title').inner_text() calendar_month = await page.frame_locator(Tk_Locator.tk_iframe).locator('div.calendar-wrapper span.month-title').inner_text()
@@ -104,7 +104,7 @@ class TiktokVideo(object):
await day_element.click() await day_element.click()
break break
# time set # time set
await page.frame_locator(Tk_Locator.tk_iframe).locator("div.time-picker-input").click() await page.frame_locator(Tk_Locator.tk_iframe).locator("div.time-picker-container").click()
hour_str = publish_date.strftime("%H") hour_str = publish_date.strftime("%H")
correct_minute = int(publish_date.minute / 5) correct_minute = int(publish_date.minute / 5)
@@ -141,7 +141,7 @@ class TiktokVideo(object):
await page.wait_for_url("https://www.tiktok.com/creator-center/upload") await page.wait_for_url("https://www.tiktok.com/creator-center/upload")
await page.wait_for_selector('iframe[data-tt="Upload_index_iframe"]') await page.wait_for_selector('iframe[data-tt="Upload_index_iframe"]')
upload_button = page.frame_locator(Tk_Locator.tk_iframe).locator( upload_button = page.frame_locator(Tk_Locator.tk_iframe).locator(
'button:has-text("Select file"):visible') 'button:has-text("Select video"):visible')
async with page.expect_file_chooser() as fc_info: async with page.expect_file_chooser() as fc_info:
await upload_button.click() await upload_button.click()
@@ -188,17 +188,18 @@ class TiktokVideo(object):
print(f"success add hashtag: {len(self.tags)}") print(f"success add hashtag: {len(self.tags)}")
async def click_publish(self, page): async def click_publish(self, page):
success_flag_div = '#\\:r9\\:'
while True: while True:
try: try:
publish_button = page.frame_locator(Tk_Locator.tk_iframe).locator('div.btn-post') publish_button = page.frame_locator(Tk_Locator.tk_iframe).locator('div.btn-post')
if await publish_button.count(): if await publish_button.count():
await publish_button.click() await publish_button.click()
await page.frame_locator(Tk_Locator.tk_iframe).locator("div.uploaded-modal:visible").wait_for(state="visible", timeout=1500) await page.frame_locator(Tk_Locator.tk_iframe).locator(success_flag_div).wait_for(state="visible", timeout=1500)
print(" [-] video published success") print(" [-] video published success")
break break
except Exception as e: except Exception as e:
if await page.frame_locator(Tk_Locator.tk_iframe).locator("div.uploaded-modal:visible").count(): if await page.frame_locator(Tk_Locator.tk_iframe).locator(success_flag_div).count():
print(" [-]video published success") print(" [-]video published success")
break break
else: else: