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

1. tk chrome iframe or div

Browse Source 
2. tk click publish
This commit is contained in:
pookz@stme
2024-07-04 11:04:54 +08:00
parent e40fd39b67
commit 743f5a8dc2
3 changed files with 86 additions and 52 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tk_uploader/main.py
+40 -23
View File
@@ -19,7 +19,7 @@ async def cookie_auth(account_file):
# 创建一个新的页面 # 创建一个新的页面
page = await context.new_page() page = await context.new_page()
# 访问指定的 URL # 访问指定的 URL
await page.goto("https://www.tiktok.com/creator-center/content") await page.goto("https://www.tiktok.com/tiktokstudio/upload?lang=en")
await page.wait_for_load_state('networkidle') await page.wait_for_load_state('networkidle')
try: try:
# 选择所有的 select 元素 # 选择所有的 select 元素
@@ -62,29 +62,31 @@ async def get_tiktok_cookie(account_file):
context = await set_init_script(context) context = await set_init_script(context)
# Pause the page, and start recording manually. # Pause the page, and start recording manually.
page = await context.new_page() page = await context.new_page()
await page.goto("https://www.tiktok.com/creator-center/content") await page.goto("https://www.tiktok.com/login?lang=en")
await page.pause() await page.pause()
# 点击调试器的继续，保存cookie # 点击调试器的继续，保存cookie
await context.storage_state(path=account_file) await context.storage_state(path=account_file)
class TiktokVideo(object): class TiktokVideo(object):
def __init__(self, title, file_path, tags, publish_date: datetime, account_file): def __init__(self, title, file_path, tags, publish_date, account_file):
self.title = title self.title = title
self.file_path = file_path self.file_path = file_path
self.tags = tags self.tags = tags
self.publish_date = publish_date self.publish_date = publish_date
self.account_file = account_file self.account_file = account_file
self.locator_base = None
async def set_schedule_time(self, page, publish_date): async def set_schedule_time(self, page, publish_date):
schedule_input_element = page.frame_locator(Tk_Locator.tk_iframe).get_by_label('Schedule') schedule_input_element = self.locator_base.get_by_label('Schedule')
await schedule_input_element.wait_for(state='visible') # 确保按钮可见 await schedule_input_element.wait_for(state='visible') # 确保按钮可见
await schedule_input_element.click() await schedule_input_element.click()
scheduled_picker = page.frame_locator(Tk_Locator.tk_iframe).locator('div.scheduled-picker') scheduled_picker = self.locator_base.locator('div.scheduled-picker')
await scheduled_picker.locator('div.TUXInputBox').nth(1).click() await scheduled_picker.locator('div.TUXInputBox').nth(1).click()
calendar_month = await page.frame_locator(Tk_Locator.tk_iframe).locator('div.calendar-wrapper span.month-title').inner_text() calendar_month = await self.locator_base.locator('div.calendar-wrapper span.month-title').inner_text()
n_calendar_month = datetime.strptime(calendar_month, '%B').month n_calendar_month = datetime.strptime(calendar_month, '%B').month
@@ -92,13 +94,13 @@ class TiktokVideo(object):
if n_calendar_month != schedule_month: if n_calendar_month != schedule_month:
if n_calendar_month < schedule_month: if n_calendar_month < schedule_month:
arrow = page.frame_locator(Tk_Locator.tk_iframe).locator('div.calendar-wrapper span.arrow').nth(-1) arrow = self.locator_base.locator('div.calendar-wrapper span.arrow').nth(-1)
else: else:
arrow = page.frame_locator(Tk_Locator.tk_iframe).locator('div.calendar-wrapper span.arrow').nth(0) arrow = self.locator_base.locator('div.calendar-wrapper span.arrow').nth(0)
await arrow.click() await arrow.click()
# day set # day set
valid_days_locator = page.frame_locator(Tk_Locator.tk_iframe).locator( valid_days_locator = self.locator_base.locator(
'div.calendar-wrapper span.day.valid') 'div.calendar-wrapper span.day.valid')
valid_days = await valid_days_locator.count() valid_days = await valid_days_locator.count()
for i in range(valid_days): for i in range(valid_days):
@@ -118,20 +120,20 @@ class TiktokVideo(object):
minute_selector = f"span.tiktok-timepicker-right:has-text('{minute_str}')" minute_selector = f"span.tiktok-timepicker-right:has-text('{minute_str}')"
# pick hour first # pick hour first
await page.frame_locator(Tk_Locator.tk_iframe).locator(hour_selector).click() await self.locator_base.locator(hour_selector).click()
# click time button again # click time button again
# 等待某个特定的元素出现或状态变化，表明UI已更新 # 等待某个特定的元素出现或状态变化，表明UI已更新
await page.wait_for_timeout(1000) # 等待500毫秒 await page.wait_for_timeout(1000) # 等待500毫秒
await scheduled_picker.locator('div.TUXInputBox').nth(0).click() await scheduled_picker.locator('div.TUXInputBox').nth(0).click()
# pick minutes after # pick minutes after
await page.frame_locator(Tk_Locator.tk_iframe).locator(minute_selector).click() await self.locator_base.locator(minute_selector).click()
# click title to remove the focus. # click title to remove the focus.
await page.frame_locator(Tk_Locator.tk_iframe).locator("h1:has-text('Upload video')").click() await self.locator_base.locator("h1:has-text('Upload video')").click()
async def handle_upload_error(self, page): async def handle_upload_error(self, page):
tiktok_logger.info("video upload error retrying.") tiktok_logger.info("video upload error retrying.")
select_file_button = page.frame_locator(Tk_Locator.tk_iframe).locator('button[aria-label="Select file"]') select_file_button = self.locator_base.locator('button[aria-label="Select file"]')
async with page.expect_file_chooser() as fc_info: async with page.expect_file_chooser() as fc_info:
await select_file_button.click() await select_file_button.click()
file_chooser = await fc_info.value file_chooser = await fc_info.value
@@ -147,8 +149,16 @@ class TiktokVideo(object):
tiktok_logger.info(f'[+]Uploading-------{self.title}.mp4') tiktok_logger.info(f'[+]Uploading-------{self.title}.mp4')
await page.wait_for_url("https://www.tiktok.com/tiktokstudio/upload", timeout=10000) await page.wait_for_url("https://www.tiktok.com/tiktokstudio/upload", timeout=10000)
await page.wait_for_selector('iframe[data-tt="Upload_index_iframe"]')
upload_button = page.frame_locator(Tk_Locator.tk_iframe).locator( try:
await page.wait_for_selector('iframe[data-tt="Upload_index_iframe"], div.upload-container', timeout=10000)
tiktok_logger.info("Either iframe or div appeared.")
except Exception as e:
tiktok_logger.error("Neither iframe nor div appeared within the timeout.")
await self.choose_base_locator(page)
upload_button = self.locator_base.locator(
'button:has-text("Select video"):visible') 'button:has-text("Select video"):visible')
await upload_button.wait_for(state='visible') # 确保按钮可见 await upload_button.wait_for(state='visible') # 确保按钮可见
@@ -159,7 +169,7 @@ class TiktokVideo(object):
await self.add_title_tags(page) await self.add_title_tags(page)
# detact upload status # detact upload status
await self.detact_upload_status(page) await self.detect_upload_status(page)
if self.publish_date != 0: if self.publish_date != 0:
await self.set_schedule_time(page, self.publish_date) await self.set_schedule_time(page, self.publish_date)
@@ -174,7 +184,7 @@ class TiktokVideo(object):
async def add_title_tags(self, page): async def add_title_tags(self, page):
editor_locator = page.frame_locator(Tk_Locator.tk_iframe).locator('div.public-DraftEditor-content') editor_locator = self.locator_base.locator('div.public-DraftEditor-content')
await editor_locator.click() await editor_locator.click()
await page.keyboard.press("End") await page.keyboard.press("End")
@@ -209,15 +219,15 @@ class TiktokVideo(object):
success_flag_div = '#\\:r9\\:' 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 = self.locator_base.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(success_flag_div).wait_for(state="visible", timeout=1500) await self.locator_base.locator(success_flag_div).wait_for(state="visible", timeout=3000)
tiktok_logger.success(" [-] video published success") tiktok_logger.success(" [-] video published success")
break break
except Exception as e: except Exception as e:
if await page.frame_locator(Tk_Locator.tk_iframe).locator(success_flag_div).count(): if await self.locator_base.locator(success_flag_div).count():
tiktok_logger.success(" [-]video published success") tiktok_logger.success(" [-]video published success")
break break
else: else:
@@ -226,22 +236,29 @@ class TiktokVideo(object):
await page.screenshot(full_page=True) await page.screenshot(full_page=True)
await asyncio.sleep(0.5) await asyncio.sleep(0.5)
async def detact_upload_status(self, page): async def detect_upload_status(self, page):
while True: while True:
try: try:
if await page.frame_locator(Tk_Locator.tk_iframe).locator('div.btn-post > button').get_attribute("disabled") is None: if await self.locator_base.locator('div.btn-post > button').get_attribute("disabled") is None:
tiktok_logger.info(" [-]video uploaded.") tiktok_logger.info(" [-]video uploaded.")
break break
else: else:
tiktok_logger.info(" [-] video uploading...") tiktok_logger.info(" [-] video uploading...")
await asyncio.sleep(2) await asyncio.sleep(2)
if await page.frame_locator(Tk_Locator.tk_iframe).locator('button[aria-label="Select file"]').count(): if await self.locator_base.locator('button[aria-label="Select file"]').count():
tiktok_logger.info(" [-] found some error while uploading now retry...") tiktok_logger.info(" [-] found some error while uploading now retry...")
await self.handle_upload_error(page) await self.handle_upload_error(page)
except: except:
tiktok_logger.info(" [-] video uploading...") tiktok_logger.info(" [-] video uploading...")
await asyncio.sleep(2) await asyncio.sleep(2)
async def choose_base_locator(self, page):
# await page.wait_for_selector('div.upload-container')
if await page.locator('iframe[data-tt="Upload_index_iframe"]').count():
self.locator_base = self.locator_base
else:
self.locator_base = page.locator(Tk_Locator.default)
async def main(self): async def main(self):
async with async_playwright() as playwright: async with async_playwright() as playwright:
await self.upload(playwright) await self.upload(playwright)
tk_uploader/main_chrome.py
+45 -29
View File
@@ -71,7 +71,7 @@ async def get_tiktok_cookie(account_file):
class TiktokVideo(object): class TiktokVideo(object):
def __init__(self, title, file_path, tags, publish_date: datetime, account_file, thumbnail_path=None): def __init__(self, title, file_path, tags, publish_date, account_file, thumbnail_path=None):
self.title = title self.title = title
self.file_path = file_path self.file_path = file_path
self.tags = tags self.tags = tags
@@ -79,19 +79,20 @@ class TiktokVideo(object):
self.thumbnail_path = thumbnail_path self.thumbnail_path = thumbnail_path
self.account_file = account_file self.account_file = account_file
self.local_executable_path = LOCAL_CHROME_PATH self.local_executable_path = LOCAL_CHROME_PATH
self.locator_base = None
async def set_schedule_time(self, page, publish_date): async def set_schedule_time(self, page, publish_date):
schedule_input_element = page.frame_locator(Tk_Locator.tk_iframe).get_by_label('Schedule') schedule_input_element = self.locator_base.get_by_label('Schedule')
await schedule_input_element.wait_for(state='visible') # 确保按钮可见 await schedule_input_element.wait_for(state='visible') # 确保按钮可见
await schedule_input_element.click() await schedule_input_element.click()
if await page.frame_locator(Tk_Locator.tk_iframe).locator('div.TUXButton-content >> text=Allow').count(): if await self.locator_base.locator('div.TUXButton-content >> text=Allow').count():
await page.frame_locator(Tk_Locator.tk_iframe).locator('div.TUXButton-content >> text=Allow').click() await self.locator_base.locator('div.TUXButton-content >> text=Allow').click()
scheduled_picker = page.frame_locator(Tk_Locator.tk_iframe).locator('div.scheduled-picker') scheduled_picker = self.locator_base.locator('div.scheduled-picker')
await scheduled_picker.locator('div.TUXInputBox').nth(1).click() await scheduled_picker.locator('div.TUXInputBox').nth(1).click()
calendar_month = await page.frame_locator(Tk_Locator.tk_iframe).locator( calendar_month = await self.locator_base.locator(
'div.calendar-wrapper span.month-title').inner_text() 'div.calendar-wrapper span.month-title').inner_text()
n_calendar_month = datetime.strptime(calendar_month, '%B').month n_calendar_month = datetime.strptime(calendar_month, '%B').month
@@ -100,13 +101,13 @@ class TiktokVideo(object):
if n_calendar_month != schedule_month: if n_calendar_month != schedule_month:
if n_calendar_month < schedule_month: if n_calendar_month < schedule_month:
arrow = page.frame_locator(Tk_Locator.tk_iframe).locator('div.calendar-wrapper span.arrow').nth(-1) arrow = self.locator_base.locator('div.calendar-wrapper span.arrow').nth(-1)
else: else:
arrow = page.frame_locator(Tk_Locator.tk_iframe).locator('div.calendar-wrapper span.arrow').nth(0) arrow = self.locator_base.locator('div.calendar-wrapper span.arrow').nth(0)
await arrow.click() await arrow.click()
# day set # day set
valid_days_locator = page.frame_locator(Tk_Locator.tk_iframe).locator( valid_days_locator = self.locator_base.locator(
'div.calendar-wrapper span.day.valid') 'div.calendar-wrapper span.day.valid')
valid_days = await valid_days_locator.count() valid_days = await valid_days_locator.count()
for i in range(valid_days): for i in range(valid_days):
@@ -127,20 +128,20 @@ class TiktokVideo(object):
# pick hour first # pick hour first
await page.wait_for_timeout(500) # 等待500毫秒 await page.wait_for_timeout(500) # 等待500毫秒
await page.frame_locator(Tk_Locator.tk_iframe).locator(hour_selector).click() await self.locator_base.locator(hour_selector).click()
# click time button again # click time button again
await page.wait_for_timeout(500) # 等待500毫秒 await page.wait_for_timeout(500) # 等待500毫秒
await scheduled_picker.locator('div.TUXInputBox').nth(0).click() await scheduled_picker.locator('div.TUXInputBox').nth(0).click()
await page.wait_for_timeout(500) # 等待500毫秒 await page.wait_for_timeout(500) # 等待500毫秒
# pick minutes after # pick minutes after
await page.frame_locator(Tk_Locator.tk_iframe).locator(minute_selector).click() await self.locator_base.locator(minute_selector).click()
# click title to remove the focus. # click title to remove the focus.
# await page.frame_locator(Tk_Locator.tk_iframe).locator("h1:has-text('Upload video')").click() # await self.locator_base.locator("h1:has-text('Upload video')").click()
async def handle_upload_error(self, page): async def handle_upload_error(self, page):
tiktok_logger.info("video upload error retrying.") tiktok_logger.info("video upload error retrying.")
select_file_button = page.frame_locator(Tk_Locator.tk_iframe).locator('button[aria-label="Select file"]') select_file_button = self.locator_base.locator('button[aria-label="Select file"]')
async with page.expect_file_chooser() as fc_info: async with page.expect_file_chooser() as fc_info:
await select_file_button.click() await select_file_button.click()
file_chooser = await fc_info.value file_chooser = await fc_info.value
@@ -158,8 +159,16 @@ class TiktokVideo(object):
tiktok_logger.info(f'[+]Uploading-------{self.title}.mp4') tiktok_logger.info(f'[+]Uploading-------{self.title}.mp4')
await page.wait_for_url("https://www.tiktok.com/tiktokstudio/upload", timeout=10000) await page.wait_for_url("https://www.tiktok.com/tiktokstudio/upload", timeout=10000)
await page.wait_for_selector('iframe[data-tt="Upload_index_iframe"]')
upload_button = page.frame_locator(Tk_Locator.tk_iframe).locator( try:
await page.wait_for_selector('iframe[data-tt="Upload_index_iframe"], div.upload-container', timeout=10000)
tiktok_logger.info("Either iframe or div appeared.")
except Exception as e:
tiktok_logger.error("Neither iframe nor div appeared within the timeout.")
await self.choose_base_locator(page)
upload_button = self.locator_base.locator(
'button:has-text("Select video"):visible') 'button:has-text("Select video"):visible')
await upload_button.wait_for(state='visible') # 确保按钮可见 await upload_button.wait_for(state='visible') # 确保按钮可见
@@ -169,8 +178,8 @@ class TiktokVideo(object):
await file_chooser.set_files(self.file_path) await file_chooser.set_files(self.file_path)
await self.add_title_tags(page) await self.add_title_tags(page)
# detact upload status # detect upload status
await self.detact_upload_status(page) await self.detect_upload_status(page)
if self.thumbnail_path: if self.thumbnail_path:
tiktok_logger.info(f'[+] Uploading thumbnail file {self.title}.png') tiktok_logger.info(f'[+] Uploading thumbnail file {self.title}.png')
await self.upload_thumbnails(page) await self.upload_thumbnails(page)
@@ -189,7 +198,7 @@ class TiktokVideo(object):
async def add_title_tags(self, page): async def add_title_tags(self, page):
editor_locator = page.frame_locator(Tk_Locator.tk_iframe).locator('div.public-DraftEditor-content') editor_locator = self.locator_base.locator('div.public-DraftEditor-content')
await editor_locator.click() await editor_locator.click()
await page.keyboard.press("End") await page.keyboard.press("End")
@@ -221,13 +230,13 @@ class TiktokVideo(object):
await page.keyboard.press("End") await page.keyboard.press("End")
async def upload_thumbnails(self, page): async def upload_thumbnails(self, page):
await page.frame_locator(Tk_Locator.tk_iframe).locator(".cover-selector-image-container").click() await self.locator_base.locator(".cover-selector-image-container").click()
await page.frame_locator(Tk_Locator.tk_iframe).locator(".cover-edit-container >> text=Upload cover").click() await self.locator_base.locator(".cover-edit-container >> text=Upload cover").click()
async with page.expect_file_chooser() as fc_info: async with page.expect_file_chooser() as fc_info:
await page.frame_locator(Tk_Locator.tk_iframe).locator(".upload-image-upload-area").click() await self.locator_base.locator(".upload-image-upload-area").click()
file_chooser = await fc_info.value file_chooser = await fc_info.value
await file_chooser.set_files(self.thumbnail_path) await file_chooser.set_files(self.thumbnail_path)
await page.frame_locator(Tk_Locator.tk_iframe).locator('div.cover-edit-panel:not(.hide-panel)').get_by_role( await self.locator_base.locator('div.cover-edit-panel:not(.hide-panel)').get_by_role(
"button", name="Confirm").click() "button", name="Confirm").click()
await page.wait_for_timeout(3000) # wait 3s, fix it later await page.wait_for_timeout(3000) # wait 3s, fix it later
@@ -245,15 +254,15 @@ class TiktokVideo(object):
success_flag_div = 'div.common-modal-confirm-modal' success_flag_div = 'div.common-modal-confirm-modal'
while True: while True:
try: try:
publish_button = page.frame_locator(Tk_Locator.tk_iframe).locator('div.button-group button').nth(0) publish_button = self.locator_base.locator('div.button-group button').nth(0)
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(success_flag_div).wait_for(state="visible", timeout=1500) await self.locator_base.locator(success_flag_div).wait_for(state="visible", timeout=3000)
tiktok_logger.success(" [-] video published success") tiktok_logger.success(" [-] video published success")
break break
except Exception as e: except Exception as e:
if await page.frame_locator(Tk_Locator.tk_iframe).locator(success_flag_div).count(): if await self.locator_base.locator(success_flag_div).count():
tiktok_logger.success(" [-]video published success") tiktok_logger.success(" [-]video published success")
break break
else: else:
@@ -262,18 +271,18 @@ class TiktokVideo(object):
await page.screenshot(full_page=True) await page.screenshot(full_page=True)
await asyncio.sleep(0.5) await asyncio.sleep(0.5)
async def detact_upload_status(self, page): async def detect_upload_status(self, page):
while True: while True:
try: try:
# if await page.frame_locator(Tk_Locator.tk_iframe).locator('div.btn-post > button').get_attribute("disabled") is None: # if await self.locator_base.locator('div.btn-post > button').get_attribute("disabled") is None:
if await page.frame_locator(Tk_Locator.tk_iframe).locator( if await self.locator_base.locator(
'div.button-group > button >> text=Post').get_attribute("disabled") is None: 'div.button-group > button >> text=Post').get_attribute("disabled") is None:
tiktok_logger.info(" [-]video uploaded.") tiktok_logger.info(" [-]video uploaded.")
break break
else: else:
tiktok_logger.info(" [-] video uploading...") tiktok_logger.info(" [-] video uploading...")
await asyncio.sleep(2) await asyncio.sleep(2)
if await page.frame_locator(Tk_Locator.tk_iframe).locator( if await self.locator_base.locator(
'button[aria-label="Select file"]').count(): 'button[aria-label="Select file"]').count():
tiktok_logger.info(" [-] found some error while uploading now retry...") tiktok_logger.info(" [-] found some error while uploading now retry...")
await self.handle_upload_error(page) await self.handle_upload_error(page)
@@ -281,6 +290,13 @@ class TiktokVideo(object):
tiktok_logger.info(" [-] video uploading...") tiktok_logger.info(" [-] video uploading...")
await asyncio.sleep(2) await asyncio.sleep(2)
async def choose_base_locator(self, page):
# await page.wait_for_selector('div.upload-container')
if await page.locator('iframe[data-tt="Upload_index_iframe"]').count():
self.locator_base = page.frame_locator(Tk_Locator.tk_iframe)
else:
self.locator_base = page.locator(Tk_Locator.default)
async def main(self): async def main(self):
async with async_playwright() as playwright: async with async_playwright() as playwright:
await self.upload(playwright) await self.upload(playwright)
tk_uploader/tk_config.py
+1
View File
@@ -1,3 +1,4 @@
class Tk_Locator(object): class Tk_Locator(object):
tk_iframe = '[data-tt="Upload_index_iframe"]' tk_iframe = '[data-tt="Upload_index_iframe"]'
default = 'body'