| Kasimier T. Buchcik | 755307a | 2005-04-01 15:55:45 +0000 | [diff] [blame] | 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 | |