From 2fe500fbdb533f88588e4a55e68d007a7cb7398b Mon Sep 17 00:00:00 2001
From: Your Name <you@example.com>
Date: Sun, 12 Feb 2023 21:54:21 -0500
Subject: [PATCH] fix google book cover

---
 catalog/common/downloaders.py |  4 +++-
 catalog/sites/google_books.py | 13 ++++++++++---
 2 files changed, 13 insertions(+), 4 deletions(-)

diff --git a/catalog/common/downloaders.py b/catalog/common/downloaders.py
index 2baf4e3f..69288186 100644
--- a/catalog/common/downloaders.py
+++ b/catalog/common/downloaders.py
@@ -231,8 +231,10 @@ class ImageDownloaderMixin:
             return RESPONSE_NETWORK_ERROR
 
     @classmethod
-    def download_image(cls, image_url, page_url):
+    def download_image(cls, image_url, page_url, headers=None):
         imgdl = cls(image_url, page_url)
+        if headers is not None:
+            imgdl.headers = headers
         try:
             image = imgdl.download().content
             image_extention = imgdl.extention
diff --git a/catalog/sites/google_books.py b/catalog/sites/google_books.py
index b4e34f6a..f922a37c 100644
--- a/catalog/sites/google_books.py
+++ b/catalog/sites/google_books.py
@@ -56,8 +56,15 @@ class GoogleBooks(AbstractSite):
         brief = re.sub(r"<.*?>", "", brief.replace("<br", "\n<br"))
         img_url = None
         if "imageLinks" in b["volumeInfo"]:
-            img_url = b["volumeInfo"]["imageLinks"]["thumbnail"]
-            img_url = img_url.replace("zoom=1", "")
+            # if "extraLarge" in b["volumeInfo"]["imageLinks"]:
+            #     img_url = b["volumeInfo"]["imageLinks"]["extraLarge"]
+            # elif "large" in b["volumeInfo"]["imageLinks"]:
+            #     img_url = b["volumeInfo"]["imageLinks"]["large"]
+            # elif "thumbnail" in b["volumeInfo"]["imageLinks"]:
+            #     img_url = b["volumeInfo"]["imageLinks"]["thumbnail"]
+            if "thumbnail" in b["volumeInfo"]["imageLinks"]:
+                img_url = b["volumeInfo"]["imageLinks"]["thumbnail"]
+                img_url = img_url.replace("zoom=1", "")
         isbn10 = None
         isbn13 = None
         for iid in (
@@ -71,7 +78,7 @@ class GoogleBooks(AbstractSite):
                 isbn13 = iid["identifier"]
         isbn = isbn13  # if isbn13 is not None else isbn10
 
-        raw_img, ext = BasicImageDownloader.download_image(img_url, self.url)
+        raw_img, ext = BasicImageDownloader.download_image(img_url, None, headers={})
         data = {
             "title": title,
             "subtitle": subtitle,