Merge branch 'ku6/fix-extension-error' of https://github.com/Xiaochen…

…Cui/you-get into XiaochenCui-ku6/fix-extension-error
ckwsens · May 12, 2017 · 13daffd · 13daffd
2 parents ded1d22 + 65712f8
commit 13daffd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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 = match1(urls[0], r'.*\.(\w+)\??[^\.]*')
     assert ext in ('flv', 'mp4', 'f4v'), ext
     ext = {'f4v': 'flv'}.get(ext, ext)
     size = 0