Add option to TikTok image downloader for user avatars

digitalmethodsinitiative · Dec 2, 2024 · 6881cba · 6881cba
1 parent 0abe885
commit 6881cba
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/processors/visualisation/download_tiktok.py b/processors/visualisation/download_tiktok.py
@@ -161,6 +161,7 @@ class TikTokImageDownloader(BasicProcessor):
             "options": {
                 "thumbnail": "Video Thumbnail",
                 "music": "Music Thumbnail",
+                "author_avatar": "User avatar"
             },
             "default": "thumbnail"
         }
@@ -217,6 +218,8 @@ def process(self):
             url_column = "thumbnail_url"
         elif self.parameters.get("thumb_type") == "music":
             url_column = "music_thumbnail"
+        elif self.parameters.get("thumb_type") == "author_avatar":
+            url_column = "author_avatar"
         else:
             self.dataset.update_status("No image column selected.", is_final=True)
             self.dataset.finish(0)