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

Use fileinput instead of sed.

Browse Source
This commit is contained in:
Anthony Sottile
2014-12-23 12:04:19 -08:00
parent 25e50df1e5
commit e3312c773e
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
pre_commit_hooks/trailing_whitespace_fixer.py
View File

@ -1,12 +1,18 @@
from __future__ import print_function
import argparse
import fileinput
import sys
from plumbum import local
from pre_commit_hooks.util import entry
def _fix_file(filename):
for line in fileinput.input([filename], inplace=True):
print(line.rstrip())
@entry
def fix_trailing_whitespace(argv):
parser = argparse.ArgumentParser()
@ -20,7 +26,7 @@ def fix_trailing_whitespace(argv):
if bad_whitespace_files:
for bad_whitespace_file in bad_whitespace_files:
print('Fixing {0}'.format(bad_whitespace_file))
local['sed']['-i', '-e', 's/[[:space:]]*$//', bad_whitespace_file]()
_fix_file(bad_whitespace_file)
return 1
else:
return 0