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

add a encoding optional for read python code

Browse source
This commit is contained in:
xxc 2018-12-12 15:48:38 +08:00 • committed by GitHub
parent c65fcd3fc5
commit 1622fbe28e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
pre_commit_hooks/check_docstring_first.py
View file

@ -53,12 +53,13 @@ def check_docstring_first(src, filename='<unknown>'):
def main(argv=None): def main(argv=None):
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument('filenames', nargs='*') parser.add_argument('filenames', nargs='*')
parser.add_argument('--encoding', default='UTF-8')
args = parser.parse_args(argv) args = parser.parse_args(argv)
retv = 0 retv = 0
for filename in args.filenames: for filename in args.filenames:
with io.open(filename, encoding='UTF-8') as f: with io.open(filename, encoding=args.encoding) as f:
contents = f.read() contents = f.read()
retv |= check_docstring_first(contents, filename=filename) retv |= check_docstring_first(contents, filename=filename)