Skip to content

Commit 6881cba

Browse files
committed
Add option to TikTok image downloader for user avatars
1 parent 0abe885 commit 6881cba

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

processors/visualisation/download_tiktok.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -161,6 +161,7 @@ class TikTokImageDownloader(BasicProcessor):
161161
"options": {
162162
"thumbnail": "Video Thumbnail",
163163
"music": "Music Thumbnail",
164+
"author_avatar": "User avatar"
164165
},
165166
"default": "thumbnail"
166167
}
@@ -217,6 +218,8 @@ def process(self):
217218
url_column = "thumbnail_url"
218219
elif self.parameters.get("thumb_type") == "music":
219220
url_column = "music_thumbnail"
221+
elif self.parameters.get("thumb_type") == "author_avatar":
222+
url_column = "author_avatar"
220223
else:
221224
self.dataset.update_status("No image column selected.", is_final=True)
222225
self.dataset.finish(0)

0 commit comments

Comments
 (0)