Added Support For Image File Types

PyFileDownloader.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: PyFileDownloader
-Version: 1.1.0
+Version: 1.2.0
 Summary: A Python module to download any file-type from the internet.
 Home-page: https://github.com/NotCookey/PyFileDownloader
 Author: NotCookey

PyFileDownloader/PyFileDownloader.py

@@ -21,24 +21,42 @@ def headers(self) -> dict:
 
 		filename = response.headers.get("Content-Disposition")
 		filesize = response.headers.get("Content-Length")
-		filetype = response.headers.get("Content-Type").split(";")[0]
+		filetype = response.headers.get("Content-Type")
 
 		if filename:
-			parsed_headers = werkzeug.http.parse_options_header(
-				re.findall("filename=(.+)", filename)[0]
-			)
+			parsed_headers = werkzeug.http.parse_options_header(filename)
 
 			for keys in parsed_headers:
 				if "filename" in keys:
 					filename = keys["filename"]
 			return {"filename": filename, "filesize": filesize, "filetype": filetype}
 		else:
-			if "image" in filetype:
-				return {
-					"filename": "download.jpg",
-					"filesize": filesize,
-					"filetype": filetype,
-				}
+			if filetype:
+				if "image" in filetype:
+					if "jpeg" in filetype:
+						return {
+							"filename": "download.jpg",
+							"filesize": filesize,
+							"filetype": filetype,
+						}
+					if "gif" in filetype:
+						return {
+							"filename": "download.gif",
+							"filesize": filesize,
+							"filetype": filetype,
+						}
+					if "png" in filetype:
+						return {
+							"filename": "download.png",
+							"filesize": filesize,
+							"filetype": filetype,
+						}
+					if "webp" in filetype:
+						return {
+							"filename": "download.webp",
+							"filesize": filesize,
+							"filetype": filetype,
+						}
 			else:
 				return {
 					"filename": "download.html",

build/lib/PyFileDownloader/PyFileDownloader.py

@@ -21,24 +21,42 @@ def headers(self) -> dict:
 
 		filename = response.headers.get("Content-Disposition")
 		filesize = response.headers.get("Content-Length")
-		filetype = response.headers.get("Content-Type").split(";")[0]
+		filetype = response.headers.get("Content-Type")
 
 		if filename:
-			parsed_headers = werkzeug.http.parse_options_header(
-				re.findall("filename=(.+)", filename)[0]
-			)
+			parsed_headers = werkzeug.http.parse_options_header(filename)
 
 			for keys in parsed_headers:
 				if "filename" in keys:
 					filename = keys["filename"]
 			return {"filename": filename, "filesize": filesize, "filetype": filetype}
 		else:
-			if "image" in filetype:
-				return {
-					"filename": "download.jpg",
-					"filesize": filesize,
-					"filetype": filetype,
-				}
+			if filetype:
+				if "image" in filetype:
+					if "jpeg" in filetype:
+						return {
+							"filename": "download.jpg",
+							"filesize": filesize,
+							"filetype": filetype,
+						}
+					if "gif" in filetype:
+						return {
+							"filename": "download.gif",
+							"filesize": filesize,
+							"filetype": filetype,
+						}
+					if "png" in filetype:
+						return {
+							"filename": "download.png",
+							"filesize": filesize,
+							"filetype": filetype,
+						}
+					if "webp" in filetype:
+						return {
+							"filename": "download.webp",
+							"filesize": filesize,
+							"filetype": filetype,
+						}
 			else:
 				return {
 					"filename": "download.html",

setup.py

@@ -7,7 +7,7 @@
 with codecs.open(os.path.join(here, "README.md"), encoding="utf-8") as fh:
     long_description = "\n" + fh.read()
 
-VERSION = '1.1.0'
+VERSION = '1.2.0'
 DESCRIPTION = 'A Python module to download any file-type from the internet.'
 LONG_DESCRIPTION = 'A Python module to download any file-type from the internet.'