mxformat.py

#!/usr/bin/env python
'''
Reformat a folder of MaterialX documents in place, optionally upgrading
the documents to the latest version of the standard.
'''

import argparse
import os
import xml.etree.ElementTree as ET

import MaterialX as mx

def is_well_formed(xml_string):
 error = ''
 try:
 ET.fromstring(xml_string) 
 except ET.ParseError as e:
 error = str(e)
 return error

def main():
 parser = argparse.ArgumentParser(description="Reformat a folder of MaterialX documents in place.")
 parser.add_argument('-y', '--yes', dest='yes', action="store_true", help="Proceed without asking for confirmation from the user.")
 parser.add_argument('-u', '--upgrade', dest='upgrade', action="store_true", help='Upgrade documents to the latest version of the standard.')
 parser.add_argument('-v', '--validate', dest='validate', action="store_true", help='Perform MaterialX validation on documents after reformatting.')
 parser.add_argument('-x', '--xml_syntax', dest='xml_syntax', action="store_true", help='Check XML syntax after reformatting.')
 parser.add_argument(dest="inputFolder", help="An input folder to scan for MaterialX documents.")
 opts = parser.parse_args()

 validDocs = dict()
 for root, dirs, files in os.walk(opts.inputFolder):
 for filename in files:
 fullpath = os.path.join(root, filename)
 if fullpath.endswith('.mtlx'):
 doc = mx.createDocument()
 try:
 readOptions = mx.XmlReadOptions()
 readOptions.readComments = True
 readOptions.readNewlines = True 
 readOptions.upgradeVersion = opts.upgrade
 try:
 mx.readFromXmlFile(doc, fullpath, mx.FileSearchPath(), readOptions)
 except Exception as err:
 print('Skipping "' + filename + '" due to exception: ' + str(err))
 continue
 validDocs[fullpath] = doc
 except mx.Exception:
 pass

 if not validDocs:
 print('No MaterialX documents were found in "%s"' % (opts.inputFolder))
 return

 print('Found %s MaterialX files in "%s"' % (len(validDocs), opts.inputFolder))

 mxVersion = mx.getVersionIntegers()

 if not opts.yes:
 if opts.upgrade:
 question = 'Would you like to upgrade all %i documents to MaterialX v%i.%i in place (y/n)?' % (len(validDocs), mxVersion[0], mxVersion[1])
 else:
 question = 'Would you like to reformat all %i documents in place (y/n)?' % len(validDocs)
 answer = input(question)
 if answer != 'y' and answer != 'Y':
 return

 validate = opts.validate
 if validate:
 print(f'- Validate documents')
 xml_syntax = opts.xml_syntax
 if xml_syntax:
 print(f'- Check XML syntax')
 for (filename, doc) in validDocs.items():
 if xml_syntax:
 xml_string = mx.writeToXmlString(doc)
 errors = is_well_formed(xml_string)
 if errors:
 print(f'- Warning: Document {filename} is not well-formed XML: {errors}') 
 if validate:
 is_valid, errors = doc.validate()
 if not is_valid:
 print(f'- Warning: Document {filename} is invalid. Errors {errors}.')
 mx.writeToXmlFile(doc, filename)

 if opts.upgrade:
 print('Upgraded %i documents to MaterialX v%i.%i' % (len(validDocs), mxVersion[0], mxVersion[1]))
 else:
 print('Reformatted %i documents ' % len(validDocs))

if __name__ == '__main__':
 main()