Merge pull request #6 from Jemeni11/imgur-support

Released Version 4.1.0
Jemeni11 · Apr 17, 2024 · b783544 · b783544
2 parents 75268e9 + a855d56
commit b783544
Show file tree

Hide file tree

Showing 5 changed files with 17 additions and 5 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -13,6 +13,11 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 [//]: # (- **Fixed** for any bug fixes.)
 [//]: # (- **Security** in case of vulnerabilities.)
 
+## [4.1.0] - 2024-04-17
+
+### Added
+- Added a user agent string to help download imgur images.
+
 ## [4.0.0] - 2024-02-28
 
 ### Fixed
@@ -87,6 +92,7 @@ match the new README format. The content type was updated from `text/markdown` t
 - Released FicImageScript
 
 
+[4.1.0]: https://github.com/Jemeni11/FicImage/compare/v4.0.0...v4.1.0
 [4.0.0]: https://github.com/Jemeni11/FicImage/compare/v3.0.0...v4.0.0
 [3.0.0]: https://github.com/Jemeni11/FicImage/compare/v2.1.0...v3.0.0
 [2.1.0]: https://github.com/Jemeni11/FicImage/compare/v2.0.0...v2.1.0

diff --git a/FicImage/image.py b/FicImage/image.py
@@ -49,9 +49,15 @@ def get_image_from_url(
 				imgdata = PIL_Image_to_bytes(compressed_base64_image, file_ext)
 
 			return imgdata, file_ext, f"image/{file_ext}"
-
+		elif url.startswith("https://imgur.com/"):
+			url = "https://i.imgur.com/" + url.split("https://imgur.com/")[-1]
+
 		with requests.Session() as session:
-			img = session.get(url, stream=True)
+			headers = {
+				'user-agent': "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) "
+				              "Chrome/122.0.0.0 Safari/537.36"
+			}
+			img = session.get(url, stream=True, headers=headers)
 
 			if img.headers.get("content-type", "") == "image/svg+xml":
 				return img.content, "svg", "image/svg+xml"

diff --git a/FicImage/main.py b/FicImage/main.py
@@ -7,7 +7,7 @@
 from .image import get_image_from_url
 from .utils import config_check, load_config_json, default_ficimage_settings
 
-__version__ = "4.0.0"
+__version__ = "4.1.0"
 
 
 def update_epub(path_to_epub, config_file_path, debug):

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "FicImageScript"
-version = "4.0.0"
+version = "4.1.0"
 authors = [
   { name="Emmanuel C. Jemeni", email="[email protected]" }
 ]

diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setuptools.setup(
 	name="FicImageScript",
-	version="4.0.0",
+	version="4.1.0",
 	author="Emmanuel C. Jemeni",
 	author_email="[email protected]",
 	description="FicImage is an application designed to enhance the reading experience of FicHub epubs.",