Is this not somewhere where you'd use traits? Make your generated code a trait and just mix it into your normal class...