Spec document jsonschema dialect cache

Author: p1c2u

openapi_spec_validator/validation/keywords.py

@@ -244,10 +244,9 @@ def __call__(
 
 
 class OpenAPIV31SchemaValidator(SchemaValidator):
-    default_jsonschema_dialect_id = OAS31_BASE_DIALECT_URI
-
     def __init__(self, registry: "KeywordValidatorRegistry"):
         super().__init__(registry)
+        self._default_jsonschema_dialect_id: str | None = None
         self._validator_classes_by_dialect: dict[
             str, type[Validator] | None
         ] = {}
@@ -283,14 +282,18 @@ def _get_schema_dialect_id(
                     )
                 dialect_id = dialect_value
             else:
-                jsonschema_dialect_id = self._get_jsonschema_dialect_id(schema)
+                jsonschema_dialect_id = (
+                    self._get_default_jsonschema_dialect_id(schema)
+                )
                 schema_to_check = {
                     **schema_to_check,
                     "$schema": jsonschema_dialect_id,
                 }
                 dialect_id = jsonschema_dialect_id
         else:
-            jsonschema_dialect_id = self._get_jsonschema_dialect_id(schema)
+            jsonschema_dialect_id = self._get_default_jsonschema_dialect_id(
+                schema
+            )
             schema_to_check = schema_value
             dialect_id = jsonschema_dialect_id
 
@@ -312,14 +315,19 @@ def _get_validator_class_for_dialect(
         self._validator_classes_by_dialect[dialect_id] = validator_cls
         return validator_cls
 
-    def _get_jsonschema_dialect_id(self, schema: SchemaPath) -> str:
-        schema_root = self._get_schema_root(schema)
-        try:
-            return (schema_root // "jsonSchemaDialect").read_str()
-        except KeyError:
-            return self.default_jsonschema_dialect_id
+    def _get_default_jsonschema_dialect_id(self, schema: SchemaPath) -> str:
+        if self._default_jsonschema_dialect_id is not None:
+            return self._default_jsonschema_dialect_id
+
+        spec_root = self._get_spec_root(schema)
+        dialect_id = (spec_root / "jsonSchemaDialect").read_str(
+            default=OAS31_BASE_DIALECT_URI
+        )
+
+        self._default_jsonschema_dialect_id = dialect_id
+        return dialect_id
 
-    def _get_schema_root(self, schema: SchemaPath) -> SchemaPath:
+    def _get_spec_root(self, schema: SchemaPath) -> SchemaPath:
         # jsonschema-path currently has no public API for root traversal.
         return schema._clone_with_parts(())