| # 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 |
