def __init__(self, title):
self.title = title
self.html = HTMLFile(title)
self.opf = OPFFile(title)
self.ncx = NCXFile(title)
self.opf.setHtml(self.html)
self.opf.setNcx(self.ncx)
self.counters = collections.defaultdict(int)
self.images = {}
self.kindlegen = 'kindlegen'
class Book(object):
def __init__(self, title):
self.title = title
self.html = HTMLFile(title)
self.opf = OPFFile(title)
self.ncx = NCXFile(title)
self.opf.setHtml(self.html)
self.opf.setNcx(self.ncx)
self.counters = collections.defaultdict(int)
self.images = {}
self.kindlegen = 'kindlegen'
def setKindlegen(self, kindlegen):
self.kindlegen = kindlegen
def generateId(self, category='object'):
self.counters[category] += 1
return "{category}{counter}".format(
category=category,
counter=self.counters[category]
)
def addHeading(self, text, level=1):
if not (1 <= level <= 6):
raise IndexError("level %d is illegal." % level)
self.addTocEntry(text, level)
self.html.addHtml(u'<h{lvl}>{heading}</h{lvl}>'.format(
lvl=level,
heading=self.html.escape(text))
)
def addParagraph(self, text, indent=None):
style = ''
if indent is not None:
indent = re.sub(r'[<>"/]', r'', indent)
style = ' style="text-indent: {indent}"'.format(indent=indent)
self.html.addHtml(u'<p{style}>{content}</p>'.format(
style=style,
content=self.html.escape(text))
)
def addAuthoringInfo(self, author='', date=None, verb="published"):
info = verb
if len(author) > 0:
info += " by {author}".format(author=author)
if date is not None:
dateStr = date.strftime("%x, %X")
info += " on {date}".format(date=dateStr)
info = "<small>{info}</small>".format(info=self.html.escape(info))
self.html.addHtml(info)
self.html.addHtml("<br/>")
def addImage(self, src, width=0.95, center=True):
_, ext = os.path.splitext(src)
imageId = self.generateId("image") + ext
self.images[imageId] = src
html = u'<img src="{id}" width="{widthpercent}%%" />'.format(
id=imageId,
widthpercent=int(width*100.0)
)
if center:
html = u'<p style="text-align: center">{html}</p>'.format(html=html)
self.html.addHtml(html)
def addTocEntry(self, title, level=1):
anchodId = self.generateId("anchor")
self.ncx.addTocEntry(title, self.html.filename, anchodId, level)
self.html.addHtml(u'<a name="{id}"></a>'.format(id=anchodId))
def addPagebreak(self):
# this shouldn't be in this class
pb = self.html.document.createElementNS('', 'mbp:pagebreak')
self.html.body.appendChild(pb)
@contextmanager
def tempObject(self):
for tempName, src in self.images.iteritems():
urllib.urlretrieve(src, tempName)
yield
for tempName in self.images.keys():
os.remove(tempName)
def createMobi(self):
with self.tempObject():
with self.html.tempObject():
with self.ncx.tempObject():
with self.opf.tempObject():
call_kindlegen(self.kindlegen, self.opf.filename)
