From f0972628ef14e46c14a610331320c5454e30e2ef Mon Sep 17 00:00:00 2001 From: Max Shytikov Date: Tue, 22 Apr 2014 03:09:13 +0200 Subject: [PATCH] Fixed migration for one file --- vcr/migration.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/vcr/migration.py b/vcr/migration.py index 90a2da9..d3e909a 100644 --- a/vcr/migration.py +++ b/vcr/migration.py @@ -92,9 +92,12 @@ def main(): path = sys.argv[1] if not os.path.isabs(path): path = os.path.abspath(path) - for root, dirs, files in os.walk(path): - for file_name in files: - file_path = os.path.join(root, file_name) + files = [path] + if os.path.isdir(path): + files = (os.path.join(root, name) + for (root, dirs, files) in os.walk(path) + for name in files) + for file_path in files: migrated = try_migrate(file_path) status = 'OK' if migrated else 'FAIL' sys.stderr.write("[{}] {}\n".format(status, file_path))