home *** CD-ROM | disk | FTP | other *** search
- # 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
-