36 lines
937 B
Python
36 lines
937 B
Python
# Copyright 2006 Google, Inc. All Rights Reserved.
|
|
# Licensed to PSF under a Contributor Agreement.
|
|
|
|
"""Fixer that turns 'long' into 'int' everywhere.
|
|
|
|
This also strips the trailing 'L' or 'l' from long loterals.
|
|
"""
|
|
|
|
# Local imports
|
|
from .. import pytree
|
|
from .. import fixer_base
|
|
from ..fixer_util import Name, Number
|
|
|
|
|
|
class FixLong(fixer_base.BaseFix):
|
|
|
|
PATTERN = """
|
|
(long_type = 'long' | number = NUMBER)
|
|
"""
|
|
|
|
static_long = Name("long")
|
|
static_int = Name("int")
|
|
|
|
def transform(self, node, results):
|
|
long_type = results.get("long_type")
|
|
number = results.get("number")
|
|
new = None
|
|
if long_type:
|
|
assert node == self.static_long, node
|
|
new = self.static_int.clone()
|
|
if number and node.value[-1] in ("l", "L"):
|
|
new = Number(node.value[:-1])
|
|
if new is not None:
|
|
new.set_prefix(node.get_prefix())
|
|
return new
|