github/gallery-dl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix error if no file extension is found

Browse Source
This commit is contained in:
Mike Fährmann
2017-04-26 12:31:42 +02:00
parent 701c016b97
commit 48a5b11204
1 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
gallery_dl/downloader/http.py
View File

@@ -12,9 +12,12 @@ import time
import requests import requests
import requests.exceptions as rexcepts import requests.exceptions as rexcepts
import mimetypes import mimetypes
import logging
from .common import BasicDownloader from .common import BasicDownloader
from .. import config from .. import config
log = logging.getLogger("http")
class Downloader(BasicDownloader): class Downloader(BasicDownloader):
@@ -63,8 +66,13 @@ class Downloader(BasicDownloader):
# set 'extension' keyword from Content-Type header # set 'extension' keyword from Content-Type header
mtype = response.headers.get("Content-Type", "image/jpeg") mtype = response.headers.get("Content-Type", "image/jpeg")
exts = mimetypes.guess_all_extensions(mtype, strict=False) exts = mimetypes.guess_all_extensions(mtype, strict=False)
exts.sort() if exts:
pathfmt.set_extension(exts[-1][1:]) 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(): if pathfmt.exists():
self.out.skip(pathfmt.path) self.out.skip(pathfmt.path)
response.close() response.close()