doc = etree.parse(doc_file, parser)
result = doc.xslt(style)
- add_anchors(result.getroot())
- add_table_of_contents(result.getroot())
- result.write(output_filename, xml_declaration=True, pretty_print=True, encoding='utf-8')
+ if result.find('//h1') is not None:
+ add_anchors(result.getroot())
+ add_table_of_contents(result.getroot())
+ result.write(output_filename, xml_declaration=False, pretty_print=True, encoding='utf-8')
+ return True
+ else:
+ return False
class Fragment(object):
fragment = Fragment(id=element.get('fid'), themes=element.text)
# Append parents
- if element.getparent().tag != 'body':
+ if element.getparent().get('id', None) != 'book-text':
parents = [element.getparent()]
- while parents[-1].getparent().tag != 'body':
+ while parents[-1].getparent().get('id', None) != 'book-text':
parents.append(parents[-1].getparent())
parents.reverse()