mirror of
https://github.com/pre-commit/pre-commit.git
synced 2026-01-13 04:20:28 -06:00
88 lines
2.5 KiB
Python
88 lines
2.5 KiB
Python
from __future__ import print_function
|
|
from __future__ import unicode_literals
|
|
|
|
import argparse
|
|
import jsonschema
|
|
import jsonschema.exceptions
|
|
import os.path
|
|
import re
|
|
import sys
|
|
import yaml
|
|
|
|
from pre_commit.jsonschema_extensions import apply_defaults
|
|
|
|
|
|
def is_regex_valid(regex):
|
|
try:
|
|
re.compile(regex)
|
|
return True
|
|
except re.error:
|
|
return False
|
|
|
|
|
|
def get_validator(
|
|
json_schema,
|
|
exception_type,
|
|
additional_validation_strategy=lambda obj: None,
|
|
):
|
|
"""Returns a function which will validate a yaml file for correctness
|
|
|
|
Args:
|
|
json_schema - JSON schema to validate file with
|
|
exception_type - Error type to raise on failure
|
|
additional_validation_strategy - Strategy for additional validation of
|
|
the object read from the file. The function should either raise
|
|
exception_type on failure.
|
|
"""
|
|
def validate(filename, load_strategy=yaml.load):
|
|
if not os.path.exists(filename):
|
|
raise exception_type('File {0} does not exist'.format(filename))
|
|
|
|
file_contents = open(filename, 'r').read()
|
|
|
|
try:
|
|
obj = load_strategy(file_contents)
|
|
except Exception as e:
|
|
raise exception_type(
|
|
'Invalid yaml: {0}\n{1}'.format(os.path.relpath(filename), e),
|
|
)
|
|
|
|
try:
|
|
jsonschema.validate(obj, json_schema)
|
|
except jsonschema.exceptions.ValidationError as e:
|
|
raise exception_type(
|
|
'Invalid content: {0}\n{1}'.format(
|
|
os.path.relpath(filename), e
|
|
),
|
|
)
|
|
|
|
obj = apply_defaults(obj, json_schema)
|
|
|
|
additional_validation_strategy(obj)
|
|
|
|
return obj
|
|
|
|
return validate
|
|
|
|
|
|
def get_run_function(filenames_help, validate_strategy, exception_cls):
|
|
def run(argv=None):
|
|
argv = argv if argv is not None else sys.argv[1:]
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('filenames', nargs='*', help=filenames_help)
|
|
args = parser.parse_args(argv)
|
|
|
|
retval = 0
|
|
for filename in args.filenames:
|
|
try:
|
|
validate_strategy(filename)
|
|
except exception_cls as e:
|
|
print(e.args[0])
|
|
# If there was an inner exception, print the stringified
|
|
# version of that.
|
|
if len(e.args) > 1:
|
|
print(str(e.args[1]))
|
|
retval = 1
|
|
return retval
|
|
return run
|