mirror of
https://github.com/pre-commit/pre-commit-hooks.git
synced 2026-04-06 03:56:54 +00:00
Adds fix-yaml hook
This adds missing --- header to yaml file, which is required for strict linting via yamllint. Future versions may add other fixes for yaml files. Fixes: #348
This commit is contained in:
parent
c8bad492e1
commit
2a92b74ea1
3 changed files with 59 additions and 0 deletions
34
pre_commit_hooks/fix_yaml.py
Normal file
34
pre_commit_hooks/fix_yaml.py
Normal file
|
|
@ -0,0 +1,34 @@
|
|||
from __future__ import absolute_import
|
||||
from __future__ import print_function
|
||||
from __future__ import unicode_literals
|
||||
|
||||
import argparse
|
||||
import io
|
||||
|
||||
from six import text_type
|
||||
|
||||
|
||||
def main(argv=None):
|
||||
parser = argparse.ArgumentParser()
|
||||
parser.add_argument('filenames', nargs='*', help='Filenames to fix')
|
||||
args = parser.parse_args(argv)
|
||||
|
||||
retv = 0
|
||||
|
||||
for filename in args.filenames:
|
||||
with io.open(filename, 'r+', encoding="utf-8") as f:
|
||||
line = f.readline().strip()
|
||||
if line != text_type('---'):
|
||||
print("[%s]" % line)
|
||||
f.seek(0, 0)
|
||||
content = f.read()
|
||||
f.seek(0, 0)
|
||||
f.write('---\n' + content)
|
||||
print('{}: Added --- header to YAML file'.format(filename))
|
||||
retv = 1
|
||||
|
||||
return retv
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
exit(main())
|
||||
Loading…
Add table
Add a link
Reference in a new issue