create a code generator using python