mirror of
https://github.com/gryf/ebook-converter.git
synced 2026-01-31 10:55:44 +01:00
gryf
0f9792df36
Here is the first batch of modules, which are needed for converting several formats to LRF. Some of the logic has been change, more cleanups will follow.
66 lines
2.2 KiB
Python
66 lines
2.2 KiB
Python
#!/usr/bin/env python2
|
|
# vim:fileencoding=utf-8
|
|
from __future__ import absolute_import, division, print_function, unicode_literals
|
|
|
|
__license__ = 'GPL v3'
|
|
__copyright__ = '2013, Kovid Goyal <kovid at kovidgoyal.net>'
|
|
|
|
from collections import OrderedDict
|
|
from ebook_converter.polyglot.builtins import iteritems, unicode_type
|
|
|
|
|
|
class Note(object):
|
|
|
|
def __init__(self, namespace, parent, rels):
|
|
self.type = namespace.get(parent, 'w:type', 'normal')
|
|
self.parent = parent
|
|
self.rels = rels
|
|
self.namespace = namespace
|
|
|
|
def __iter__(self):
|
|
for p in self.namespace.descendants(self.parent, 'w:p', 'w:tbl'):
|
|
yield p
|
|
|
|
|
|
class Footnotes(object):
|
|
|
|
def __init__(self, namespace):
|
|
self.namespace = namespace
|
|
self.footnotes = {}
|
|
self.endnotes = {}
|
|
self.counter = 0
|
|
self.notes = OrderedDict()
|
|
|
|
def __call__(self, footnotes, footnotes_rels, endnotes, endnotes_rels):
|
|
XPath, get = self.namespace.XPath, self.namespace.get
|
|
if footnotes is not None:
|
|
for footnote in XPath('./w:footnote[@w:id]')(footnotes):
|
|
fid = get(footnote, 'w:id')
|
|
if fid:
|
|
self.footnotes[fid] = Note(self.namespace, footnote, footnotes_rels)
|
|
|
|
if endnotes is not None:
|
|
for endnote in XPath('./w:endnote[@w:id]')(endnotes):
|
|
fid = get(endnote, 'w:id')
|
|
if fid:
|
|
self.endnotes[fid] = Note(self.namespace, endnote, endnotes_rels)
|
|
|
|
def get_ref(self, ref):
|
|
fid = self.namespace.get(ref, 'w:id')
|
|
notes = self.footnotes if ref.tag.endswith('}footnoteReference') else self.endnotes
|
|
note = notes.get(fid, None)
|
|
if note is not None and note.type == 'normal':
|
|
self.counter += 1
|
|
anchor = 'note_%d' % self.counter
|
|
self.notes[anchor] = (unicode_type(self.counter), note)
|
|
return anchor, unicode_type(self.counter)
|
|
return None, None
|
|
|
|
def __iter__(self):
|
|
for anchor, (counter, note) in iteritems(self.notes):
|
|
yield anchor, counter, note
|
|
|
|
@property
|
|
def has_notes(self):
|
|
return bool(self.notes)
|