From 52b3cb6020d6b735879fccde04d1feeac5b0f47b Mon Sep 17 00:00:00 2001
From: ayanamist <ayanamist@gmail.com>
Date: Thu, 5 Oct 2017 11:55:29 +0800
Subject: [PATCH] support bilibili watchlater url

---
 src/you_get/extractors/bilibili.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/you_get/extractors/bilibili.py b/src/you_get/extractors/bilibili.py
index f75397d6..036ddfb6 100644
--- a/src/you_get/extractors/bilibili.py
+++ b/src/you_get/extractors/bilibili.py
@@ -104,6 +104,12 @@ class Bilibili(VideoExtractor):
                 self.parse_bili_xml(api_xml)
 
     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
             socket.setdefaulttimeout(2) # fail fast, very speedy!