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

Add wrapper for testing fixing functionality

Browse source
This commit is contained in:
Keyan Pishdadian 2015-02-11 14:41:48 -05:00
parent c90c4e5e8c
commit 4a9e34a574
1 changed files with 5 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

7
scripts/flaskext_migrate.py
View file

@ -83,8 +83,11 @@ def fix_standard_imports(red):
return red
if __name__ == "__main__":
input_file = sys.argv[1]
def fix(input_file):
ast = read_source(input_file)
new_ast = fix_imports(ast)
write_source(new_ast, input_file)
if __name__ == "__main__":
input_file = sys.argv[1]
fix(input_file)