orbit-oss/flask
4
5
Fork 2
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Change wrapper logic for testing

Browse source
This commit is contained in:
Keyan Pishdadian 2015-02-11 14:53:39 -05:00
parent f0ba70cda4
commit 77668e1d65
1 changed files with 3 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

7
scripts/flaskext_migrate.py
View file

@ -83,10 +83,9 @@ def fix_standard_imports(red):
return red return red
def fix(input_file): def fix(ast):
ast = read_source(input_file) """Wrapper which allows for testing when not running from shell"""
new_ast = fix_imports(ast) return fix_imports(ast).dumps()
write_source(new_ast, input_file)
if __name__ == "__main__": if __name__ == "__main__":
input_file = sys.argv[1] input_file = sys.argv[1]