21 lines
594 B
Python
21 lines
594 B
Python
|
from lib2to3.fixer_base import BaseFix
|
||
|
|
from lib2to3.fixer_util import Name, syms
|
||
|
|
|
||
|
|
|
||
|
|
class FixExtImport(BaseFix):
|
||
|
|
|
||
|
|
PATTERN = "fixnode='oldname'"
|
||
|
|
|
||
|
|
def transform(self, node, results):
|
||
|
|
fixnode = results['fixnode']
|
||
|
|
fixnode.replace(Name('newname', prefix=fixnode.prefix))
|
||
|
|
|
||
|
|
if node.type == syms.import_from and \
|
||
|
|
getattr(results['imp'], 'value', None) == 'flask.ext':
|
||
|
|
return 0
|
||
|
|
# TODO: Case 2
|
||
|
|
|
||
|
|
|
||
|
|
# CASE 1 - from flask.ext.foo import bam --> from flask_foo import bam
|
||
|
|
# CASE 2 - from flask.ext import foo --> import flask_foo as foo
|