Can it rewrite the python code at the function level?

[Jarvx]

It can be achieved while rewriting rules might need to be carefully crafted.