root
/
flask
mirror of https://github.com/pallets/flask.git
1
0
Fork 0
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Change wrapper logic for testing

This commit is contained in:
Keyan Pishdadian 2015-02-11 14:53:39 -05:00
parent 4a9e34a574
commit 96d81c297a
1 changed files with 3 additions and 4 deletions
Show Stats 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]