_ Git - youtube-dl/blob - youtube_dl/extractor/sunporno.py

   1 from __future__ import unicode_literals
   2
   3 import re
   4
   5 from .common import InfoExtractor
   6 from ..utils import (
   7     parse_duration,
   8     int_or_none,
   9     qualities,
  10     determine_ext,
  11 )
  12
  13
  14 class SunPornoIE(InfoExtractor):
  15     _VALID_URL = r'https?://(?:www\.)?sunporno\.com/videos/(?P<id>\d+)'
  16     _TEST = {
  17         'url': 'http://www.sunporno.com/videos/807778/',
  18         'md5': '6457d3c165fd6de062b99ef6c2ff4c86',
  19         'info_dict': {
  20             'id': '807778',
  21             'ext': 'flv',
  22             'title': 'md5:0a400058e8105d39e35c35e7c5184164',
  23             'description': 'md5:a31241990e1bd3a64e72ae99afb325fb',
  24             'thumbnail': 're:^https?://.*\.jpg$',
  25             'duration': 302,
  26             'age_limit': 18,
  27         }
  28     }
  29
  30     def _real_extract(self, url):
  31         mobj = re.match(self._VALID_URL, url)
  32         video_id = mobj.group('id')
  33
  34         webpage = self._download_webpage(url, video_id)
  35
  36         title = self._html_search_regex(r'<title>([^<]+)</title>', webpage, 'title')
  37         description = self._html_search_meta('description', webpage, 'description')
  38         thumbnail = self._html_search_regex(
  39             r'poster="([^"]+)"', webpage, 'thumbnail', fatal=False)
  40
  41         duration = parse_duration(self._search_regex(
  42             r'<span>Duration: (\d+:\d+)</span>', webpage, 'duration', fatal=False))
  43
  44         view_count = int_or_none(self._html_search_regex(
  45             r'<span class="views">(\d+)</span>', webpage, 'view count', fatal=False))
  46         comment_count = int_or_none(self._html_search_regex(
  47             r'(\d+)</b> Comments?', webpage, 'comment count', fatal=False))
  48
  49         formats = []
  50         quality = qualities(['mp4', 'flv'])
  51         for video_url in re.findall(r'<source src="([^"]+)"', webpage):
  52             video_ext = determine_ext(video_url)
  53             formats.append({
  54                 'url': video_url,
  55                 'format_id': video_ext,
  56                 'quality': quality(video_ext),
  57             })
  58         self._sort_formats(formats)
  59
  60         return {
  61             'id': video_id,
  62             'title': title,
  63             'description': description,
  64             'thumbnail': thumbnail,
  65             'duration': duration,
  66             'view_count': view_count,
  67             'comment_count': comment_count,
  68             'formats': formats,
  69             'age_limit': 18,
  70         }