X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/e09496eb5725b2dd860dfc6b91cbfc88b6237041..933a58adfc8abf2d4c9843afb7fda4cb4d06d030:/lib/librarian/html.py?ds=inline diff --git a/lib/librarian/html.py b/lib/librarian/html.py index 3dd70f56e..94514df3c 100644 --- a/lib/librarian/html.py +++ b/lib/librarian/html.py @@ -53,9 +53,13 @@ def transform(input_filename, output_filename): 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): @@ -117,9 +121,9 @@ def extract_fragments(input_filename): 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()