you-get/src/you_get/extractors/tiktok.py

24 lines
816 B
Python
Raw Normal View History

2018-11-30 20:29:22 +03:00
#!/usr/bin/env python
__all__ = ['tiktok_download']
from ..common import *
def tiktok_download(url, output_dir='.', merge=True, info_only=False, **kwargs):
html = get_html(url)
title = r1(r'<title>(.*?)</title>', html)
2018-11-30 20:54:05 +03:00
video_id = r1(r'/video/(\d+)', url) or r1(r'musical\?id=(\d+)', html)
title = '%s [%s]' % (title, video_id)
dataText = r1(r'var data = \[(.*)\] ', html) or r1(r'var data = (\{.*\})', html)
2018-11-30 20:29:22 +03:00
data = json.loads(dataText)
source = 'http:' + data['video']['play_addr']['url_list'][0]
mime, ext, size = url_info(source)
print_info(site_info, title, mime, size)
if not info_only:
download_urls([source], title, ext, size, output_dir, merge=merge)
site_info = "TikTok.com"
download = tiktok_download
download_playlist = playlist_not_supported('tiktok')