From 44f2700fd085f5268a1e980103221203572c982d Mon Sep 17 00:00:00 2001
From: "pookz@stme" <pookz@stme.com>
Date: Mon, 12 Aug 2024 17:17:01 +0800
Subject: [PATCH] 1. add kuaishou cli

---
 README.MD                  |  2 +-
 cli_main.py                | 11 ++++++++++-
 utils/base_social_media.py |  3 ++-
 3 files changed, 13 insertions(+), 3 deletions(-)

diff --git a/README.MD b/README.MD
index 35d561a..2d85685 100644
--- a/README.MD
+++ b/README.MD
@@ -87,7 +87,7 @@ usage: cli_main.py [-h] platform account_name action ...
 Upload video to multiple social-media.
 
 positional arguments:
-  platform      Choose social-media platform: douyin tencent tiktok
+  platform      Choose social-media platform: douyin tencent tiktok kuaishou
   account_name  Account name for the platform: xiaoA
   action        Choose action
     upload      upload operation
diff --git a/cli_main.py b/cli_main.py
index 6792593..747f8ec 100644
--- a/cli_main.py
+++ b/cli_main.py
@@ -6,10 +6,11 @@ from pathlib import Path
 
 from conf import BASE_DIR
 from uploader.douyin_uploader.main import douyin_setup, DouYinVideo
+from uploader.ks_uploader.main import ks_setup, KSVideo
 from uploader.tencent_uploader.main import weixin_setup, TencentVideo
 from uploader.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
+    SOCIAL_MEDIA_TENCENT, SOCIAL_MEDIA_TIKTOK, SOCIAL_MEDIA_KUAISHOU
 from utils.constant import TencentZoneTypes
 from utils.files_times import get_title_and_hashtags
 
@@ -63,6 +64,8 @@ async def main():
             await tiktok_setup(str(account_file), handle=True)
         elif args.platform == SOCIAL_MEDIA_TENCENT:
             await weixin_setup(str(account_file), handle=True)
+        elif args.platform == SOCIAL_MEDIA_KUAISHOU:
+            await ks_setup(str(account_file), handle=True)
     elif args.action == 'upload':
         title, tags = get_title_and_hashtags(args.video_file)
         video_file = args.video_file
@@ -84,6 +87,12 @@ async def main():
             await weixin_setup(account_file, handle=True)
             category = TencentZoneTypes.LIFESTYLE.value  # 标记原创需要否则不需要传
             app = TencentVideo(title, video_file, tags, publish_date, account_file, category)
+        elif args.platform == SOCIAL_MEDIA_KUAISHOU:
+            await ks_setup(account_file, handle=True)
+            app = KSVideo(title, video_file, tags, publish_date, account_file)
+        else:
+            print("Wrong platform, please check your input")
+            exit()
 
         await app.main()
 
diff --git a/utils/base_social_media.py b/utils/base_social_media.py
index ddfb14e..7697875 100644
--- a/utils/base_social_media.py
+++ b/utils/base_social_media.py
@@ -7,10 +7,11 @@ SOCIAL_MEDIA_DOUYIN = "douyin"
 SOCIAL_MEDIA_TENCENT = "tencent"
 SOCIAL_MEDIA_TIKTOK = "tiktok"
 SOCIAL_MEDIA_BILIBILI = "bilibili"
+SOCIAL_MEDIA_KUAISHOU = "kuaishou"
 
 
 def get_supported_social_media() -> List[str]:
-    return [SOCIAL_MEDIA_DOUYIN, SOCIAL_MEDIA_TENCENT, SOCIAL_MEDIA_TIKTOK]
+    return [SOCIAL_MEDIA_DOUYIN, SOCIAL_MEDIA_TENCENT, SOCIAL_MEDIA_TIKTOK, SOCIAL_MEDIA_KUAISHOU]
 
 
 def get_cli_action() -> List[str]: