change_xsl_generic.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# Written by Børre Gaup <borre.gaup@samediggi.no>

import sys
from lxml import etree

def usage():
    print 'This is a script that changes empty values in a corpus xsl file'
    print 'Call the program like this: change_xsl.py variable-value-pairs filename'
    print 'This requires an odd number of args to the script'
    print 'If a value contains a space, use "-chars around it.'
    print 'e.g. change_xsl_generic.py sub_name "Jens Kristensen" sub_email jens.kristensen@samediggi.no kraken.html.xsl'

# check that there is exactly one argument given to the program
if ( len(sys.argv) % 2 ) != 0:
    usage()
    sys.exit()

# Initiate an empty dict
change_variables = {}

# read in the variable-value pairs, add them in the dict
for index in range(1, len(sys.argv) - 1, 2):
    change_variables[sys.argv[index]] = sys.argv[index + 1].decode('utf8')

xsl_filename = sys.argv[len(sys.argv) - 1]
if (xsl_filename.rfind('.xsl') > 0):
    try:
        tree = etree.parse(xsl_filename)
    except Exception, inst:
        print >>sys.stderr, "Unexpected error opening {}: {}".format(xsl_filename, inst)
        sys.exit(254)

    root = tree.getroot()
    for key, value in change_variables.iteritems():
        variable = root.find("{http://www.w3.org/1999/XSL/Transform}variable[@name='" + key + "']")
        if variable is not None:
            variable.attrib['select'] = "'" + value + "'"
        else:
            print >>sys.stderr, 'Sorry, the xsl variable %s does not exist' % (key)
    root.tail = '\n'
    try:
        tree.write(xsl_filename, encoding="utf-8", xml_declaration = True)
    except IOError:
        print >>sys.stderr, 'cannot write', xsl_filename
        sys.exit(254)

else:
    print >>sys.stderr, "This is not an xsl file: " + xsl_filename
    print >>sys.stderr
    usage()