[common] Add new helper function _match_id

This commit is contained in:
Philipp Hagemeister 2014-09-28 09:31:58 +02:00
parent f4b1c7adb8
commit ed9266db90
2 changed files with 9 additions and 2 deletions

View File

@ -22,8 +22,7 @@ class ABCIE(InfoExtractor):
} }
def _real_extract(self, url): def _real_extract(self, url):
mobj = re.match(self._VALID_URL, url) video_id = self._match_id(url)
video_id = mobj.group('id')
webpage = self._download_webpage(url, video_id) webpage = self._download_webpage(url, video_id)
urls_info_json = self._search_regex( urls_info_json = self._search_regex(

View File

@ -165,6 +165,14 @@ class InfoExtractor(object):
cls._VALID_URL_RE = re.compile(cls._VALID_URL) cls._VALID_URL_RE = re.compile(cls._VALID_URL)
return cls._VALID_URL_RE.match(url) is not None return cls._VALID_URL_RE.match(url) is not None
@classmethod
def _match_id(cls, url):
if '_VALID_URL_RE' not in cls.__dict__:
cls._VALID_URL_RE = re.compile(cls._VALID_URL)
m = cls._VALID_URL_RE.match(url)
assert m
return m.group('id')
@classmethod @classmethod
def working(cls): def working(cls):
"""Getter method for _WORKING.""" """Getter method for _WORKING."""