From 4a9e34a57476c92a4147f9ecafc357a681f1a19a Mon Sep 17 00:00:00 2001 From: Keyan Pishdadian Date: Wed, 11 Feb 2015 14:41:48 -0500 Subject: [PATCH] Add wrapper for testing fixing functionality --- scripts/flaskext_migrate.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/scripts/flaskext_migrate.py b/scripts/flaskext_migrate.py index 53bc80e4..b7a1baf0 100644 --- a/scripts/flaskext_migrate.py +++ b/scripts/flaskext_migrate.py @@ -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)