Add tests, remove manual testing file #1135

This commit is contained in:
Keyan Pishdadian 2015-02-11 15:06:53 -05:00
parent 96d81c297a
commit 0cf5881312
2 changed files with 40 additions and 34 deletions

View file

@ -1,34 +0,0 @@
from flask.ext.foo import \
bam, \
crackle
from flask.ext import foo
from flask.ext.foo import (bam,
a,
b
)
import flask.ext.foo
from flask.ext import foo
import sys
def migrate(old_file):
new_file = open("temp.py", "w")
for line in old_file:
if line[0, 15] is "from flask.ext":
if line[15] == '.':
import_statement = line[16::].split(' ')
extension = import_statement[0]
line = line. replace("flask.ext." + extension,
"flask_" + extension)
else:
pass
new_file.write(line)
if __name__ == "__main__":
old_file = open(sys.arv[1])