pre-commit/pre-commit-hooks
1
0
Fork 0
mirror of https://github.com/pre-commit/pre-commit-hooks.git synced 2026-04-06 20:16:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add line ending conversion

Browse source
This commit is contained in:
Morgan Courbet 2017-06-28 22:15:31 +02:00
parent 466f9e1732
commit aaf134c2bc
No known key found for this signature in database
GPG key ID: 467299D324A21B24
1 changed files with 38 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

49
pre_commit_hooks/mixed_line_ending.py
View file

@ -12,14 +12,15 @@ class CLIOption(Enum):
class LineEnding(CLIOption): class LineEnding(CLIOption):
CR = '\r', '\\r', 'cr' CR = '\r', '\\r', 'cr', re.compile(r'\r', re.DOTALL)
CRLF = '\r\n', '\\r\\n', 'crlf' CRLF = '\r\n', '\\r\\n', 'crlf', re.compile(r'\r\n', re.DOTALL)
LF = '\n', '\\n', 'lf' LF = '\n', '\\n', 'lf', re.compile(r'(?<!\r)\n', re.DOTALL)
def __init__(self, string, strPrint, optName): def __init__(self, string, strPrint, optName, regex):
self.string = string self.string = string
self.strPrint = strPrint self.strPrint = strPrint
self.optName = optName self.optName = optName
self.regex = regex
class MixedLineEndingOption(CLIOption): class MixedLineEndingOption(CLIOption):
@ -40,10 +41,17 @@ class MixedLineDetection(Enum):
self.line_ending_char = line_ending_char self.line_ending_char = line_ending_char
# Matches CRLF ANY_LINE_ENDING_PATTERN = re.compile(
CRLF_PATTERN = re.compile(LineEnding.CRLF.string, re.DOTALL) # match either
# Matches LF (without preceding CR) '(' +
LF_PATTERN = re.compile('(?<!' + LineEnding.CR.strPrint + ')' + LineEnding.LF.strPrint, re.DOTALL) # \r\n
LineEnding.CRLF.regex.pattern +
# or \n
'|' + LineEnding.LF.regex.pattern +
# or \r
'|' + LineEnding.CR.regex.pattern +
')'
)
def mixed_line_ending(argv=None): def mixed_line_ending(argv=None):
@ -54,7 +62,10 @@ def mixed_line_ending(argv=None):
_check_filenames(options['filenames']) _check_filenames(options['filenames'])
for filename in options['filenames']: for filename in options['filenames']:
print(_detect_line_ending(filename)) detect_result = _detect_line_ending(filename)
if detect_result.conversion:
_convert_line_ending(filename, detect_result.line_ending_char)
return 0 return 0
@ -99,9 +110,10 @@ def _check_filenames(filenames):
def _detect_line_ending(filename): def _detect_line_ending(filename):
f = open(filename, 'r') f = open(filename, 'r')
buf = f.read() buf = f.read()
f.close()
crlf_nb = len(CRLF_PATTERN.findall(buf)) crlf_nb = len(LineEnding.CRLF.regex.findall(buf))
lf_nb = len(LF_PATTERN.findall(buf)) lf_nb = len(LineEnding.LF.regex.findall(buf))
crlf_found = crlf_nb > 0 crlf_found = crlf_nb > 0
lf_found = lf_nb > 0 lf_found = lf_nb > 0
@ -118,5 +130,20 @@ def _detect_line_ending(filename):
return MixedLineDetection.MIXED_MOSTLY_LF return MixedLineDetection.MIXED_MOSTLY_LF
def _convert_line_ending(filename, line_ending):
# read the file
fin = open(filename, 'r')
bufin = fin.read()
fin.close()
# convert line ending
bufout = ANY_LINE_ENDING_PATTERN.sub(line_ending, bufin)
# write the result in the file
fout = open(filename, 'w')
fout.write(bufout)
fout.close()
if __name__ == '__main__': if __name__ == '__main__':
sys.exit(mixed_line_ending()) sys.exit(mixed_line_ending())