leave_AllBinaryOperators? match multiple classes

[nschloe]

I would like to control the whitespace around all binary operators. So far, I got

    def leave_Add(self, orig_node, updated_node):
        return updated_node.with_changes(whitespace_before=s0, whitespace_after=s0)

    def leave_Subtract(self, orig_node, updated_node):
        return updated_node.with_changes(whitespace_before=s0, whitespace_after=s0)

    def leave_Multiply(self, orig_node, updated_node):
        return updated_node.with_changes(whitespace_before=s0, whitespace_after=s0)

    def leave_Divide(self, orig_node, updated_node):
        return updated_node.with_changes(whitespace_before=s0, whitespace_after=s0)

    def leave_Modulo(self, orig_node, updated_node):
        return updated_node.with_changes(whitespace_before=s0, whitespace_after=s0)

    def leave_Power(self, orig_node, updated_node):
        return updated_node.with_changes(whitespace_before=s0, whitespace_after=s0)

    def leave_MatrixMultiply(self, orig_node, updated_node):
        return updated_node.with_changes(whitespace_before=s0, whitespace_after=s0)

which works well. It seems a little repetetive though, and perhaps I forgot an operation. Is there a way to shorten this?

[nschloe]

I found

class WhitespaceRemover2(m.MatcherDecoratableTransformer):
    @m.leave(
        m.Add()
        | m.Subtract()
        | m.Multiply()
        | m.Divide()
        | m.Modulo()
        | m.MatrixMultiply()
        | m.Power()
    )
    def rm_space_around_operators(self, orig_node, updated_node):
        return updated_node.with_changes(
            whitespace_before=s0,
            whitespace_after=s0,
        )

Perhaps this can be improved further.