fix_ne.py 571 Bytes
Newer Older
1 2 3 4 5 6 7 8
# Copyright 2006 Google, Inc. All Rights Reserved.
# Licensed to PSF under a Contributor Agreement.

"""Fixer that turns <> into !=."""

# Local imports
from .. import pytree
from ..pgen2 import token
9
from .. import fixer_base
10 11


12
class FixNe(fixer_base.BaseFix):
13 14
    # This is so simple that we don't need the pattern compiler.

15 16
    _accept_type = token.NOTEQUAL

17 18
    def match(self, node):
        # Override
19
        return node.value == "<>"
20 21

    def transform(self, node, results):
22
        new = pytree.Leaf(token.NOTEQUAL, "!=", prefix=node.prefix)
23
        return new