Wrote something similar, but while you can mix generated and hand-written code, the actual python code for that is moved out of the file.

https://jwdevantier.github.io/gcgen/