Expose annotation processor, in Jython

expose_processor.py

import os
import os.path
import sys

from org.objectweb.asm import ClassWriter
from org.python.expose.generate import ExposedTypeProcessor


def generate(exposer):
    writer = ClassWriter(ClassWriter.COMPUTE_FRAMES)
    exposer.generate(writer)
    write(exposer.getClassName(), writer.toByteArray()) 


def write(class_name, bytecode):
    dest_dir = "test"
    dest = os.path.join(dest_dir, class_name.replace(".", "/") + ".class")
    try:
        os.makedirs(os.path.dirname(dest))
    except OSError:
        pass
    with open(dest, "wb") as out:
        out.write(bytecode)


def main():
    for path in sys.argv[1:]:
        with open(path) as f:
            etp = ExposedTypeProcessor(f)
            for exposer in etp.getMethodExposers():
                generate(exposer)
            for exposer in etp.getDescriptorExposers():
                generate(exposer)
            if etp.getNewExposer():
                generate(etp.getNewExposer())
            generate(etp.getTypeExposer())
            write(etp.getExposedClassName(), etp.getBytecode())


main()