support bilibili watchlater url

This commit is contained in:
ayanamist 2017-10-05 11:55:29 +08:00
parent c3b31c5d31
commit 52b3cb6020

View File

@ -104,6 +104,12 @@ class Bilibili(VideoExtractor):
self.parse_bili_xml(api_xml) self.parse_bili_xml(api_xml)
def prepare(self, **kwargs): def prepare(self, **kwargs):
m = re.match(r"http(?:s)?://www.bilibili.com/watchlater/#/av(\d+)(?:/p(\d+))?", self.url, re.I)
if m is not None:
self.url = 'http://www.bilibili.com/video/av{}'.format(m.group(1))
if m.group(2) is not None:
self.url += "/index_{}.html".format(m.group(2))
if socket.getdefaulttimeout() == 600: # no timeout specified if socket.getdefaulttimeout() == 600: # no timeout specified
socket.setdefaulttimeout(2) # fail fast, very speedy! socket.setdefaulttimeout(2) # fail fast, very speedy!