Fixed the default_url_fetcher for Python 2, which returns a message…

… type with no `get_filename` method
Kozea · Apr 22, 2014 · a8a951b · a8a951b
1 parent a084a5b
commit a8a951b
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/weasyprint/urls.py b/weasyprint/urls.py
@@ -264,7 +264,12 @@ def default_url_fetcher(url):
         url = iri_to_uri(url)
         result, mime_type, charset = urlopen_contenttype(Request(
             url, headers={'User-Agent': VERSION_STRING}))
-        filename = result.info().get_filename()
+        filename = None
+        try:
+            filename = result.info().get_filename()
+        except AttributeError:
+            # Python 2 doesn't return a message with the get_filename method
+            pass
         return dict(file_obj=result, redirected_url=result.geturl(),
                     mime_type=mime_type, encoding=charset, filename=filename)
     else: