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

...

Browse Source
This commit is contained in:
Kovid Goyal 2010-03-10 08:24:23 -07:00
commit bb9d414c4a
1 changed files with 39 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
src/calibre/ebooks/metadata/topaz.py
View File

@ -115,12 +115,19 @@ class MetadataUpdater(object):
first = raw.find('metadata') first = raw.find('metadata')
if first < 0: if first < 0:
raise ValueError('Invalid Topaz file') raise ValueError('Invalid Topaz file')
second = raw.find('metadata', first+10)
if second < 0:
raise ValueError('Invalid Topaz file')
self.md_start = second-1
self.data = StreamSlicer(stream) self.data = StreamSlicer(stream)
self.header_records, = unpack('>B',self.data[4]) self.header_records, = unpack('>B',self.data[4])
self.get_topaz_headers()
# Seek the metadata block
md_block_offset, spam = self.decode_vwi(self.data[first+9:first+13])
md_block_offset += self.base
if self.data[md_block_offset+1:md_block_offset+9] != 'metadata':
raise ValueError('Invalid Topaz file')
else:
self.md_start = md_block_offset
offset = self.get_md_header(self.md_start) offset = self.get_md_header(self.md_start)
self.metadata = {} self.metadata = {}
self.md_end = self.get_original_metadata(offset) self.md_end = self.get_original_metadata(offset)
@ -195,13 +202,16 @@ class MetadataUpdater(object):
def generate_dkey(self): def generate_dkey(self):
for x in self.topaz_headers: for x in self.topaz_headers:
#print "dkey['blocks']: %s" % self.topaz_headers[x]['blocks']
if self.topaz_headers[x]['tag'] == 'dkey': if self.topaz_headers[x]['tag'] == 'dkey':
if self.topaz_headers[x]['blocks']:
offset = self.base + self.topaz_headers[x]['blocks'][0]['hdr_offset'] offset = self.base + self.topaz_headers[x]['blocks'][0]['hdr_offset']
len_uncomp = self.topaz_headers[x]['blocks'][0]['len_uncomp'] len_uncomp = self.topaz_headers[x]['blocks'][0]['len_uncomp']
break break
else:
return None
dkey = self.topaz_headers[x] dkey = self.topaz_headers[x]
dks = StringIO.StringIO() dks = StringIO.StringIO()
dks.write('d@')
dks.write(self.encode_vwi(len(dkey['tag']))) dks.write(self.encode_vwi(len(dkey['tag'])))
offset += 1 offset += 1
dks.write(dkey['tag']) dks.write(dkey['tag'])
@ -242,6 +252,8 @@ class MetadataUpdater(object):
offset = 5 offset = 5
topaz_headers = {} topaz_headers = {}
dkey_offset = 0
highest_payload_offset = 0
for x in range(self.header_records): for x in range(self.header_records):
marker = self.data[offset] marker = self.data[offset]
offset += 1 offset += 1
@ -254,6 +266,12 @@ class MetadataUpdater(object):
blocks = {} blocks = {}
for val in range(num_vals): for val in range(num_vals):
hdr_offset, consumed = self.decode_vwi(self.data[offset:offset+4]) hdr_offset, consumed = self.decode_vwi(self.data[offset:offset+4])
if tag == 'dkey':
dkey_offset = hdr_offset
if tag != 'metadata':
if hdr_offset > highest_payload_offset:
highest_payload_offset = hdr_offset
#print "adjusting highest_payload_offset to 0x%x (%s)" % (hdr_offset,tag)
offset += consumed offset += consumed
len_uncomp, consumed = self.decode_vwi(self.data[offset:offset+4]) len_uncomp, consumed = self.decode_vwi(self.data[offset:offset+4])
offset += consumed offset += consumed
@ -262,11 +280,13 @@ class MetadataUpdater(object):
blocks[val] = dict(hdr_offset=hdr_offset,len_uncomp=len_uncomp,len_comp=len_comp) blocks[val] = dict(hdr_offset=hdr_offset,len_uncomp=len_uncomp,len_comp=len_comp)
topaz_headers[x] = dict(tag=tag,blocks=blocks) topaz_headers[x] = dict(tag=tag,blocks=blocks)
self.topaz_headers = topaz_headers self.topaz_headers = topaz_headers
self.highest_payload_offset = highest_payload_offset
eod = self.data[offset] self.eod = self.data[offset]
offset += 1 offset += 1
self.base = offset self.base = offset
self.base_value = None
if dkey_offset:
self.base_value = self.data[offset:offset + dkey_offset]
return md_header_offset, topaz_headers return md_header_offset, topaz_headers
def generate_metadata_stream(self): def generate_metadata_stream(self):
@ -332,7 +352,8 @@ class MetadataUpdater(object):
self.metadata[item]['metadata'] = value self.metadata[item]['metadata'] = value
return return
self.get_topaz_headers() if self.md_start > self.highest_payload_offset:
raise ValueError('Unable to update metadata')
try: try:
from calibre.ebooks.conversion.config import load_defaults from calibre.ebooks.conversion.config import load_defaults
@ -357,10 +378,15 @@ class MetadataUpdater(object):
self.stream.seek(0) self.stream.seek(0)
self.stream.truncate(0) self.stream.truncate(0)
self.stream.write(head) self.stream.write(head)
self.stream.write(self.eod)
if self.base_value:
self.stream.write(self.base_value)
if dkey:
self.stream.write(dkey) self.stream.write(dkey)
self.stream.write(updated_metadata) self.stream.write(updated_metadata)
self.stream.write(tail) self.stream.write(tail)
def set_metadata(stream, mi): def set_metadata(stream, mi):
mu = MetadataUpdater(stream) mu = MetadataUpdater(stream)
mu.update(mi) mu.update(mi)