1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
|
#!/usr/bin/env python3
# Copyright 2004 Gentoo Foundation
# Distributed under the terms of the GNU General Public Licence v2
import sys
from xml.dom.minidom import parse
from optparse import OptionParser, make_option
from xml.dom import NotFoundErr
class IOWrapper:
def __init__(self, object):
self.stream = object
def stream(self):
return self.stream
def write(self, data):
if self.stream == sys.stdin:
sys.stdout.write(data.encode('utf-8'))
else:
file = open(self.stream, 'w')
file.write(data.encode('utf-8'))
file.close()
class Rewriter:
def __init__(self, stream):
self.stream = stream
self.document = parse(stream.stream)
def modifyAttribute(self, elementTag, attribute, value, index=None):
matches = self.document.getElementsByTagName(elementTag)
if matches:
if index is None:
for match in matches:
match.setAttribute(attribute, value)
else:
matches[index].setAttribute(attribute, value)
def deleteAttribute(self, elementTag, attribute, index=None):
matches = self.document.getElementsByTagName(elementTag)
if matches:
if index is None:
for match in matches:
try:
match.removeAttribute(attribute)
except NotFoundErr:
continue
else:
try:
matches[index].removeAttribute(attribute)
except NotFoundErr:
return
def write(self):
self.stream.write(self.document.toxml())
def main():
usage = "Copyright 2004 Gentoo Foundation\n"
usage += "Distributed under the terms of the GNU General Public Lincense v2\n"
usage += "Please contact the Gentoo Java Herd <java@gentoo.org> with problems.\n"
usage += "\n"
usage += "Usage:\n"
usage += " xml-rewrite.py [-f] --delete -e tag [-e tag] -a attribute [-i index]\n"
usage += " xml-rewrite.py [-f] --change -e tag [-e tag] -a attribute -v value [-i index]\n"
usage += "\n"
usage += "If the -f parameter is not utilized, the script will read and\n"
usage += "write to stdin and stdout respectively. The use of quotes on\n"
usage += "parameters will break the script.\n"
def error(message):
print("ERROR: " + message)
sys.exit(1)
options_list = [
make_option("-f", "--file", type="string", dest="file",
help="Read input from file instead of stdin"),
make_option(
"-c",
"--change",
action="store_true",
dest="doAdd",
default=False,
help="Change the value of an attribute. If it does not exist, it will be created."),
make_option(
"-d",
"--delete",
action="store_true",
dest="doDelete",
default=False,
help="Delete an attribute from matching elements."),
make_option(
"-e",
"--element",
action="append",
dest="elements",
help="Tag of the element of which the attributes to be changed. These can be chained for multiple elements."),
make_option(
"-a",
"--attribute",
type="string",
dest="attribute",
help="Attribute of the matching elements to change."),
make_option("-v", "--value", type="string", dest="value",
help="Value to set the attribute to."),
make_option(
"-i",
"--index",
type="int",
dest="index",
help="Index of the match. If none is specified, the changes will be applied to all matches within the document.")
]
parser = OptionParser(usage, options_list)
(options, args) = parser.parse_args()
# Invalid Arguments Must be smited!
if not options.doAdd and not options.doDelete:
print(usage)
print()
error("No action was specified.")
if options.doAdd and options.doDelete:
error("Unable to perform multiple actions simultaneously.")
if not options.elements or not options.attribute:
error("At least one element and attribute must be specified.")
if options.doAdd and not options.value:
error("You must specify values for the attributes to be modified.")
# End Invalid Arguments Check
if options.file:
source = options.file
else:
source = sys.stdin
rewriter = Rewriter(IOWrapper(source))
if options.doDelete:
for element in options.elements:
rewriter.deleteAttribute(element, options.attribute, options.index)
if options.doAdd:
for element in options.elements:
rewriter.modifyAttribute(
element,
options.attribute,
options.value,
options.index)
rewriter.write()
if __name__ == '__main__':
main()
|
GitWeb