pre-commit-hooks/pre_commit_hooks/autopep8_wrapper.py
2017-06-08 08:44:22 +00:00

28 lines
720 B
Python

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