From 37217e85b80c389208cd3396c2cdc4b89bd46bcf Mon Sep 17 00:00:00 2001 From: Keyan Pishdadian Date: Wed, 11 Feb 2015 15:22:07 -0500 Subject: [PATCH] Fix issue with wrapper logic --- scripts/flaskext_migrate.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/scripts/flaskext_migrate.py b/scripts/flaskext_migrate.py index de7290e4..e23b732b 100644 --- a/scripts/flaskext_migrate.py +++ b/scripts/flaskext_migrate.py @@ -89,4 +89,6 @@ def fix(ast): if __name__ == "__main__": input_file = sys.argv[1] - fix(input_file) + ast = read_source(input_file) + ast = fix_imports(ast) + write_source(ast, input_file)