Use Enum to list line ending types in mixed_line_ending

This commit is contained in:
Morgan Courbet 2017-06-15 20:29:28 +02:00
parent 3d4fb41d8a
commit 16b7c7af5d
No known key found for this signature in database
GPG key ID: 467299D324A21B24

View file

@ -1,13 +1,25 @@
import argparse
import sys
from enum import Enum
class LineEnding(Enum):
CRLF = '\r\n', '\\r\\n', 'crlf'
LF = '\n', '\\n', 'lf'
def __init__(self, str, strPrint, optName):
self.str = str
self.strPrint = strPrint
self.optName = optName
def mixed_line_ending(argv=None):
parser = argparse.ArgumentParser()
parser.add_argument(
'-f',
'--fix',
choices=['auto', 'crlf', 'lf', 'no'],
choices=['auto', 'no'] + [m.optName for m in LineEnding],
default='auto',
help='Replace line ending with the specified. Default is "auto"',
nargs=1)