[downloader:http] add MIME type and signature for .ico files

(closes #1211)
mikf · Jan 1, 2021 · 0594821 · 0594821
1 parent b0beed7
commit 0594821
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/gallery_dl/downloader/http.py b/gallery_dl/downloader/http.py
@@ -283,7 +283,10 @@ def _adjust_extension(pathfmt, file_header):
     "image/x-ms-bmp": "bmp",
     "image/webp"    : "webp",
     "image/svg+xml" : "svg",
-
+    "image/ico"     : "ico",
+    "image/icon"    : "ico",
+    "image/x-icon"  : "ico",
+    "image/vnd.microsoft.icon" : "ico",
     "image/x-photoshop"        : "psd",
     "application/x-photoshop"  : "psd",
     "image/vnd.adobe.photoshop": "psd",
@@ -314,14 +317,16 @@ def _adjust_extension(pathfmt, file_header):
     "application/octet-stream": "bin",
 }
 
-# taken from https://en.wikipedia.org/wiki/List_of_file_signatures
+# https://en.wikipedia.org/wiki/List_of_file_signatures
 FILE_SIGNATURES = {
     "jpg" : b"\xFF\xD8\xFF",
     "png" : b"\x89PNG\r\n\x1A\n",
     "gif" : (b"GIF87a", b"GIF89a"),
     "bmp" : b"BM",
     "webp": b"RIFF",
     "svg" : b"<?xml",
+    "ico" : b"\x00\x00\x01\x00",
+    "cur" : b"\x00\x00\x02\x00",
     "psd" : b"8BPS",
     "webm": b"\x1A\x45\xDF\xA3",
     "ogg" : b"OggS",
@@ -333,8 +338,7 @@ def _adjust_extension(pathfmt, file_header):
     "pdf" : b"%PDF-",
     "swf" : (b"CWS", b"FWS"),
     # check 'bin' files against all other file signatures
-    "bin" : b"\x00\x00\x00\x00",
+    "bin" : b"\x00\x00\x00\x00\x00\x00\x00\x00",
 }
 
-
 __downloader__ = HttpDownloader