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

Make cover extraction in lrf-meta much more efficient.

Browse Source
This commit is contained in:
Kovid Goyal 2008-05-29 13:04:20 -07:00
parent d77746ea07
commit 96aa2c8b0b
1 changed files with 9 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
src/calibre/ebooks/lrf/meta.py
View File

@ -563,17 +563,14 @@ class LRFMetaFile(object):
def get_cover(self): def get_cover(self):
from calibre.ebooks.lrf.objects import get_object from calibre.ebooks.lrf.objects import get_object
for id, offset, size in self.get_objects_by_type(0x06): #Blocks for id, offset, size in self.get_objects_by_type(0x0C):
block = get_object(None, self._file, id, offset, size, self.xor_key) image = get_object(None, self._file, id, offset, size, self.xor_key)
tid, ref = struct.unpack('<HI', block.stream) id, offset, size = self.get_object_by_id(image.refstream)[:3]
if tid == 0xF503: image_stream = get_object(None, self._file, id, offset, size, self.xor_key)
obj_id, offset, size, obj_type = self.get_object_by_id(ref) return image_stream.file.rpartition('.')[-1], image_stream.stream
if obj_type == 0x0C: # Image return None
image = get_object(None, self._file, obj_id, offset, size, self.xor_key)
id, offset, size = self.get_object_by_id(image.refstream)[:3]
image_stream = get_object(None, self._file, id, offset, size, self.xor_key)
return image_stream.file.rpartition('.')[-1], image_stream.stream
return (None, None)
def option_parser(): def option_parser():
from optparse import OptionParser from optparse import OptionParser
@ -680,7 +677,7 @@ def main(args=sys.argv):
if options.get_cover: if options.get_cover:
ext, data = lrf.get_cover() ext, data = lrf.get_cover()
if data: if data:
cover = os.path.basename(args[1])+"_cover."+ext cover = os.path.splitext(os.path.basename(args[1]))[0]+"_cover."+ext
open(cover, 'wb').write(data) open(cover, 'wb').write(data)
print 'Cover:', cover print 'Cover:', cover
else: else: