Fix bug in current registration of validators

Author: agoose77

jsonschema/tests/test_validators.py

@@ -1030,6 +1030,7 @@ def test_custom_validator(self):
         Validator = validators.create(
             meta_schema={"id": "meta schema id"},
             version="12",
+            id_of=lambda s: s.get("id", ""),
         )
         schema = {"$schema": "meta schema id"}
         self.assertIs(

jsonschema/validators.py

@@ -52,8 +52,9 @@ def validates(version):
 
     def _validates(cls):
         validators[version] = cls
-        if u"id" in cls.META_SCHEMA:
-            meta_schemas[cls.META_SCHEMA[u"id"]] = cls
+        meta_schema_id = cls.ID_OF(cls.META_SCHEMA)
+        if meta_schema_id:
+            meta_schemas[meta_schema_id] = cls
         return cls
     return _validates
 
@@ -204,6 +205,7 @@ class Validator(object):
         VALIDATORS = dict(validators)
         META_SCHEMA = dict(meta_schema)
         TYPE_CHECKER = type_checker
+        ID_OF = staticmethod(id_of)
 
         _DEFAULT_TYPES = dict(default_types)