fix #106

0967d731 · Mort Yao · 106c85e2 · 0967d731
隐藏空白更改
内联并排

Showing with 6 addition and 3 deletion

src/you_get/common.py src/you_get/common.py +6 -3

未找到文件。
--- a/src/you_get/common.py
+++ b/src/you_get/common.py
@@ -133,9 +133,12 @@ def url_info(url, faker = False):
        ext = mapping[type]
    else:
        type = None
-        filename = parse.unquote(r1(r'filename="?(.+)"?', headers['content-disposition']))
-        if len(filename.split('.')) > 1:
-            ext = filename.split('.')[-1]
+        if headers['content-disposition']:
+            filename = parse.unquote(r1(r'filename="?(.+)"?', headers['content-disposition']))
+            if len(filename.split('.')) > 1:
+                ext = filename.split('.')[-1]
+            else:
+                ext = None
        else:
            ext = None