/Lib/lib2to3/fixes/fix_types.py
http://unladen-swallow.googlecode.com/ · Python · 62 lines · 45 code · 1 blank · 16 comment · 0 complexity · 08728aeba77665139ce3f967cb24c2f1 MD5 · raw file
- # Copyright 2007 Google, Inc. All Rights Reserved.
- # Licensed to PSF under a Contributor Agreement.
- """Fixer for removing uses of the types module.
- These work for only the known names in the types module. The forms above
- can include types. or not. ie, It is assumed the module is imported either as:
- import types
- from types import ... # either * or specific types
- The import statements are not modified.
- There should be another fixer that handles at least the following constants:
- type([]) -> list
- type(()) -> tuple
- type('') -> str
- """
- # Local imports
- from ..pgen2 import token
- from .. import fixer_base
- from ..fixer_util import Name
- _TYPE_MAPPING = {
- 'BooleanType' : 'bool',
- 'BufferType' : 'memoryview',
- 'ClassType' : 'type',
- 'ComplexType' : 'complex',
- 'DictType': 'dict',
- 'DictionaryType' : 'dict',
- 'EllipsisType' : 'type(Ellipsis)',
- #'FileType' : 'io.IOBase',
- 'FloatType': 'float',
- 'IntType': 'int',
- 'ListType': 'list',
- 'LongType': 'int',
- 'ObjectType' : 'object',
- 'NoneType': 'type(None)',
- 'NotImplementedType' : 'type(NotImplemented)',
- 'SliceType' : 'slice',
- 'StringType': 'bytes', # XXX ?
- 'StringTypes' : 'str', # XXX ?
- 'TupleType': 'tuple',
- 'TypeType' : 'type',
- 'UnicodeType': 'str',
- 'XRangeType' : 'range',
- }
- _pats = ["power< 'types' trailer< '.' name='%s' > >" % t for t in _TYPE_MAPPING]
- class FixTypes(fixer_base.BaseFix):
- PATTERN = '|'.join(_pats)
- def transform(self, node, results):
- new_value = _TYPE_MAPPING.get(results["name"].value)
- if new_value:
- return Name(new_value, prefix=node.get_prefix())
- return None