pre-commit/pre-commit-hooks
1
0
Fork 0
mirror of https://github.com/pre-commit/pre-commit-hooks.git synced 2025-08-14 09:27:21 +08:00
Code Issues Packages Projects Releases Wiki Activity

trailing-whitespace hook: restoring original file in case of failure - fixes #134

Browse Source
This commit is contained in:
Lucas Cimon
2016-08-18 16:39:06 +02:00
parent 775a7906cd
commit bc5e7f2d72
2 changed files with 24 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
pre_commit_hooks/trailing_whitespace_fixer.py
View File

@ -9,7 +9,7 @@ from pre_commit_hooks.util import cmd_output
def _fix_file(filename, markdown=False):
for line in fileinput.input([filename], inplace=True):
for line in fileinput.input([filename], inplace=True, backup='.bak'):
# preserve trailing two-space for non-blank lines in markdown files
if markdown and (not line.isspace()) and (line.endswith(" \n")):
line = line.rstrip(' \n')
@ -64,14 +64,20 @@ def fix_trailing_whitespace(argv=None):
.format(ext)
)
if bad_whitespace_files:
for bad_whitespace_file in bad_whitespace_files:
print('Fixing {0}'.format(bad_whitespace_file))
_, extension = os.path.splitext(bad_whitespace_file.lower())
return_code = 0
for bad_whitespace_file in bad_whitespace_files:
print('Fixing {0}'.format(bad_whitespace_file))
_, extension = os.path.splitext(bad_whitespace_file.lower())
try:
_fix_file(bad_whitespace_file, all_markdown or extension in md_exts)
return 1
else:
return 0
return_code = 1
# pylint: disable=broad-except
except Exception as error: # pragma: no cover
# e.g. error can be a UnicodeDecodeError in Python 3
print('Ignoring {} that caused a {}'.format(bad_whitespace_file, error.__class__))
os.remove(bad_whitespace_file)
os.rename(bad_whitespace_file + '.bak', bad_whitespace_file)
return return_code
if __name__ == '__main__':