social-dl/app/api/tiktok.py

24 lines
865 B
Python
Raw Permalink Normal View History

from app.api.tiktok_scraper import Scraper as Tiktok_Scraper
from app.core.scraper_config import ScraperConfig
tiktok_scraper = Tiktok_Scraper(quiet=True)
class Tiktok(ScraperConfig):
def __init__(self, url):
super().__init__()
self.url = url
self.set_sauce(url)
async def download_or_extract(self):
media = await tiktok_scraper.hybrid_parsing(self.url)
if not media or "status" not in media or media["status"] == "failed":
return
if "video_data" in media:
self.link = media["video_data"]["nwm_video_url_HQ"]
self.thumb = media["cover_data"]["dynamic_cover"]["url_list"][0]
self.video = self.success = True
if "image_data" in media:
self.link = media["image_data"]["no_watermark_image_list"]
self.group = self.success = True