"""Fixer that turns 1L into 1, 0755 into 0o755.

"""

# Copyright 2007 Georg Brandl.

# Licensed to PSF under a Contributor Agreement.

# Local imports

from ..pgen2 import token

from .. import fixer_base

from ..fixer_util import Number

class FixNumliterals(fixer_base.BaseFix):

    # This is so simple that we don't need the pattern compiler.

    def match(self, node):

        # Override

        return (node.type == token.NUMBER and

                (node.value.startswith("0") or node.value[-1] in "Ll"))

    def transform(self, node, results):

        val = node.value

        if val[-1] in 'Ll':

            val = val[:-1]

        elif val.startswith('0') and val.isdigit() and len(set(val)) > 1:

            val = "0o" + val[1:]

        return Number(val, prefix=node.get_prefix())