| 1 | #!/usr/bin/env python3
|
|---|
| 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 |
|
|---|
