From 155e434607fc322f5f2431f4f33ea57698b117d6 Mon Sep 17 00:00:00 2001
From: XiaochenCui <jcnlcxc@163.com>
Date: Sun, 7 May 2017 06:38:03 +0800
Subject: [PATCH] fix the error when parse ku6 url (such as
 http://v.ku6.com/show/dBKZbnO51PWi7N2xRa3Bmw...html), case the sub-url end
 with mp4?AccessKeyId=...&Expires=...&Signature=...

---
 src/you_get/extractors/ku6.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/you_get/extractors/ku6.py b/src/you_get/extractors/ku6.py
index 808619f4..1c6361d2 100644
--- a/src/you_get/extractors/ku6.py
+++ b/src/you_get/extractors/ku6.py
@@ -14,7 +14,7 @@ def ku6_download_by_id(id, title = None, output_dir = '.', merge = True, info_on
     title = title or t
     assert title
     urls = f.split(',')
-    ext = re.sub(r'.*\.', '', urls[0])
+    ext = re.match(r'.*\.(\w+)\??[^\.]*', urls[0]).group(1)
     assert ext in ('flv', 'mp4', 'f4v'), ext
     ext = {'f4v': 'flv'}.get(ext, ext)
     size = 0