Cutlery/calibre
1
0
Fork 0
mirror of https://github.com/kovidgoyal/calibre.git synced 2025-07-09 03:04:10 -04:00
Code Issues Packages Projects Releases Wiki Activity

Download remote background images and cache

Browse Source
This commit is contained in:
Kovid Goyal 2024-02-16 11:34:06 +05:30
parent e15ab16153
commit be8082790a
No known key found for this signature in database
GPG Key ID: 06BC317B515ACE7C
1 changed files with 16 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
src/calibre/gui2/viewer/web_view.py
View File

@ -22,11 +22,14 @@ from calibre.constants import (
from calibre.ebooks.metadata.book.base import field_metadata from calibre.ebooks.metadata.book.base import field_metadata
from calibre.ebooks.oeb.polish.utils import guess_type from calibre.ebooks.oeb.polish.utils import guess_type
from calibre.gui2 import choose_images, config, error_dialog, safe_open_url from calibre.gui2 import choose_images, config, error_dialog, safe_open_url
from calibre.gui2.viewer import link_prefix_for_location_links, performance_monitor, url_for_book_in_library from calibre.gui2.viewer import (
link_prefix_for_location_links, performance_monitor, url_for_book_in_library,
)
from calibre.gui2.viewer.config import viewer_config_dir, vprefs from calibre.gui2.viewer.config import viewer_config_dir, vprefs
from calibre.gui2.viewer.tts import TTS from calibre.gui2.viewer.tts import TTS
from calibre.gui2.webengine import RestartingWebEngineView from calibre.gui2.webengine import RestartingWebEngineView
from calibre.srv.code import get_translations_data from calibre.srv.code import get_translations_data
from calibre.utils.filenames import make_long_path_useable
from calibre.utils.localization import _, localize_user_manual_link from calibre.utils.localization import _, localize_user_manual_link
from calibre.utils.resources import get_path as P from calibre.utils.resources import get_path as P
from calibre.utils.serialize import json_loads from calibre.utils.serialize import json_loads
@ -93,9 +96,20 @@ def background_image(encoded_fname=''):
img_path = os.path.join(viewer_config_dir, 'background-images', fname) img_path = os.path.join(viewer_config_dir, 'background-images', fname)
mt = guess_type(fname)[0] or 'image/jpeg' mt = guess_type(fname)[0] or 'image/jpeg'
try: try:
with open(img_path, 'rb') as f: with open(make_long_path_useable(img_path), 'rb') as f:
return mt, f.read() return mt, f.read()
except FileNotFoundError: except FileNotFoundError:
if fname.startswith('https://') or fname.startswith('http://'):
from calibre import browser
br = browser()
try:
with br.open(fname) as src:
data = src.read()
except Exception:
return mt, b''
with open(make_long_path_useable(img_path), 'wb') as dest:
dest.write(data)
return mt, data
return mt, b'' return mt, b''