flask/scripts/flaskext_migrate.py
Keyan Pishdadian 0b2556f3ec Add lib2to3 fixer base code
Switching to using RedBaron instead of lib2to3, committing to save lib2to3 code in case using Baron is not successful.
2015-02-10 11:23:16 -05:00

14 lines
362 B
Python

# 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
from redbaron import RedBaron
import sys
with open("test.py", "r") as source_code:
red = RedBaron(source_code.read())
print red.dumps()
# with open("code.py", "w") as source_code:
# source_code.write(red.dumps())