youtube-dl/youtube_dl/extractor/tiktok.py

48 lines
1.8 KiB
Python

# coding: utf-8
from .common import InfoExtractor
from ..utils import (
int_or_none,
)
class TikTokIE(InfoExtractor):
_VALID_URL = r'https?://(?:www\.)?tiktok\.com/.*/video/(?P<id>[^/?#&]+)'
_TESTS = [{
'url': 'https://www.tiktok.com/@ben_brainard/video/6858321293117312262',
# 'md5': 'd584b572e92fcd48888051f238022420',
'info_dict': {
'id': '6858321293117312262',
'ext': 'mp4',
'title': 'ben_brainard',
'description': 'Hurricane... hurricane... hurricane... I got nothing.',
'uploader': 'ben_brainard',
'timestamp': 1596827366,
'upload_date': '20200807',
'comment_count': int
}
}]
def _real_extract(self, url):
video_id = self._match_id(url)
webpage = self._download_webpage(url, video_id)
# print webpage
description = self._search_regex(r'<title[^>]*>([^<]+)</title>', webpage, 'description')
mediaUrl = self._search_regex(r'"video":{"urls":\["([^"]+)"', webpage, 'url')
# data = self._parse_html5_media_entries(url,webpage,video_id)
uploader = self._search_regex(r'"authorName":"([^"]+)"', webpage, 'uploader')
commentCount = self._search_regex(r'"commentCount":([^,]+)', webpage, 'comment_count')
viewCount = self._search_regex(r'"playCount":([^,]+)', webpage, 'view_count')
timestamp = self._search_regex(r'"createTime":"([^"]+)"', webpage, 'timestamp')
return {
'id': video_id,
'url': mediaUrl,
'title': uploader,
'description': description,
'ext': 'mp4',
'uploader': uploader,
'comment_count': int_or_none(commentCount),
'view_count': int_or_none(viewCount),
'timestamp': int_or_none(timestamp)
}