Paillat-dev/viralfactory
1
0
Fork 0
mirror of https://github.com/Paillat-dev/viralfactory.git synced 2026-01-02 01:06:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

🏷️ 🎨 Add title and description parameter to TikTokUploadEngine.upload. Also run linter

Browse Source
This commit is contained in:
Paillat
2024-04-21 21:59:17 +02:00
parent e9a5328d1d
commit f3db42e341
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/engines/UploadEngine/TikTokUploadEngine.py
View File

@@ -14,20 +14,18 @@ class TikTokUploadEngine(BaseUploadEngine):
self.hashtags = options[0] self.hashtags = options[0]
super().__init__() super().__init__()
def upload(self): def upload(self, title: str, description: str, path: str):
cookies = self.get_setting(type="cookies")["cookies"] cookies = self.get_setting(type="cookies")["cookies"]
if cookies is None: if cookies is None:
gr.Warning( gr.Warning(
"Skipping upload to TikTok because no cookies were provided. Please provide cookies in the settings." "Skipping upload to TikTok because no cookies were provided. Please provide cookies in the settings."
) )
return return
title: str = self.ctx.title
description: str = self.ctx.description
hashtags = self.hashtags.strip().split(" ") hashtags = self.hashtags.strip().split(" ")
title = title.strip() title = title.strip()
description = description.strip() description = description.strip()
#we get all the hastags from title and description and add them to the list of hashtags, while removing them from the title and description # we get all the hastags from title and description and add them to the list of hashtags, while removing them from the title and description
for word in title.split(): for word in title.split():
if word.startswith("#"): if word.startswith("#"):
hashtags.append(word) hashtags.append(word)
@@ -39,11 +37,13 @@ class TikTokUploadEngine(BaseUploadEngine):
hashtags_str = " ".join(hashtags) + " " if hashtags else "" hashtags_str = " ".join(hashtags) + " " if hashtags else ""
failed = upload_video( failed = upload_video(
filename=self.ctx.get_file_path("final.mp4"), filename=path,
description=f"{title} {hashtags_str}\n{description}", description=f"{title} {hashtags_str}\n{description}",
cookies_str=cookies, cookies_str=cookies,
browser="chrome", browser="chrome",
comment=True, stitch=False, duet=False comment=True,
stitch=False,
duet=False,
) )
for _ in failed: for _ in failed:
gr.Error(f"Failed to upload to TikTok") gr.Error(f"Failed to upload to TikTok")