30 lines
841 B
Python
30 lines
841 B
Python
|
#!/usr/bin/env python
|
||
|
|
|
||
|
|
# Make a reST file compliant to our pre-commit hook.
|
||
|
|
# Currently just remove trailing whitespace.
|
||
|
|
|
||
|
|
from __future__ import with_statement
|
||
|
|
import sys, re, shutil
|
||
|
|
|
||
|
|
ws_re = re.compile(r'\s+(\r?\n)$')
|
||
|
|
|
||
|
|
def main(argv=sys.argv):
|
||
|
|
rv = 0
|
||
|
|
for filename in argv[1:]:
|
||
|
|
try:
|
||
|
|
with open(filename, 'rb') as f:
|
||
|
|
lines = f.readlines()
|
||
|
|
new_lines = [ws_re.sub(r'\1', line) for line in lines]
|
||
|
|
if new_lines != lines:
|
||
|
|
print 'Fixing %s...' % filename
|
||
|
|
shutil.copyfile(filename, filename + '.bak')
|
||
|
|
with open(filename, 'wb') as f:
|
||
|
|
f.writelines(new_lines)
|
||
|
|
except Exception, err:
|
||
|
|
print 'Cannot fix %s: %s' % (filename, err)
|
||
|
|
rv = 1
|
||
|
|
return rv
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
sys.exit(main())
|