Staging
v0.5.1
https://github.com/python/cpython
Raw File
Tip revision: 65b2eb9dfebe873e8ee5821f13d5a8c07d9c89c2 authored by Georg Brandl on 05 December 2010, 11:42:38 UTC
Fix title.
Tip revision: 65b2eb9
fix_future.py
"""Remove __future__ imports

from __future__ import foo is replaced with an empty line.
"""
# Author: Christian Heimes

# Local imports
from .. import fixer_base
from ..fixer_util import BlankLine

class FixFuture(fixer_base.BaseFix):
    BM_compatible = True

    PATTERN = """import_from< 'from' module_name="__future__" 'import' any >"""

    # This should be run last -- some things check for the import
    run_order = 10

    def transform(self, node, results):
        new = BlankLine()
        new.prefix = node.prefix
        return new
back to top