From cec81dc04a23808077505265a3f8bb0e9b6053cb Mon Sep 17 00:00:00 2001 From: =?utf8?q?Micha=C5=82=20G=C3=B3rny?= Date: Tue, 24 Apr 2012 20:29:12 +0200 Subject: [PATCH 1/1] Initial FictionBook 2 output support. This is a work-in-progress and still needs a lot of polishing. --- librarian/fb2.py | 35 ++++++++++++++++++ librarian/fb2/fb2.xslt | 48 +++++++++++++++++++++++++ librarian/fb2/footnotes.xslt | 37 +++++++++++++++++++ librarian/fb2/inline.xslt | 35 ++++++++++++++++++ librarian/fb2/paragraphs.xslt | 30 ++++++++++++++++ librarian/fb2/poems.xslt | 61 +++++++++++++++++++++++++++++++ librarian/fb2/sections.xslt | 43 ++++++++++++++++++++++ librarian/parser.py | 4 +++ scripts/book2fb2 | 68 +++++++++++++++++++++++++++++++++++ 9 files changed, 361 insertions(+) create mode 100644 librarian/fb2.py create mode 100644 librarian/fb2/fb2.xslt create mode 100644 librarian/fb2/footnotes.xslt create mode 100644 librarian/fb2/inline.xslt create mode 100644 librarian/fb2/paragraphs.xslt create mode 100644 librarian/fb2/poems.xslt create mode 100644 librarian/fb2/sections.xslt create mode 100755 scripts/book2fb2 diff --git a/librarian/fb2.py b/librarian/fb2.py new file mode 100644 index 0000000..b0ad410 --- /dev/null +++ b/librarian/fb2.py @@ -0,0 +1,35 @@ +# -*- coding: utf-8 -*- +# +# This file is part of Librarian, licensed under GNU Affero GPLv3 or later. +# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information. +# +import os.path +from copy import deepcopy +from lxml import etree + +from librarian import functions, OutputFile + + +functions.reg_substitute_entities() + +def transform(wldoc, verbose=False, + cover=None, flags=None): + """ produces a FB2 file + + cover: a cover.Cover object or True for default + flags: less-advertising, working-copy + """ + + document = deepcopy(wldoc) + del wldoc + + if flags: + for flag in flags: + document.edoc.getroot().set(flag, 'yes') + + style_filename = os.path.join(os.path.dirname(__file__), 'fb2/fb2.xslt') + style = etree.parse(style_filename) + + result = document.transform(style) + + return OutputFile.from_string(unicode(result).encode('utf-8')) diff --git a/librarian/fb2/fb2.xslt b/librarian/fb2/fb2.xslt new file mode 100644 index 0000000..e9ab6ce --- /dev/null +++ b/librarian/fb2/fb2.xslt @@ -0,0 +1,48 @@ + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + <xsl:apply-templates mode="para" + select="autor_utworu|nazwa_utworu"/> + + + + + + + diff --git a/librarian/fb2/footnotes.xslt b/librarian/fb2/footnotes.xslt new file mode 100644 index 0000000..663b2a3 --- /dev/null +++ b/librarian/fb2/footnotes.xslt @@ -0,0 +1,37 @@ + + + + + + + + + + + fn + +

+ + + + + + + + + note + #fn + + [] + + + diff --git a/librarian/fb2/inline.xslt b/librarian/fb2/inline.xslt new file mode 100644 index 0000000..4151845 --- /dev/null +++ b/librarian/fb2/inline.xslt @@ -0,0 +1,35 @@ + + + + + + + + + + + + + + + + + + + + + + + + + + diff --git a/librarian/fb2/paragraphs.xslt b/librarian/fb2/paragraphs.xslt new file mode 100644 index 0000000..9c7def8 --- /dev/null +++ b/librarian/fb2/paragraphs.xslt @@ -0,0 +1,30 @@ + + + + + + + + + +

+ + + + + + + +

+ + + diff --git a/librarian/fb2/poems.xslt b/librarian/fb2/poems.xslt new file mode 100644 index 0000000..a9ef901 --- /dev/null +++ b/librarian/fb2/poems.xslt @@ -0,0 +1,61 @@ + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + diff --git a/librarian/fb2/sections.xslt b/librarian/fb2/sections.xslt new file mode 100644 index 0000000..7f34bc3 --- /dev/null +++ b/librarian/fb2/sections.xslt @@ -0,0 +1,43 @@ + + + + + + + + + + + + + + + +
+ +
+ + + + + <p><xsl:apply-templates mode="inline"/></p> + + diff --git a/librarian/parser.py b/librarian/parser.py index 2ece72f..6343d21 100644 --- a/librarian/parser.py +++ b/librarian/parser.py @@ -185,6 +185,10 @@ class WLDocument(object): from librarian import mobi return mobi.transform(self, *args, **kwargs) + def as_fb2(self, *args, **kwargs): + from librarian import fb2 + return fb2.transform(self, *args, **kwargs) + def save_output_file(self, output_file, output_path=None, output_dir_path=None, make_author_dir=False, ext=None): if output_dir_path: diff --git a/scripts/book2fb2 b/scripts/book2fb2 new file mode 100755 index 0000000..81a002b --- /dev/null +++ b/scripts/book2fb2 @@ -0,0 +1,68 @@ +#!/usr/bin/env python +# -*- coding: utf-8 -*- +# +# This file is part of Librarian, licensed under GNU Affero GPLv3 or later. +# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information. +# +import os.path +import optparse + +from librarian import DirDocProvider, ParseError +from librarian.parser import WLDocument + + +if __name__ == '__main__': + # Parse commandline arguments + usage = """Usage: %prog [options] SOURCE [SOURCE...] + Convert SOURCE files to FB2 format.""" + + parser = optparse.OptionParser(usage=usage) + + parser.add_option('-v', '--verbose', action='store_true', dest='verbose', default=False, + help='print status messages to stdout') + parser.add_option('-c', '--with-cover', action='store_true', dest='with_cover', default=False, + help='create default cover') + parser.add_option('-w', '--working-copy', action='store_true', dest='working_copy', default=False, + help='mark the output as a working copy') + parser.add_option('-d', '--make-dir', action='store_true', dest='make_dir', default=False, + help='create a directory for author and put the PDF in it') + parser.add_option('-o', '--output-file', dest='output_file', metavar='FILE', + help='specifies the output file') + parser.add_option('-O', '--output-dir', dest='output_dir', metavar='DIR', + help='specifies the directory for output') + + options, input_filenames = parser.parse_args() + + if len(input_filenames) < 1: + parser.print_help() + exit(1) + + flags = [] + if options.working_copy: + flags.append('working-copy') + + # Do some real work + try: + for main_input in input_filenames: + if options.verbose: + print main_input + + path, fname = os.path.realpath(main_input).rsplit('/', 1) + provider = DirDocProvider(path) + if not (options.output_file or options.output_dir): + output_file = os.path.splitext(main_input)[0] + '.fb2' + else: + output_file = None + + doc = WLDocument.from_file(main_input, provider=provider) + fb2 = doc.as_fb2(cover=options.with_cover, flags=flags) + + doc.save_output_file(fb2, + output_file, options.output_dir, options.make_dir, 'fb2') + + except ParseError, e: + print '%(file)s:%(name)s:%(message)s' % { + 'file': main_input, + 'name': e.__class__.__name__, + 'message': e + } -- 2.20.1