Add autopep8-wrapper hook.

This commit is contained in:
Anthony Sottile 2014-08-22 11:00:43 -07:00
parent 0121199c0a
commit 86b1c9da8e
5 changed files with 62 additions and 0 deletions

View file

@ -0,0 +1,28 @@
from __future__ import absolute_import
from __future__ import print_function
from __future__ import unicode_literals
import autopep8
import io
import sys
def main(argv=None):
argv = argv if argv is not None else sys.argv[1:]
args = autopep8.parse_args(argv)
retv = 0
for filename in args.files:
original_contents = io.open(filename).read()
new_contents = autopep8.fix_code(original_contents, args)
if original_contents != new_contents:
print('Fixing {0}'.format(filename))
retv = 1
with io.open(filename, 'w') as output_file:
output_file.write(new_contents)
return retv
if __name__ == '__main__':
exit(main())