summaryrefslogtreecommitdiffstats
path: root/Lib/lib2to3/fixes/fix_ne.py
blob: ecd18e098d1a9cc95074a0ed4470e5d21382c5c9 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
# 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
from . import basefix


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

    def match(self, node):
        # Override
        return node.type == token.NOTEQUAL and node.value == "<>"

    def transform(self, node, results):
      new = pytree.Leaf(token.NOTEQUAL, "!=")
      new.set_prefix(node.get_prefix())
      return new