1 | #!/usr/bin/python -u
|
---|
2 | # $Id: xslt.py 8353 2009-03-17 16:57:50Z mzjn $
|
---|
3 |
|
---|
4 | import sys
|
---|
5 | import libxml2
|
---|
6 | import libxslt
|
---|
7 | from docbook import adjustColumnWidths
|
---|
8 |
|
---|
9 | # Check the arguments
|
---|
10 | usage = "Usage: %s xmlfile.xml xslfile.xsl [outputfile] [param1=val [param2=val]...]" % sys.argv[0]
|
---|
11 |
|
---|
12 | xmlfile = None
|
---|
13 | xslfile = None
|
---|
14 | outfile = "-"
|
---|
15 | params = {}
|
---|
16 |
|
---|
17 | try:
|
---|
18 | xmlfile = sys.argv[1]
|
---|
19 | xslfile = sys.argv[2]
|
---|
20 | except IndexError:
|
---|
21 | print usage
|
---|
22 | sys.exit(1)
|
---|
23 |
|
---|
24 | def quote(astring):
|
---|
25 | if astring.find("'") < 0:
|
---|
26 | return "'" + astring + "'"
|
---|
27 | else:
|
---|
28 | return '"' + astring + '"'
|
---|
29 |
|
---|
30 | try:
|
---|
31 | outfile = sys.argv[3]
|
---|
32 | if outfile.find("=") > 0:
|
---|
33 | name, value = outfile.split("=", 2)
|
---|
34 | params[name] = quote(value)
|
---|
35 | outfile = None
|
---|
36 |
|
---|
37 | count = 4
|
---|
38 | while (sys.argv[count]):
|
---|
39 | try:
|
---|
40 | name, value = sys.argv[count].split("=", 2)
|
---|
41 | if params.has_key(name):
|
---|
42 | print "Warning: '%s' re-specified; replacing value" % name
|
---|
43 | params[name] = quote(value)
|
---|
44 | except ValueError:
|
---|
45 | print "Invalid parameter specification: '" + sys.argv[count] + "'"
|
---|
46 | print usage
|
---|
47 | sys.exit(1)
|
---|
48 | count = count+1
|
---|
49 | except IndexError:
|
---|
50 | pass
|
---|
51 |
|
---|
52 | # ======================================================================
|
---|
53 | # Memory debug specific
|
---|
54 | # libxml2.debugMemory(1)
|
---|
55 |
|
---|
56 | # Setup environment
|
---|
57 | libxml2.lineNumbersDefault(1)
|
---|
58 | libxml2.substituteEntitiesDefault(1)
|
---|
59 | libxslt.registerExtModuleFunction("adjustColumnWidths",
|
---|
60 | "http://nwalsh.com/xslt/ext/xsltproc/python/Table",
|
---|
61 | adjustColumnWidths)
|
---|
62 |
|
---|
63 | # Initialize and run
|
---|
64 | styledoc = libxml2.parseFile(xslfile)
|
---|
65 | style = libxslt.parseStylesheetDoc(styledoc)
|
---|
66 | doc = libxml2.parseFile(xmlfile)
|
---|
67 | result = style.applyStylesheet(doc, params)
|
---|
68 |
|
---|
69 | # Save the result
|
---|
70 | if outfile:
|
---|
71 | style.saveResultToFilename(outfile, result, 0)
|
---|
72 | else:
|
---|
73 | print result
|
---|
74 |
|
---|
75 | # Free things up
|
---|
76 | style.freeStylesheet()
|
---|
77 | doc.freeDoc()
|
---|
78 | result.freeDoc()
|
---|
79 |
|
---|
80 | # Memory debug specific
|
---|
81 | #libxslt.cleanup()
|
---|
82 | #if libxml2.debugMemory(1) != 0:
|
---|
83 | # print "Memory leak %d bytes" % (libxml2.debugMemory(1))
|
---|
84 | # libxml2.dumpMemory()
|
---|