1
0
mirror of https://github.com/l1ving/youtube-dl synced 2025-03-13 02:12:16 +08:00
2018-04-28 03:11:57 -04:00

99 lines
3.9 KiB
Python

# coding: utf-8
from __future__ import unicode_literals
from collections import OrderedDict
from datetime import datetime, timedelta
import re
import subprocess as sp
import time
from .common import InfoExtractor
from ..utils import compat_str, int_or_none
class PewTubeIE(InfoExtractor):
_VALID_URL = r'https?://(?:www\.)?pewtube\.com/user/[^/]+/(?P<id>[^/]+)'
_TEST = {
'url': 'https://pew.tube/user/MrBond/4jLJf06',
'md5': 'TODO: md5 sum of the first 10241 bytes of the video file (use --test)',
'info_dict': {
'id': '4jLJf06',
'ext': 'mp4',
'title': 'Mr. Bond - Good Old Nationalist',
'thumbnail': r're:^https?://.*\.jpg$',
# TODO more properties, either as:
# * A value
# * MD5 checksum; start the string with md5:
# * A regular expression; start the string with re:
# * Any Python type (for example int or float)
}
}
def _real_extract(self, url):
video_id = self._match_id(url)
webpage = self._download_webpage(url, video_id, tries=5, timeout=15)
title = self._html_search_regex(r'<h2 .*upload\-title\-value[^>]+>([^<]+)</h2>', webpage, 'title')
thumbnail = self._html_search_regex(r'<video .*poster="(https[^"]+)"', webpage, 'thumbnail', fatal=False)
if not thumbnail:
thumbnail = self._og_search_thumbnail(webpage)
video_url = self._html_search_regex(r'<(?:audio|video)[^>]+>.*<source.*src="([^"]+)".*</(?:audio|video)>', webpage, 'video URL')
if not video_url:
video_url = self._og_search_video_url(webpage)
uploader = self._html_search_regex(r'<h3.*class="uploader-name"[^>]+>.*<a href="/user[^>]+>([^<]+)', webpage, 'uploader')
description = self._html_search_meta('description', webpage, 'description')
view_count = self._html_search_regex(r'<h3(?:[^>]+)?>(\d+)(?:\s+)[vV]iews</h3>', webpage, 'view count')
like_count = 0
for like_class in ('like', 'laugh', 'love',):
like_count += int(self._html_search_regex(r'<p class="{}">(\d+)</p>'.format(like_class), webpage, '{} count'))
dislike_count = 0
for dislike_class in ('dislike', 'sad', 'discount',):
dislike_count += int(self._html_search_regex(r'<p class="{}">(\d+)</p>'.format(like_class), webpage, '{} count'))
like_count = None if not like_count else like_count
dislike_count = None if not dislike_count else dislike_count
if not thumbnail:
thumbnail = self._og_search_thumbnail(webpage)
if not description:
description = self._og_search_description(webpage)
upload_date_s = ou = self._html_search_regex(r'<h4>Uploaded ([^<]+)</h4>', webpage, 'upload date')
today = datetime.today()
upload_date_s = re.sub(r'(?:\s+)ago\s+\&nbsp;?$', '', upload_date_s)
n, unit = re.search('^(?P<n>(?:\d+|a))\s(?:\s+)?(?P<unit>.*)', upload_date_s).groups()
if n == 'a':
n = 1
else:
n = int(n)
unit = unit.lower().encode('utf-8')
total_seconds = 0
if not unit.endswith('s'):
unit += 's'
if unit == 'months':
unit = 'weeks'
n *= 4
elif unit == 'years':
unit = 'weeks'
n *= 52
kwargs = dict()
kwargs[unit] = n
td = timedelta(**kwargs)
upload_date = (today - td).strftime('%Y%m%d')
return {
'id': video_id,
'title': title,
'description': description,
'uploader': uploader,
'creator': uploader,
'uploader_url': '/'.join(url.split('/')[:-1]),
'thumbnail': thumbnail,
'view_count': int_or_none(view_count),
'like_count': like_count,
'dislike_count': dislike_count,
'upload_date': upload_date,
'url': video_url,
}