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 2012-04-20 20:49:23 +05:30
parent fbcd3eb279
commit 03ed4010f5
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/calibre/ebooks/mobi/writer8/skeleton.py
View File

@ -165,7 +165,7 @@ class Chunker(object):
# Set this to a list to enable dumping of the original and rebuilt # Set this to a list to enable dumping of the original and rebuilt
# html files for debugging # html files for debugging
self.orig_dumps = None orig_dumps = None
for i, item in enumerate(self.oeb.spine): for i, item in enumerate(self.oeb.spine):
root = self.remove_namespaces(self.data(item)) root = self.remove_namespaces(self.data(item))
@ -193,8 +193,8 @@ class Chunker(object):
# for all chunks # for all chunks
self.skeletons.append(Skeleton(i, item, root, chunks)) self.skeletons.append(Skeleton(i, item, root, chunks))
if self.orig_dumps: if orig_dumps:
self.dump() self.dump(orig_dumps)
# Create the SKEL and Chunk tables # Create the SKEL and Chunk tables
self.skel_table = [] self.skel_table = []
@ -356,7 +356,7 @@ class Chunker(object):
return re.sub(br'<[^>]+(kindle:pos:fid:\d{4}:\d{10})', sub, text) return re.sub(br'<[^>]+(kindle:pos:fid:\d{4}:\d{10})', sub, text)
def dump(self): def dump(self, orig_dumps):
import tempfile, shutil, os import tempfile, shutil, os
tdir = os.path.join(tempfile.gettempdir(), 'skeleton') tdir = os.path.join(tempfile.gettempdir(), 'skeleton')
self.log('Skeletons dumped to:', tdir) self.log('Skeletons dumped to:', tdir)
@ -368,7 +368,7 @@ class Chunker(object):
os.makedirs(x) os.makedirs(x)
for i, skeleton in enumerate(self.skeletons): for i, skeleton in enumerate(self.skeletons):
with open(os.path.join(orig, '%04d.html'%i), 'wb') as f: with open(os.path.join(orig, '%04d.html'%i), 'wb') as f:
f.write(self.orig_dumps[i]) f.write(orig_dumps[i])
with open(os.path.join(rebuilt, '%04d.html'%i), 'wb') as f: with open(os.path.join(rebuilt, '%04d.html'%i), 'wb') as f:
f.write(skeleton.rebuild()) f.write(skeleton.rebuild())