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

Handle data URLs for covers in Get Books results

Browse Source
This commit is contained in:
Kovid Goyal 2015-12-20 10:37:57 +05:30
parent 21346e3e20
commit dbf36e96e7
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/calibre/gui2/store/search/download_thread.py
View File

@ -6,7 +6,7 @@ __license__ = 'GPL 3'
__copyright__ = '2011, John Schember <john@nachtimwald.com>' __copyright__ = '2011, John Schember <john@nachtimwald.com>'
__docformat__ = 'restructuredtext en' __docformat__ = 'restructuredtext en'
import traceback import traceback, base64
from contextlib import closing from contextlib import closing
from threading import Thread from threading import Thread
from Queue import Queue from Queue import Queue
@ -139,6 +139,8 @@ class CoverThreadPool(GenericDownloadThreadPool):
self.tasks.put((search_result, update_callback, timeout)) self.tasks.put((search_result, update_callback, timeout))
GenericDownloadThreadPool.add_task(self) GenericDownloadThreadPool.add_task(self)
def decode_data_url(url):
return base64.standard_b64decode(url.partition(',')[2])
class CoverThread(Thread): class CoverThread(Thread):
@ -159,8 +161,11 @@ class CoverThread(Thread):
try: try:
result, callback, timeout = self.tasks.get() result, callback, timeout = self.tasks.get()
if result and result.cover_url: if result and result.cover_url:
with closing(self.br.open(result.cover_url, timeout=timeout)) as f: if result.cover_url.startswith('data:'):
result.cover_data = f.read() result.cover_data = decode_data_url(result.cover_url)
else:
with closing(self.br.open(result.cover_url, timeout=timeout)) as f:
result.cover_data = f.read()
result.cover_data = scale_image(result.cover_data, 64, 64)[2] result.cover_data = scale_image(result.cover_data, 64, 64)[2]
callback() callback()
self.tasks.task_done() self.tasks.task_done()