| 123456789101112131415161718192021222324252627282930313233 |
- # coding: utf-8
- """
- Fixer for the cmp() function on Py2, which was removed in Py3.
- Adds this import line::
- from past.builtins import cmp
- if cmp() is called in the code.
- """
- from __future__ import unicode_literals
- from lib2to3 import fixer_base
- from libfuturize.fixer_util import touch_import_top
- expression = "name='cmp'"
- class FixCmp(fixer_base.BaseFix):
- BM_compatible = True
- run_order = 9
- PATTERN = """
- power<
- ({0}) trailer< '(' args=[any] ')' >
- rest=any* >
- """.format(expression)
- def transform(self, node, results):
- name = results["name"]
- touch_import_top(u'past.builtins', name.value, node)
|
