Merge pull request #15 from pre-commit/cross_platform_trim_trailing_whitespace

Use fileinput instead of sed.
This commit is contained in:
Anthony Sottile 2014-12-24 05:00:30 -08:00
commit 2bd608119f

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