how to create a code generator in python