mirror of
https://github.com/soimort/you-get.git
synced 2025-02-11 04:32:27 +03:00
48 lines
1.3 KiB
Python
48 lines
1.3 KiB
Python
# coding=utf-8
|
|
|
|
import re
|
|
import json
|
|
from urllib.parse import unquote
|
|
|
|
from ..common import (
|
|
url_size,
|
|
print_info,
|
|
get_content,
|
|
fake_headers,
|
|
download_urls,
|
|
playlist_not_supported,
|
|
)
|
|
|
|
|
|
__all__ = ['douyin_download_by_url']
|
|
|
|
|
|
def douyin_download_by_url(url, **kwargs):
|
|
page_content = get_content(url, headers=fake_headers)
|
|
# The video player and video source are rendered client-side, the data
|
|
# contains in a <script id="RENDER_DATA" type="application/json"> tag
|
|
# quoted, unquote the whole page content then search using regex with
|
|
# regular string.
|
|
page_content = unquote(page_content)
|
|
title = re.findall(r'"desc":"([^"]*)"', page_content)[0].strip()
|
|
video_format = 'mp4'
|
|
# video URLs are in this pattern {"src":"THE_URL"}, in json format
|
|
urls_pattern = r'"playAddr":(\[.*?\])'
|
|
urls = json.loads(re.findall(urls_pattern, page_content)[0])
|
|
video_url = 'https:' + urls[0]['src']
|
|
size = url_size(video_url, faker=True)
|
|
print_info(
|
|
site_info='douyin.com', title=title,
|
|
type=video_format, size=size
|
|
)
|
|
if not kwargs['info_only']:
|
|
download_urls(
|
|
urls=[video_url], title=title, ext=video_format, total_size=size,
|
|
faker=True,
|
|
**kwargs
|
|
)
|
|
|
|
|
|
download = douyin_download_by_url
|
|
download_playlist = playlist_not_supported('douyin')
|