mirror of
https://github.com/pre-commit/pre-commit-hooks.git
synced 2025-08-14 17:31:28 +08:00
41 lines
1.0 KiB
Python
41 lines
1.0 KiB
Python
from __future__ import print_function
|
|
|
|
import argparse
|
|
import os.path
|
|
import re
|
|
import sys
|
|
from typing import Optional
|
|
from typing import Sequence
|
|
|
|
|
|
def main(argv=None): # type: (Optional[Sequence[str]]) -> int
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('filenames', nargs='*')
|
|
parser.add_argument(
|
|
'--django', default=False, action='store_true',
|
|
help='Use Django-style test naming pattern (test*.py)',
|
|
)
|
|
args = parser.parse_args(argv)
|
|
|
|
retcode = 0
|
|
test_name_pattern = 'test.*.py' if args.django else '.*_test.py'
|
|
for filename in args.filenames:
|
|
base = os.path.basename(filename)
|
|
if (
|
|
not re.match(test_name_pattern, base) and
|
|
not base == '__init__.py' and
|
|
not base == 'conftest.py'
|
|
):
|
|
retcode = 1
|
|
print(
|
|
'{} does not match pattern "{}"'.format(
|
|
filename, test_name_pattern,
|
|
),
|
|
)
|
|
|
|
return retcode
|
|
|
|
|
|
if __name__ == '__main__':
|
|
sys.exit(main())
|