I recently added list comprehensions etc for python #24 which works nice. I also wanted to add support to turn eg
to
or
but this does not directly work in the current framework since the nodes we want to put on separate lines here are not the children of a single parent, they are rather in a tree of binary operations. Also the whitespace before/after the operator is tricky. Maybe nvim-treesitter/nvim-treesitter-textobjects#235 could help for the whitespace at least.