mirror of
https://gitlab.gnome.org/GNOME/libxml2
synced 2025-03-28 21:33:13 +00:00
a62b31f43f
* In conda or Gentoo Prefix, we don't want to use the system python and instead rely on PATH lookup.
100 lines
3.3 KiB
Python
Executable File
100 lines
3.3 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
import sys, os
|
|
import libxml2
|
|
|
|
|
|
libxml2.debugMemory(1)
|
|
baseDir = os.path.join('msxsdtest', 'Particles')
|
|
filenames = os.listdir(baseDir)
|
|
mainXSD = str()
|
|
signature = str()
|
|
dictXSD = dict()
|
|
|
|
def gatherFiles():
|
|
for file in filenames:
|
|
if (file[-5] in ["a", "b", "c"]) and (file[-3:] == 'xsd'):
|
|
# newfilename = string.replace(filename, ' ', '_')
|
|
signature = file[:-5]
|
|
mainXSD = signature + ".xsd"
|
|
imports = []
|
|
for sub in filenames:
|
|
if (mainXSD != sub) and (sub[-3:] == 'xsd') and sub.startswith(signature):
|
|
imports.append(sub)
|
|
if len(imports) != 0:
|
|
dictXSD[mainXSD] = imports
|
|
|
|
def debugMsg(text):
|
|
#pass
|
|
print "DEBUG:", text
|
|
|
|
|
|
def fixup():
|
|
for mainXSD in dictXSD:
|
|
debugMsg("fixing '%s'..." % mainXSD)
|
|
schemaDoc = None
|
|
xpmainCtx = None
|
|
# Load the schema document.
|
|
schemaFile = os.path.join(baseDir, mainXSD)
|
|
schemaDoc = libxml2.parseFile(schemaFile)
|
|
if (schemaDoc is None):
|
|
print "ERROR: doc '%s' not found" % mainXSD
|
|
sys.exit(1)
|
|
try:
|
|
xpmainCtx = schemaDoc.xpathNewContext()
|
|
xpmainCtx.xpathRegisterNs("xs", "http://www.w3.org/2001/XMLSchema");
|
|
xpres = xpmainCtx.xpathEval("/xs:schema")
|
|
if len(xpres) == 0:
|
|
print "ERROR: doc '%s' has no <schema> element" % mainXSD
|
|
sys.exit(1)
|
|
schemaElem = xpres[0]
|
|
schemaNs = schemaElem.ns()
|
|
# Select all <import>s.
|
|
xpres = xpmainCtx.xpathEval("/xs:schema/xs:import")
|
|
if len(xpres) != 0:
|
|
for elem in xpres:
|
|
loc = elem.noNsProp("schemaLocation")
|
|
if (loc is not None):
|
|
debugMsg(" imports '%s'" % loc)
|
|
if loc in dictXSD[mainXSD]:
|
|
dictXSD[mainXSD].remove(loc)
|
|
for loc in dictXSD[mainXSD]:
|
|
# Read out the targetNamespace.
|
|
impTargetNs = None
|
|
impFile = os.path.join(baseDir, loc)
|
|
impDoc = libxml2.parseFile(impFile)
|
|
try:
|
|
xpimpCtx = impDoc.xpathNewContext()
|
|
try:
|
|
xpimpCtx.setContextDoc(impDoc)
|
|
xpimpCtx.xpathRegisterNs("xs", "http://www.w3.org/2001/XMLSchema");
|
|
xpres = xpimpCtx.xpathEval("/xs:schema")
|
|
impTargetNs = xpres[0].noNsProp("targetNamespace")
|
|
finally:
|
|
xpimpCtx.xpathFreeContext()
|
|
finally:
|
|
impDoc.freeDoc()
|
|
|
|
# Add the <import>.
|
|
debugMsg(" adding <import namespace='%s' schemaLocation='%s'/>" % (impTargetNs, loc))
|
|
newElem = schemaDoc.newDocNode(schemaNs, "import", None)
|
|
if (impTargetNs is not None):
|
|
newElem.newProp("namespace", impTargetNs)
|
|
newElem.newProp("schemaLocation", loc)
|
|
if schemaElem.children is not None:
|
|
schemaElem.children.addPrevSibling(newElem)
|
|
schemaDoc.saveFile(schemaFile)
|
|
finally:
|
|
xpmainCtx.xpathFreeContext()
|
|
schemaDoc.freeDoc()
|
|
|
|
try:
|
|
gatherFiles()
|
|
fixup()
|
|
finally:
|
|
libxml2.cleanupParser()
|
|
if libxml2.debugMemory(1) != 0:
|
|
print "Memory leak %d bytes" % (libxml2.debugMemory(1))
|
|
libxml2.dumpMemory()
|
|
|