Fixes #5 :wap support

sdvcrx · Mar 22, 2014 · 8e22c15 · 8e22c15
1 parent e115387
commit 8e22c15
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/bddown_core.py b/bddown_core.py
@@ -176,8 +176,10 @@ def download(args):
     if namespace.output_dir:
         output_dir = namespace.output_dir
 
-    links = filter(check_url, links)  # filter the wrong url
-    links = map(add_http, links)  # add 'http://'
+    # if is wap
+    links = [link.replace("wap/link", "share/link") for link in links]
+    links = filter(check_url, links)    # filter the wrong url
+    links = map(add_http, links)        # add 'http://'
     for url in links:
         pan = BaiduDown(url, secret=secret)
         filename = pan.filename