| 1 | from pathlib import Path |
| 2 | from typing import List |
| 3 | |
| 4 | from conf import BASE_DIR |
| 5 | |
| 6 | SOCIAL_MEDIA_DOUYIN = "douyin" |
| 7 | SOCIAL_MEDIA_TENCENT = "tencent" |
| 8 | SOCIAL_MEDIA_TIKTOK = "tiktok" |
| 9 | SOCIAL_MEDIA_BILIBILI = "bilibili" |
| 10 | SOCIAL_MEDIA_KUAISHOU = "kuaishou" |
| 11 | |
| 12 | |
| 13 | def get_supported_social_media() -> List[str]: |
| 14 | return [SOCIAL_MEDIA_DOUYIN, SOCIAL_MEDIA_TENCENT, SOCIAL_MEDIA_TIKTOK, SOCIAL_MEDIA_KUAISHOU] |
| 15 | |
| 16 | |
| 17 | def get_cli_action() -> List[str]: |
| 18 | return ["upload", "login", "watch"] |
| 19 | |
| 20 | |
| 21 | async def set_init_script(context): |
| 22 | stealth_js_path = Path(BASE_DIR / "utils/stealth.min.js") |
| 23 | await context.add_init_script(path=stealth_js_path) |
| 24 | return context |
| 25 |