fix error if no file extension is found

This commit is contained in:
Mike Fährmann
2017-04-26 12:31:42 +02:00
parent 701c016b97
commit 48a5b11204

View File

@@ -12,9 +12,12 @@ import time
import requests
import requests.exceptions as rexcepts
import mimetypes
import logging
from .common import BasicDownloader
from .. import config
log = logging.getLogger("http")
class Downloader(BasicDownloader):
@@ -63,8 +66,13 @@ class Downloader(BasicDownloader):
# set 'extension' keyword from Content-Type header
mtype = response.headers.get("Content-Type", "image/jpeg")
exts = mimetypes.guess_all_extensions(mtype, strict=False)
exts.sort()
pathfmt.set_extension(exts[-1][1:])
if exts:
exts.sort()
pathfmt.set_extension(exts[-1][1:])
else:
log.warning("No file extension found for MIME type '%s'",
mtype)
pathfmt.set_extension("txt")
if pathfmt.exists():
self.out.skip(pathfmt.path)
response.close()