TikTok: Support more URL formats

vt.tiktok.com www.tiktok.com/t/
mikf · Dec 24, 2024 · 8aec090 · 8aec090
1 parent d2eecb9
commit 8aec090
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 1 deletion.
diff --git a/gallery_dl/extractor/tiktok.py b/gallery_dl/extractor/tiktok.py
@@ -14,7 +14,8 @@
 USER_PATTERN = BASE_PATTERN + r"/+@([\w.]{0,23}\w)?"
 POST_PATTERN = USER_PATTERN + \
     r"/+(?:[pP][hH][oO][tT][oO]|[vV][iI][dD][eE][oO])/+(?:[0-9]+)/*"
-VM_POST_PATTERN = r"(?:https?://)?(?:vm\.)?tiktok\.com/+.*/*"
+VM_POST_PATTERN = r"(?:(?:https?://)?(?:(?:vm|vt)\.)?tiktok\.com/+.*/*)|" + \
+    r"(?:(?:https?://)?(?:www\.)?tiktok\.com/+t/+.*/*)"
 
 
 class TiktokExtractor(Extractor):

diff --git a/test/results/tiktok.py b/test/results/tiktok.py
@@ -164,4 +164,20 @@
     "#pattern"  : PATTERN,
     "#options"  : {"videos": False, "tiktok-range": "1-10"}
 },
+{
+    "#url"      : "https://vt.tiktok.com/ZGdhVtER2",
+    "#comment"  : "vt.tiktok.com link: single photo",
+    "#category" : ("", "tiktok", "vmpost"),
+    "#class"    : tiktok.TiktokVmpostExtractor,
+    "#pattern"  : PATTERN,
+    "#options"  : {"videos": False}
+},
+{
+    "#url"      : "https://www.tiktok.com/t/ZGdhVtER2//",
+    "#comment"  : "www.tiktok.com/t/ link: single photo",
+    "#category" : ("", "tiktok", "vmpost"),
+    "#class"    : tiktok.TiktokVmpostExtractor,
+    "#pattern"  : PATTERN,
+    "#options"  : {"videos": False}
+},
 )