From 462217f9fca36771bd1da9c65db8b2888ba71a5e Mon Sep 17 00:00:00 2001 From: Dan Buch Date: Mon, 30 May 2016 10:46:00 -0400 Subject: [PATCH] Allow zero+ paths in migration script and run the self-tests on Windows, too. --- appveyor.yml | 1 + cli-v1-to-v2 | 10 +++++++--- 2 files changed, 8 insertions(+), 3 deletions(-) diff --git a/appveyor.yml b/appveyor.yml index 173086e..e93734d 100644 --- a/appveyor.yml +++ b/appveyor.yml @@ -23,3 +23,4 @@ build_script: - python runtests vet - python runtests test - python runtests gfmxr +- python cli-v1-to-v2 --selftest diff --git a/cli-v1-to-v2 b/cli-v1-to-v2 index 1ff4892..175afa7 100755 --- a/cli-v1-to-v2 +++ b/cli-v1-to-v2 @@ -36,7 +36,7 @@ def main(sysargs=sys.argv[:]): parser = argparse.ArgumentParser( description=_DESCRIPTION, formatter_class=argparse.ArgumentDefaultsHelpFormatter) - parser.add_argument('path', nargs='+', + parser.add_argument('path', nargs='*', type=os.path.abspath, default=os.getcwd()) parser.add_argument('-w', '--write', help='write changes back to file', action='store_true', default=False) @@ -60,7 +60,11 @@ def main(sysargs=sys.argv[:]): format='%(message)s' ) - for filepath in _find_candidate_files(args.path): + paths = args.path + if len(paths) == 0: + paths = ['.'] + + for filepath in _find_candidate_files(paths): updated_source = _update_filepath(filepath) if args.write: logging.info('Updating %s', filepath) @@ -86,7 +90,7 @@ def _find_candidate_files(paths): dirs.pop(i) for filename in files: - if not filename.endswith('.go'): + if not filename.decode('utf-8').endswith('.go'): continue filepath = os.path.join(curdir, filename)