From b2b0d5929ac65f96b19b68579a46b8925c845fd2 Mon Sep 17 00:00:00 2001 From: Morgan Courbet Date: Mon, 26 Jun 2017 21:31:44 +0200 Subject: [PATCH] Use enum instead of raw argparse result --- pre_commit_hooks/mixed_line_ending.py | 18 +++++++++++++++--- 1 file changed, 15 insertions(+), 3 deletions(-) diff --git a/pre_commit_hooks/mixed_line_ending.py b/pre_commit_hooks/mixed_line_ending.py index 323708b..6de9a4e 100644 --- a/pre_commit_hooks/mixed_line_ending.py +++ b/pre_commit_hooks/mixed_line_ending.py @@ -27,9 +27,9 @@ class MixedLineEndingOption(CLIOption): def mixed_line_ending(argv=None): - args = _parse_arguments(argv) + options = _parse_arguments(argv) - print(args.fix) + print(options) return 0 @@ -49,7 +49,19 @@ def _parse_arguments(argv=None): help='Increase output verbosity') args = parser.parse_args(argv) - return args + fix = None + if args.fix == 'auto': + fix = MixedLineEndingOption.AUTO + elif args.fix == 'no': + fix = MixedLineEndingOption.NO + elif args.fix == 'crlf': + fix = MixedLineEndingOption.CRLF + elif args.fix == 'lf': + fix = MixedLineEndingOption.LF + + options = {'fix': fix, 'verbose': args.verbose} + + return options if __name__ == '__main__':