From 5400caf9e791572a0800ab69cc5af4044f98268e Mon Sep 17 00:00:00 2001 From: =?utf8?q?Marek=20St=C4=99pniowski?= Date: Thu, 28 Aug 2008 15:05:16 +0200 Subject: [PATCH] Extracted transformation in book2html to transform function. Put all command line logic behind ifmain guard. --- bin/book2html.py | 56 ++++++++++++++++++++++++++---------------------- 1 file changed, 30 insertions(+), 26 deletions(-) diff --git a/bin/book2html.py b/bin/book2html.py index 5bd2bb527..e68435974 100755 --- a/bin/book2html.py +++ b/bin/book2html.py @@ -9,32 +9,11 @@ import sys from lxml import etree -# Parse args -usage = """Usage: %prog [options] SOURCE [SOURCE...] -Convert SOURCE files to HTML format.""" - -parser = optparse.OptionParser(usage=usage) - -parser.add_option('-v', '--verbose', action='store_true', dest='verbose', default=False, - help='print status messages to stdout') - -options, input_filenames = parser.parse_args() - -if len(input_filenames) < 1: - parser.print_help() - exit(1) - -# Parse XSLT -style = etree.parse('book2html.xslt') - -# Do some real work -for input_filename in input_filenames: - if options.verbose: - print input_filename - - output_filename = os.path.splitext(input_filename)[0] + '.html' - - # Transform +def transform(input_filename, output_filename): + """Transforms file input_filename in XML to output_filename in XHTML.""" + # Parse XSLT + style = etree.parse('book2html.xslt') + doc_file = cStringIO.StringIO() expr = re.compile(r'/\s', re.MULTILINE | re.UNICODE); @@ -53,3 +32,28 @@ for input_filename in input_filenames: result = doc.xslt(style) result.write(output_filename, xml_declaration=True, pretty_print=True, encoding='utf-8') + +if __name__ == '__main__': + # Parse commandline arguments + usage = """Usage: %prog [options] SOURCE [SOURCE...] + Convert SOURCE files to HTML format.""" + + parser = optparse.OptionParser(usage=usage) + + parser.add_option('-v', '--verbose', action='store_true', dest='verbose', default=False, + help='print status messages to stdout') + + options, input_filenames = parser.parse_args() + + if len(input_filenames) < 1: + parser.print_help() + exit(1) + + # Do some real work + for input_filename in input_filenames: + if options.verbose: + print input_filename + + output_filename = os.path.splitext(input_filename)[0] + '.html' + transform(input_filename, output_filename) + -- 2.20.1