third-party/pre-commit
1
0
Fork 0
mirror of https://github.com/pre-commit/pre-commit.git synced 2026-02-17 16:24:40 +04:00
Code Issues Projects Releases Packages Wiki Activity Actions

Extend validate_base to take a load_strategy

Browse source
This commit is contained in:
Anthony Sottile 2014-03-23 20:19:09 -07:00
parent 2699026908
commit d8f8f5e1f4
3 changed files with 14 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
pre_commit/clientlib/validate_base.py
View file

@ -24,7 +24,7 @@ def get_validator(
exception_type on failure.
"""
def validate(filename=None):
def validate(filename=None, load_strategy=yaml.load):
filename = filename or os.path.join(git.get_root(), default_filename)
if not os.path.exists(filename):
@ -33,7 +33,7 @@ def get_validator(
file_contents = open(filename, 'r').read()
try:
obj = yaml.load(file_contents)
obj = load_strategy(file_contents)
except Exception as e:
raise exception_type(
'File {0} is not a valid yaml file'.format(filename), e,