1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
|
#!/usr/bin/env python
import argparse
import tempfile
import shutil
import os
import glob
import subprocess
import shlex
def get_files(xsd_dir):
print("Searching ", xsd_dir)
files = []
for path, subdirs, fns in os.walk(xsd_dir):
for fn in fns:
if fn.endswith('xsd'):
files.append(os.path.join(path, fn))
return files
def copy_xsds(xsd, dest):
"""Go ahead and copy all xsds, there will likely be many dependencies. If
not, xsds are small"""
files = get_files(os.path.dirname(xsd))
print("Copying:")
for f in files:
print(f)
for fn in files:
shutil.copy(fn, dest)
def generate_pyxb(xsd, outdir, modname):
cmd = "pyxbgen -u {x} -m {m} --binding-root {p}".format(x=xsd, m=modname,
p=outdir)
print(cmd)
subprocess.call(shlex.split(cmd))
if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument("xsd", type=str,
help="the XSD of interest")
parser.add_argument("outdir", type=str)
parser.add_argument("--mod-name", type=str,
default='DataSetXsd')
args = parser.parse_args()
tempd = tempfile.mkdtemp(suffix='xsds')
copy_xsds(args.xsd, tempd)
xsd_name = os.path.basename(args.xsd)
generate_pyxb(os.path.join(tempd, xsd_name), args.outdir, args.mod_name)
shutil.rmtree(tempd)
|
