mirror of
https://github.com/kovidgoyal/calibre.git
synced 2025-07-31 14:33:54 -04:00
...
This commit is contained in:
parent
96d51719af
commit
629da2b324
@ -11,7 +11,7 @@ import struct, datetime, sys, os, shutil
|
|||||||
from collections import OrderedDict
|
from collections import OrderedDict
|
||||||
from calibre.utils.date import utc_tz
|
from calibre.utils.date import utc_tz
|
||||||
from calibre.ebooks.mobi.langcodes import main_language, sub_language
|
from calibre.ebooks.mobi.langcodes import main_language, sub_language
|
||||||
from calibre.ebooks.mobi.writer2.utils import (decode_hex_number, decint,
|
from calibre.ebooks.mobi.utils import (decode_hex_number, decint,
|
||||||
get_trailing_data)
|
get_trailing_data)
|
||||||
from calibre.utils.magick.draw import identify_data
|
from calibre.utils.magick.draw import identify_data
|
||||||
|
|
||||||
@ -738,8 +738,7 @@ class CNCX(object) : # {{{
|
|||||||
|
|
||||||
class TextRecord(object): # {{{
|
class TextRecord(object): # {{{
|
||||||
|
|
||||||
def __init__(self, idx, record, extra_data_flags, decompress, index_record,
|
def __init__(self, idx, record, extra_data_flags, decompress):
|
||||||
doc_type):
|
|
||||||
self.trailing_data, self.raw = get_trailing_data(record.raw, extra_data_flags)
|
self.trailing_data, self.raw = get_trailing_data(record.raw, extra_data_flags)
|
||||||
self.raw = decompress(self.raw)
|
self.raw = decompress(self.raw)
|
||||||
if 0 in self.trailing_data:
|
if 0 in self.trailing_data:
|
||||||
@ -751,60 +750,6 @@ class TextRecord(object): # {{{
|
|||||||
|
|
||||||
self.idx = idx
|
self.idx = idx
|
||||||
|
|
||||||
if 'indexing' in self.trailing_data and index_record is not None:
|
|
||||||
self.interpret_indexing(doc_type, index_record.indices)
|
|
||||||
|
|
||||||
def interpret_indexing(self, doc_type, indices):
|
|
||||||
raw = self.trailing_data['indexing']
|
|
||||||
ident, consumed = decint(raw)
|
|
||||||
raw = raw[consumed:]
|
|
||||||
entry_type = ident & 0b111
|
|
||||||
index_entry_idx = ident >> 3
|
|
||||||
index_entry = None
|
|
||||||
for i in indices:
|
|
||||||
if i.index == index_entry_idx:
|
|
||||||
index_entry = i.label
|
|
||||||
break
|
|
||||||
self.trailing_data['interpreted_indexing'] = (
|
|
||||||
'Type: %s, Index Entry: %s'%(entry_type, index_entry))
|
|
||||||
if doc_type == 2: # Book
|
|
||||||
self.interpret_book_indexing(raw, entry_type)
|
|
||||||
|
|
||||||
def interpret_book_indexing(self, raw, entry_type):
|
|
||||||
arg1, consumed = decint(raw)
|
|
||||||
raw = raw[consumed:]
|
|
||||||
if arg1 != 0:
|
|
||||||
raise ValueError('TBS index entry has unknown arg1: %d'%
|
|
||||||
arg1)
|
|
||||||
if entry_type == 2:
|
|
||||||
desc = ('This record has only a single starting or a single'
|
|
||||||
' ending point')
|
|
||||||
if raw:
|
|
||||||
raise ValueError('TBS index entry has unknown extra bytes:'
|
|
||||||
' %r'%raw)
|
|
||||||
elif entry_type == 3:
|
|
||||||
desc = ('This record is spanned by a single node (i.e. it'
|
|
||||||
' has no start or end points)')
|
|
||||||
arg2, consumed = decint(raw)
|
|
||||||
if arg2 != 0:
|
|
||||||
raise ValueError('TBS index entry has unknown arg2: %d'%
|
|
||||||
arg2)
|
|
||||||
elif entry_type == 6:
|
|
||||||
if len(raw) != 1:
|
|
||||||
raise ValueError('TBS index entry has unknown extra bytes:'
|
|
||||||
' %r'%raw)
|
|
||||||
num = ord(raw[0])
|
|
||||||
# An unmatched starting or ending point each contributes 1 to
|
|
||||||
# this count. A matched pair of starting and ending points
|
|
||||||
# together contribute 1 to this count. Note that you can only
|
|
||||||
# ever have either 1 unmatched start point or 1 unmatched end
|
|
||||||
# point, never both (logically impossible).
|
|
||||||
desc = ('This record has %d starting/ending points and/or complete'
|
|
||||||
' nodes.')%num
|
|
||||||
else:
|
|
||||||
raise ValueError('Unknown TBS index entry type: %d for book'%entry_type)
|
|
||||||
self.trailing_data['interpreted_indexing'] += ' :: ' + desc
|
|
||||||
|
|
||||||
def dump(self, folder):
|
def dump(self, folder):
|
||||||
name = '%06d'%self.idx
|
name = '%06d'%self.idx
|
||||||
with open(os.path.join(folder, name+'.txt'), 'wb') as f:
|
with open(os.path.join(folder, name+'.txt'), 'wb') as f:
|
||||||
@ -910,8 +855,7 @@ class MOBIFile(object): # {{{
|
|||||||
if fntbr == 0xffffffff:
|
if fntbr == 0xffffffff:
|
||||||
fntbr = len(self.records)
|
fntbr = len(self.records)
|
||||||
self.text_records = [TextRecord(r, self.records[r],
|
self.text_records = [TextRecord(r, self.records[r],
|
||||||
self.mobi_header.extra_data_flags, decompress, self.index_record,
|
self.mobi_header.extra_data_flags, decompress) for r in xrange(1,
|
||||||
self.mobi_header.type_raw) for r in xrange(1,
|
|
||||||
min(len(self.records), ntr+1))]
|
min(len(self.records), ntr+1))]
|
||||||
self.image_records, self.binary_records = [], []
|
self.image_records, self.binary_records = [], []
|
||||||
for i in xrange(fntbr, len(self.records)):
|
for i in xrange(fntbr, len(self.records)):
|
||||||
|
@ -18,7 +18,7 @@ from calibre.ebooks.compression.palmdoc import compress_doc
|
|||||||
from calibre.ebooks.mobi.langcodes import iana2mobi
|
from calibre.ebooks.mobi.langcodes import iana2mobi
|
||||||
from calibre.utils.filenames import ascii_filename
|
from calibre.utils.filenames import ascii_filename
|
||||||
from calibre.ebooks.mobi.writer2 import PALMDOC, UNCOMPRESSED
|
from calibre.ebooks.mobi.writer2 import PALMDOC, UNCOMPRESSED
|
||||||
from calibre.ebooks.mobi.writer2.utils import (rescale_image, encint)
|
from calibre.ebooks.mobi.utils import (rescale_image, encint)
|
||||||
|
|
||||||
EXTH_CODES = {
|
EXTH_CODES = {
|
||||||
'creator': 100,
|
'creator': 100,
|
||||||
|
Loading…
x
Reference in New Issue
Block a user