add_MonoPInvokeCallback_attribute.py

import os
import sys

DECORATORS = ["// ATTRIBUTE AUTOMATICALLY ADDED.",
              "[MonoPInvokeCallback]"]

METHODS_TO_DECORATE = ["SetPendingApplicationException(",
                      "SetPendingArithmeticException(",
                      "SetPendingDivideByZeroException(",
                      "SetPendingIndexOutOfRangeException(",
                      "SetPendingInvalidCastException(",
                      "SetPendingInvalidOperationException(",
                      "SetPendingIOException(",
                      "SetPendingNullReferenceException(",
                      "SetPendingOutOfMemoryException(",
                      "SetPendingOverflowException(",
                      "SetPendingSystemException(",
                      "SetPendingArgumentException(",
                      "SetPendingArgumentNullException(",
                      "SetPendingArgumentOutOfRangeException(",
                      "EntryPoint=\"SWIGRegisterExceptionCallbacks_UsdCs",
                      "EntryPoint=\"SWIGRegisterExceptionArgumentCallbacks_UsdCs",
                      "CreateString("]

def decorate(line):
    leading_spaces = len(line) - len(line.lstrip(' '))
    decorated = ["{0}{1}".format(" "*leading_spaces, d) for d in DECORATORS]
    decorated.append(line)
    return "\n".join(decorated)

#TODO: add proper argument handling
def main():
    print("Adding MonoPInvokeCallback attribute to the appropriate methods ...\n")
    build_path = sys.argv[1]
    cwd = os.path.abspath(os.path.dirname(__file__))
    filein_path = os.path.join(cwd, build_path, "UsdCsPINVOKE.cs")
    fileout_path = os.path.join(cwd, build_path, "UsdCsPINVOKE.cs.out")
    prev_line = ""
    with open(filein_path, "r") as filein, open(fileout_path, "w") as fileout:
        for line in filein:
            if DECORATORS[1] not in prev_line and any(methods in line for methods in METHODS_TO_DECORATE):
                fileout.write(decorate(line))
            else:
                fileout.write(line)
            prev_line = line
    os.replace(fileout_path, filein_path)

if __name__ == "__main__":
    main()