from xml.etree import ElementTree as ET
import os, sys, re
debug = 0
def printTree(node,pattern):
if node != None:
for child in node:
if pattern.match(child.text):
print child.tag
else:
print child.tag, "=>", child.text
printTree(child,pattern)
def main():
pattern = re.compile('\s+')
file_name = os.popen('echo %userprofile%').read()
xml_file = os.path.join(file_name.strip(), ".p4qt\\customtools.xml")
try:
tree = ET.parse(xml_file)
except Exception, inst:
print "Unexpected error opening %s: %s" % (xml_file, inst)
return
if debug == 1:
printTree(tree.getroot(),pattern)
child = ET.SubElement(tree.getroot(),"CustomToolDef")
subChild1 = ET.SubElement(child,"Definition")
subChild2 = ET.SubElement(subChild1,"Name")
subChild2.text = "DeleteC"
subChild3 = ET.SubElement(subChild1,"Command")
subChild3.text = "p4"
subChild4 = ET.SubElement(subChild1,"Arguments")
subChild4.text = "-u smoon -p20082 client -d -f $D"
subChild5 = ET.SubElement(child,"Prompt")
subChild6 = ET.SubElement(subChild5,"PromptText")
subChild6.text = "Enter client name"
try:
tree.write(xml_file)
except Exception, inst:
print "Unexpected error opening %s: %s" % (xml_file, inst)
return
if __name__ == "__main__":
main()