1 #!/usr/bin/python -u 2 3 import sys, os 4 import libxml2 5 6 7 libxml2.debugMemory(1) 8 baseDir = os.path.join('msxsdtest', 'Particles') 9 filenames = os.listdir(baseDir) 10 mainXSD = str() 11 signature = str() 12 dictXSD = dict() 13 14 def gatherFiles(): 15 for file in filenames: 16 if (file[-5] in ["a", "b", "c"]) and (file[-3:] == 'xsd'): 17 # newfilename = string.replace(filename, ' ', '_') 18 signature = file[:-5] 19 mainXSD = signature + ".xsd" 20 imports = [] 21 for sub in filenames: 22 if (mainXSD != sub) and (sub[-3:] == 'xsd') and sub.startswith(signature): 23 imports.append(sub) 24 if len(imports) != 0: 25 dictXSD[mainXSD] = imports 26 27 def debugMsg(text): 28 #pass 29 print "DEBUG:", text 30 31 32 def fixup(): 33 for mainXSD in dictXSD: 34 debugMsg("fixing '%s'..." % mainXSD) 35 schemaDoc = None 36 xpmainCtx = None 37 # Load the schema document. 38 schemaFile = os.path.join(baseDir, mainXSD) 39 schemaDoc = libxml2.parseFile(schemaFile) 40 if (schemaDoc is None): 41 print "ERROR: doc '%s' not found" % mainXSD 42 sys.exit(1) 43 try: 44 xpmainCtx = schemaDoc.xpathNewContext() 45 xpmainCtx.xpathRegisterNs("xs", "http://www.w3.org/2001/XMLSchema"); 46 xpres = xpmainCtx.xpathEval("/xs:schema") 47 if len(xpres) == 0: 48 print "ERROR: doc '%s' has no <schema> element" % mainXSD 49 sys.exit(1) 50 schemaElem = xpres[0] 51 schemaNs = schemaElem.ns() 52 # Select all <import>s. 53 xpres = xpmainCtx.xpathEval("/xs:schema/xs:import") 54 if len(xpres) != 0: 55 for elem in xpres: 56 loc = elem.noNsProp("schemaLocation") 57 if (loc is not None): 58 debugMsg(" imports '%s'" % loc) 59 if loc in dictXSD[mainXSD]: 60 dictXSD[mainXSD].remove(loc) 61 for loc in dictXSD[mainXSD]: 62 # Read out the targetNamespace. 63 impTargetNs = None 64 impFile = os.path.join(baseDir, loc) 65 impDoc = libxml2.parseFile(impFile) 66 try: 67 xpimpCtx = impDoc.xpathNewContext() 68 try: 69 xpimpCtx.setContextDoc(impDoc) 70 xpimpCtx.xpathRegisterNs("xs", "http://www.w3.org/2001/XMLSchema"); 71 xpres = xpimpCtx.xpathEval("/xs:schema") 72 impTargetNs = xpres[0].noNsProp("targetNamespace") 73 finally: 74 xpimpCtx.xpathFreeContext() 75 finally: 76 impDoc.freeDoc() 77 78 # Add the <import>. 79 debugMsg(" adding <import namespace='%s' schemaLocation='%s'/>" % (impTargetNs, loc)) 80 newElem = schemaDoc.newDocNode(schemaNs, "import", None) 81 if (impTargetNs is not None): 82 newElem.newProp("namespace", impTargetNs) 83 newElem.newProp("schemaLocation", loc) 84 if schemaElem.children is not None: 85 schemaElem.children.addPrevSibling(newElem) 86 schemaDoc.saveFile(schemaFile) 87 finally: 88 xpmainCtx.xpathFreeContext() 89 schemaDoc.freeDoc() 90 91 try: 92 gatherFiles() 93 fixup() 94 finally: 95 libxml2.cleanupParser() 96 if libxml2.debugMemory(1) != 0: 97 print "Memory leak %d bytes" % (libxml2.debugMemory(1)) 98 libxml2.dumpMemory() 99 100
