浏览代码

[statigram] Update to fit new website name and rename extractor

Sergey M․ 11 年之前
父节点
当前提交
867274e997
共有 2 个文件被更改,包括 6 次插入4 次删除
  1. 1 1
      youtube_dl/extractor/__init__.py
  2. 5 3
      youtube_dl/extractor/iconosquare.py

+ 1 - 1
youtube_dl/extractor/__init__.py

@@ -113,6 +113,7 @@ from .hotnewhiphop import HotNewHipHopIE
 from .howcast import HowcastIE
 from .howcast import HowcastIE
 from .huffpost import HuffPostIE
 from .huffpost import HuffPostIE
 from .hypem import HypemIE
 from .hypem import HypemIE
+from .iconosquare import IconosquareIE
 from .ign import IGNIE, OneUPIE
 from .ign import IGNIE, OneUPIE
 from .imdb import (
 from .imdb import (
     ImdbIE,
     ImdbIE,
@@ -246,7 +247,6 @@ from .spankwire import SpankwireIE
 from .spiegel import SpiegelIE
 from .spiegel import SpiegelIE
 from .spike import SpikeIE
 from .spike import SpikeIE
 from .stanfordoc import StanfordOpenClassroomIE
 from .stanfordoc import StanfordOpenClassroomIE
-from .statigram import StatigramIE
 from .steam import SteamIE
 from .steam import SteamIE
 from .streamcloud import StreamcloudIE
 from .streamcloud import StreamcloudIE
 from .streamcz import StreamCZIE
 from .streamcz import StreamCZIE

+ 5 - 3
youtube_dl/extractor/statigram.py → youtube_dl/extractor/iconosquare.py

@@ -5,8 +5,8 @@ import re
 from .common import InfoExtractor
 from .common import InfoExtractor
 
 
 
 
-class StatigramIE(InfoExtractor):
-    _VALID_URL = r'https?://(www\.)?statigr\.am/p/(?P<id>[^/]+)'
+class IconosquareIE(InfoExtractor):
+    _VALID_URL = r'https?://(www\.)?(?:iconosquare\.com|statigr\.am)/p/(?P<id>[^/]+)'
     _TEST = {
     _TEST = {
         'url': 'http://statigr.am/p/522207370455279102_24101272',
         'url': 'http://statigr.am/p/522207370455279102_24101272',
         'md5': '6eb93b882a3ded7c378ee1d6884b1814',
         'md5': '6eb93b882a3ded7c378ee1d6884b1814',
@@ -15,6 +15,7 @@ class StatigramIE(InfoExtractor):
             'ext': 'mp4',
             'ext': 'mp4',
             'uploader_id': 'aguynamedpatrick',
             'uploader_id': 'aguynamedpatrick',
             'title': 'Instagram photo by @aguynamedpatrick (Patrick Janelle)',
             'title': 'Instagram photo by @aguynamedpatrick (Patrick Janelle)',
+            'description': 'md5:644406a9ec27457ed7aa7a9ebcd4ce3d',
         },
         },
     }
     }
 
 
@@ -25,7 +26,7 @@ class StatigramIE(InfoExtractor):
         html_title = self._html_search_regex(
         html_title = self._html_search_regex(
             r'<title>(.+?)</title>',
             r'<title>(.+?)</title>',
             webpage, 'title')
             webpage, 'title')
-        title = re.sub(r'(?: *\(Videos?\))? \| Statigram$', '', html_title)
+        title = re.sub(r'(?: *\(Videos?\))? \| (?:Iconosquare|Statigram)$', '', html_title)
         uploader_id = self._html_search_regex(
         uploader_id = self._html_search_regex(
             r'@([^ ]+)', title, 'uploader name', fatal=False)
             r'@([^ ]+)', title, 'uploader name', fatal=False)
 
 
@@ -33,6 +34,7 @@ class StatigramIE(InfoExtractor):
             'id': video_id,
             'id': video_id,
             'url': self._og_search_video_url(webpage),
             'url': self._og_search_video_url(webpage),
             'title': title,
             'title': title,
+            'description': self._og_search_description(webpage),
             'thumbnail': self._og_search_thumbnail(webpage),
             'thumbnail': self._og_search_thumbnail(webpage),
             'uploader_id': uploader_id
             'uploader_id': uploader_id
         }
         }