117 lines
4.2 KiB
Python
117 lines
4.2 KiB
Python
#!/usr/bin/env python3
|
|
"""Валидация примеров контракта распознавания схемами JSON Schema 2020-12.
|
|
|
|
Запуск из любого cwd: python scripts/validate_contracts.py
|
|
Зависимости: стандартная библиотека + jsonschema.
|
|
Код возврата: 0 — всё валидно, 1 — любая ошибка.
|
|
"""
|
|
import datetime
|
|
import json
|
|
import re
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
from jsonschema import Draft202012Validator, FormatChecker
|
|
|
|
ROOT = Path(__file__).resolve().parent.parent
|
|
CONTRACTS = ROOT / "contracts"
|
|
EXAMPLES = CONTRACTS / "examples"
|
|
|
|
SCHEMAS = {
|
|
"request": CONTRACTS / "recognition-request.schema.json",
|
|
"message": CONTRACTS / "recognition-result-message.schema.json",
|
|
"result": CONTRACTS / "recognition-result.schema.json",
|
|
}
|
|
|
|
# Порядок важен: result-message-* проверяется раньше result-*.
|
|
PATTERNS = [
|
|
("request-", "request"),
|
|
("result-message-", "message"),
|
|
("result-", "result"),
|
|
]
|
|
|
|
_DATE_TIME_RE = re.compile(r"^\d{4}-\d{2}-\d{2}[Tt]\d{2}:\d{2}:\d{2}(\.\d+)?([Zz]|[+-]\d{2}:\d{2})$")
|
|
|
|
|
|
def _check_date_time(value):
|
|
# Собственная проверка вместо rfc3339-validator: нужна и дата, и время, и часовой пояс.
|
|
if not isinstance(value, str):
|
|
return True
|
|
if not _DATE_TIME_RE.match(value):
|
|
return False
|
|
datetime.datetime.fromisoformat(value.replace("z", "Z")) # ValueError -> формат не прошёл
|
|
return True
|
|
|
|
|
|
def make_format_checker():
|
|
checker = FormatChecker()
|
|
checker.checks("date-time", raises=ValueError)(_check_date_time)
|
|
return checker
|
|
|
|
|
|
def short_path(error):
|
|
path = "/".join(str(p) for p in error.absolute_path)
|
|
return f"{path or '<root>'}: {error.message[:120]}"
|
|
|
|
|
|
def main():
|
|
failed = False
|
|
validators = {}
|
|
format_checker = make_format_checker()
|
|
|
|
for kind, path in SCHEMAS.items():
|
|
if not path.is_file():
|
|
print(f"FAIL схема {path.name}: файл не найден")
|
|
failed = True
|
|
continue
|
|
try:
|
|
schema = json.loads(path.read_text(encoding="utf-8"))
|
|
Draft202012Validator.check_schema(schema)
|
|
except Exception as exc: # битый JSON или нарушение метасхемы
|
|
print(f"FAIL схема {path.name}: {str(exc).splitlines()[0][:150]}")
|
|
failed = True
|
|
continue
|
|
validators[kind] = Draft202012Validator(schema, format_checker=format_checker)
|
|
print(f"OK схема {path.name}")
|
|
|
|
counts = {kind: 0 for kind in SCHEMAS}
|
|
for example in sorted(EXAMPLES.glob("*.json")):
|
|
kind = next((k for prefix, k in PATTERNS if example.name.startswith(prefix)), None)
|
|
if kind is None:
|
|
print(f"FAIL {example.name}: имя не подходит ни под один шаблон (request-*, result-message-*, result-*)")
|
|
failed = True
|
|
continue
|
|
counts[kind] += 1
|
|
validator = validators.get(kind)
|
|
if validator is None:
|
|
print(f"FAIL {example.name} [{SCHEMAS[kind].name}]: схема недоступна")
|
|
failed = True
|
|
continue
|
|
try:
|
|
instance = json.loads(example.read_text(encoding="utf-8"))
|
|
except ValueError as exc:
|
|
print(f"FAIL {example.name} [{SCHEMAS[kind].name}]: не JSON: {exc}")
|
|
failed = True
|
|
continue
|
|
errors = sorted(validator.iter_errors(instance), key=lambda e: list(e.absolute_path))
|
|
if errors:
|
|
failed = True
|
|
print(f"FAIL {example.name} [{SCHEMAS[kind].name}]")
|
|
for error in errors:
|
|
print(f" {short_path(error)}")
|
|
else:
|
|
print(f"OK {example.name} [{SCHEMAS[kind].name}]")
|
|
|
|
for kind, count in counts.items():
|
|
if count == 0:
|
|
print(f"FAIL нет ни одного примера для схемы {SCHEMAS[kind].name}")
|
|
failed = True
|
|
|
|
total = sum(counts.values())
|
|
print(f"Итого: примеров {total}, {'есть ошибки' if failed else 'всё валидно'}")
|
|
return 1 if failed else 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(main())
|