Comment by carlmr

I was going to suggest libCST, it works really well and is much less of a hassle to set up than this.

https://github.com/Instagram/LibCST