diff --git a/README.MD b/README.MD index ad19869..0ed1965 100644 --- a/README.MD +++ b/README.MD @@ -67,7 +67,7 @@ meta_file 内容(content)： ``` ### Usage -1. 设置conf 文件中的 `LOCAL_CHROME_PATH`(在douyin 或者视频号可能出现chromium 不兼容的各种问题，建议设置本地的chrome) +1. 设置conf 文件中的 `LOCAL_CHROME_PATH`(在douyin、视频号 tiktok可能出现chromium 不兼容的各种问题，建议设置本地的chrome) 2. 这里分割出来3条路 - 可自行研究源码，免费、任意 穿插在自己的项目中 - 可参考下面的各个平台的使用指南，`examples`文件夹中有各种示例代码 @@ -141,7 +141,7 @@ generate_schedule_time_next_day 默认从第二天开始（此举为避免选择 > 解决方案： > 1. 下载chrome > 2. 找到chrome的安装目录 -> 3. 将目录配置到DouYinVideo类中 self.local_executable_path = "xxx/xxx/chrome.exe" +> 3. 将本地chrome 路径conf文件中 LOCAL_CHROME_PATH = "xxx/xxx/chrome.exe" 参考项目： @@ -168,7 +168,7 @@ generate_schedule_time_next_day 默认从第二天开始（此举为避免选择 > 解决方案： > 1. 下载chrome > 2. 找到chrome的安装目录 -> 3. 将目录配置到TencentVideo类中 self.local_executable_path = "xxx/xxx/chrome.exe" +> 3. 将本地chrome 路径conf文件中 LOCAL_CHROME_PATH = "xxx/xxx/chrome.exe" --- @@ -234,11 +234,16 @@ bilibili cookie 长期有效（至少我运行2年以来是这样的） ### 5. tiktok 使用playwright模拟浏览器行为(Simulating Browser Behavior with playwright) 1. 准备视频目录结构(Prepare the video directory structure) -2. cookie获取(generate your cookie)：get_tk_cookie.py +2. 将本地chrome路径配置到conf文件中 LOCAL_CHROME_PATH = "xxx/xxx/chrome.exe"(Configure the local chrome path to the conf file) +3. cookie获取(generate your cookie)：get_tk_cookie.py ![get tiktok cookie](media/tk_login.png) -3. 上传视频(upload video)：upload_video_to_tiktok.py - +4. 上传视频(upload video)：upload_video_to_tiktok.py +24.7.3 update local chrome support: +- you can upload your custom thumbnail(place `png` file at the path of `videos`) +- if not, the system will take the first frame of the video as the thumbnail. +- chrome drive can't login by gmail oauth. the google don't support it. (if you want login by google oauth you can use the `tk_uploader/main.py` old firefox.) +- before the upload process change the language to eng. 其他部分解释： ``` diff --git a/cli_main.py b/cli_main.py index c2f61cb..e8da41e 100644 --- a/cli_main.py +++ b/cli_main.py @@ -7,7 +7,7 @@ from pathlib import Path from conf import BASE_DIR from douyin_uploader.main import douyin_setup, DouYinVideo from tencent_uploader.main import weixin_setup, TencentVideo -from tk_uploader.main import tiktok_setup, TiktokVideo +from tk_uploader.main_chrome import tiktok_setup, TiktokVideo from utils.base_social_media import get_supported_social_media, get_cli_action, SOCIAL_MEDIA_DOUYIN, \ SOCIAL_MEDIA_TENCENT, SOCIAL_MEDIA_TIKTOK from utils.constant import TencentZoneTypes diff --git a/examples/get_tk_cookie.py b/examples/get_tk_cookie.py index 9c02dbc..3ce008d 100644 --- a/examples/get_tk_cookie.py +++ b/examples/get_tk_cookie.py @@ -2,7 +2,7 @@ import asyncio from pathlib import Path from conf import BASE_DIR -from tk_uploader.main import tiktok_setup +from tk_uploader.main_chrome import tiktok_setup if __name__ == '__main__': account_file = Path(BASE_DIR / "tk_uploader" / "account.json") diff --git a/examples/upload_video_to_tiktok.py b/examples/upload_video_to_tiktok.py index c3b657c..740c927 100644 --- a/examples/upload_video_to_tiktok.py +++ b/examples/upload_video_to_tiktok.py @@ -2,7 +2,8 @@ import asyncio from pathlib import Path from conf import BASE_DIR -from tk_uploader.main import tiktok_setup, TiktokVideo +# from tk_uploader.main import tiktok_setup, TiktokVideo +from tk_uploader.main_chrome import tiktok_setup, TiktokVideo from utils.files_times import generate_schedule_time_next_day, get_title_and_hashtags @@ -17,8 +18,13 @@ if __name__ == '__main__': cookie_setup = asyncio.run(tiktok_setup(account_file, handle=True)) for index, file in enumerate(files): title, tags = get_title_and_hashtags(str(file)) + thumbnail_path = file.with_suffix('.png') print(f"video_file_name：{file}") print(f"video_title：{title}") print(f"video_hashtag：{tags}") - app = TiktokVideo(title, file, tags, publish_datetimes[index], account_file) + if thumbnail_path.exists(): + print(f"thumbnail_file_name：{thumbnail_path}") + app = TiktokVideo(title, file, tags, publish_datetimes[index], account_file, thumbnail_path) + else: + app = TiktokVideo(title, file, tags, publish_datetimes[index], account_file) asyncio.run(app.main(), debug=False) diff --git a/tk_uploader/main_chrome.py b/tk_uploader/main_chrome.py new file mode 100644 index 0000000..b296987 --- /dev/null +++ b/tk_uploader/main_chrome.py @@ -0,0 +1,286 @@ +# -*- coding: utf-8 -*- +import re +from datetime import datetime + +from playwright.async_api import Playwright, async_playwright +import os +import asyncio + +from conf import LOCAL_CHROME_PATH +from tk_uploader.tk_config import Tk_Locator +from utils.base_social_media import set_init_script +from utils.files_times import get_absolute_path +from utils.log import tiktok_logger + + +async def cookie_auth(account_file): + async with async_playwright() as playwright: + browser = await playwright.chromium.launch(headless=True) + context = await browser.new_context(storage_state=account_file) + context = await set_init_script(context) + # 创建一个新的页面 + page = await context.new_page() + # 访问指定的 URL + await page.goto("https://www.tiktok.com/tiktokstudio/upload?lang=en") + await page.wait_for_load_state('networkidle') + try: + # 选择所有的 select 元素 + select_elements = await page.query_selector_all('select') + for element in select_elements: + class_name = await element.get_attribute('class') + # 使用正则表达式匹配特定模式的 class 名称 + if re.match(r'tiktok-.*-SelectFormContainer.*', class_name): + tiktok_logger.error("[+] cookie expired") + return False + tiktok_logger.success("[+] cookie valid") + return True + except: + tiktok_logger.success("[+] cookie valid") + return True + + +async def tiktok_setup(account_file, handle=False): + account_file = get_absolute_path(account_file, "tk_uploader") + if not os.path.exists(account_file) or not await cookie_auth(account_file): + if not handle: + return False + tiktok_logger.info('[+] cookie file is not existed or expired. Now open the browser auto. Please login with your way(gmail phone, whatever, the cookie file will generated after login') + await get_tiktok_cookie(account_file) + return True + + +async def get_tiktok_cookie(account_file): + async with async_playwright() as playwright: + options = { + 'args': [ + '--lang en-GB', + ], + 'headless': False, # Set headless option here + } + # Make sure to run headed. + browser = await playwright.chromium.launch(**options) + # Setup context however you like. + context = await browser.new_context() # Pass any options + context = await set_init_script(context) + # Pause the page, and start recording manually. + page = await context.new_page() + await page.goto("https://www.tiktok.com/login?lang=en") + await page.pause() + # 点击调试器的继续，保存cookie + await context.storage_state(path=account_file) + + +class TiktokVideo(object): + def __init__(self, title, file_path, tags, publish_date: datetime, account_file, thumbnail_path=None): + self.title = title + self.file_path = file_path + self.tags = tags + self.publish_date = publish_date + self.thumbnail_path = thumbnail_path + self.account_file = account_file + self.local_executable_path = LOCAL_CHROME_PATH + + async def set_schedule_time(self, page, publish_date): + schedule_input_element = page.frame_locator(Tk_Locator.tk_iframe).get_by_label('Schedule') + await schedule_input_element.wait_for(state='visible') # 确保按钮可见 + + await schedule_input_element.click() + if await page.frame_locator(Tk_Locator.tk_iframe).locator('div.TUXButton-content >> text=Allow').count(): + await page.frame_locator(Tk_Locator.tk_iframe).locator('div.TUXButton-content >> text=Allow').click() + + scheduled_picker = page.frame_locator(Tk_Locator.tk_iframe).locator('div.scheduled-picker') + 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() + + n_calendar_month = datetime.strptime(calendar_month, '%B').month + + schedule_month = publish_date.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) + else: + arrow = page.frame_locator(Tk_Locator.tk_iframe).locator('div.calendar-wrapper span.arrow').nth(0) + await arrow.click() + + # day set + valid_days_locator = page.frame_locator(Tk_Locator.tk_iframe).locator( + 'div.calendar-wrapper span.day.valid') + valid_days = await valid_days_locator.count() + for i in range(valid_days): + day_element = valid_days_locator.nth(i) + text = await day_element.inner_text() + if text.strip() == str(publish_date.day): + await day_element.click() + break + # time set + await scheduled_picker.locator('div.TUXInputBox').nth(0).click() + + hour_str = publish_date.strftime("%H") + correct_minute = int(publish_date.minute / 5) + minute_str = f"{correct_minute:02d}" + + hour_selector = f"span.tiktok-timepicker-left:has-text('{hour_str}')" + minute_selector = f"span.tiktok-timepicker-right:has-text('{minute_str}')" + + # pick hour first + await page.wait_for_timeout(500) # 等待500毫秒 + await page.frame_locator(Tk_Locator.tk_iframe).locator(hour_selector).click() + # click time button again + await page.wait_for_timeout(500) # 等待500毫秒 + await scheduled_picker.locator('div.TUXInputBox').nth(0).click() + await page.wait_for_timeout(500) # 等待500毫秒 + # pick minutes after + await page.frame_locator(Tk_Locator.tk_iframe).locator(minute_selector).click() + + # click title to remove the focus. + # await page.frame_locator(Tk_Locator.tk_iframe).locator("h1:has-text('Upload video')").click() + + async def handle_upload_error(self, page): + tiktok_logger.info("video upload error retrying.") + select_file_button = page.frame_locator(Tk_Locator.tk_iframe).locator('button[aria-label="Select file"]') + async with page.expect_file_chooser() as fc_info: + await select_file_button.click() + file_chooser = await fc_info.value + await file_chooser.set_files(self.file_path) + + async def upload(self, playwright: Playwright) -> None: + 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 set_init_script(context) + page = await context.new_page() + + # change language to eng first + await self.change_language(page) + await page.goto("https://www.tiktok.com/tiktokstudio/upload") + 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_selector('iframe[data-tt="Upload_index_iframe"]') + upload_button = page.frame_locator(Tk_Locator.tk_iframe).locator( + 'button:has-text("Select video"):visible') + await upload_button.wait_for(state='visible') # 确保按钮可见 + + async with page.expect_file_chooser() as fc_info: + await upload_button.click() + file_chooser = await fc_info.value + await file_chooser.set_files(self.file_path) + + await self.add_title_tags(page) + # detact upload status + await self.detact_upload_status(page) + if self.thumbnail_path: + tiktok_logger.info(f'[+] Uploading thumbnail file {self.title}.png') + await self.upload_thumbnails(page) + + if self.publish_date != 0: + await self.set_schedule_time(page, self.publish_date) + + await self.click_publish(page) + + await context.storage_state(path=f"{self.account_file}") # save cookie + tiktok_logger.info(' [-] update cookie！') + await asyncio.sleep(2) # close delay for look the video status + # close all + await context.close() + await browser.close() + + async def add_title_tags(self, page): + + editor_locator = page.frame_locator(Tk_Locator.tk_iframe).locator('div.public-DraftEditor-content') + await editor_locator.click() + + await page.keyboard.press("End") + + await page.keyboard.press("Control+A") + + await page.keyboard.press("Delete") + + await page.keyboard.press("End") + + await page.wait_for_timeout(1000) # 等待1秒 + + await page.keyboard.insert_text(self.title) + await page.wait_for_timeout(1000) # 等待1秒 + await page.keyboard.press("End") + + await page.keyboard.press("Enter") + + # tag part + for index, tag in enumerate(self.tags, start=1): + tiktok_logger.info("Setting the %s tag" % index) + await page.keyboard.press("End") + await page.wait_for_timeout(1000) # 等待1秒 + await page.keyboard.insert_text("#" + tag + " ") + await page.keyboard.press("Space") + await page.wait_for_timeout(1000) # 等待1秒 + + await page.keyboard.press("Backspace") + await page.keyboard.press("End") + + async def upload_thumbnails(self, page): + await page.frame_locator(Tk_Locator.tk_iframe).locator(".cover-selector-image-container").click() + await page.frame_locator(Tk_Locator.tk_iframe).locator(".cover-edit-container >> text=Upload cover").click() + async with page.expect_file_chooser() as fc_info: + await page.frame_locator(Tk_Locator.tk_iframe).locator(".upload-image-upload-area").click() + file_chooser = await fc_info.value + 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( + "button", name="Confirm").click() + await page.wait_for_timeout(3000) # wait 3s, fix it later + + async def change_language(self, page): + # set the language to english + await page.goto("https://www.tiktok.com") + await page.wait_for_url("https://www.tiktok.com/", timeout=100000) + await page.wait_for_selector('#header-more-menu-icon') + + await page.locator('#header-more-menu-icon').hover() + await page.locator('[data-e2e="language-select"]').click() + await page.locator('#lang-setting-popup-list >> text=English').click() + + async def click_publish(self, page): + success_flag_div = 'div.common-modal-confirm-modal' + while True: + try: + publish_button = page.frame_locator(Tk_Locator.tk_iframe).locator('div.button-group button').nth(0) + if await publish_button.count(): + await publish_button.click() + + await page.frame_locator(Tk_Locator.tk_iframe).locator(success_flag_div).wait_for(state="visible", timeout=1500) + tiktok_logger.success(" [-] video published success") + break + except Exception as e: + if await page.frame_locator(Tk_Locator.tk_iframe).locator(success_flag_div).count(): + tiktok_logger.success(" [-]video published success") + break + else: + tiktok_logger.exception(f" [-] Exception: {e}") + tiktok_logger.info(" [-] video publishing") + await page.screenshot(full_page=True) + await asyncio.sleep(0.5) + + async def detact_upload_status(self, page): + while True: + try: + # if await page.frame_locator(Tk_Locator.tk_iframe).locator('div.btn-post > button').get_attribute("disabled") is None: + if await page.frame_locator(Tk_Locator.tk_iframe).locator( + 'div.button-group > button >> text=Post').get_attribute("disabled") is None: + tiktok_logger.info(" [-]video uploaded.") + break + else: + tiktok_logger.info(" [-] video uploading...") + await asyncio.sleep(2) + if await page.frame_locator(Tk_Locator.tk_iframe).locator( + 'button[aria-label="Select file"]').count(): + tiktok_logger.info(" [-] found some error while uploading now retry...") + await self.handle_upload_error(page) + except: + tiktok_logger.info(" [-] video uploading...") + await asyncio.sleep(2) + + async def main(self): + async with async_playwright() as playwright: + await self.upload(playwright) diff --git a/videos/demo.png b/videos/demo.png new file mode 100644 index 0000000..9a749c5 Binary files /dev/null and b/videos/demo.png differ