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

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