Use fastjsonschema to validate user configuration

aiida/manage/configuration/config.py

@@ -124,13 +124,12 @@ def _backup(cls, filepath):
     @staticmethod
     def validate(config: dict, filepath: Optional[str] = None):
         """Validate a configuration dictionary."""
-        import jsonschema
+        from .schema.fastjsonschema_validate_v9 import JsonSchemaValueException, validate
+
         try:
-            jsonschema.validate(instance=config, schema=config_schema())
-        except jsonschema.ValidationError as error:
-            raise ConfigValidationError(
-                message=error.message, keypath=error.path, schema=error.schema, filepath=filepath
-            )
+            validate(data=config)
+        except JsonSchemaValueException as error:
+            raise ConfigValidationError(message=error.message, keypath=error.path, schema=None, filepath=filepath)
 
     def __init__(self, filepath: str, config: dict, validate: bool = True):
         """Instantiate a configuration object from a configuration dictionary and its filepath.

aiida/manage/configuration/options.py

@@ -62,7 +62,7 @@ def validate(self, value: Any, cast: bool = True) -> Any:
 
         """
         # pylint: disable=too-many-branches
-        import jsonschema
+        import fastjsonschema
 
         from aiida.manage.caching import _validate_identifier_pattern
 
@@ -90,9 +90,9 @@ def validate(self, value: Any, cast: bool = True) -> Any:
                 pass
 
         try:
-            jsonschema.validate(instance=value, schema=self.schema)
-        except jsonschema.ValidationError as exc:
-            raise ConfigValidationError(message=exc.message, keypath=[self.name, *(exc.path or [])], schema=exc.schema)
+            fastjsonschema.validate(data=value, definition=self.schema)
+        except fastjsonschema.JsonSchemaException as exc:
+            raise ConfigValidationError(message=exc.message, keypath=[self.name, *(exc.path or [])], schema=self.schema)  # pylint: disable=no-member
 
         # special caching validation
         if self.name in ('caching.enabled_for', 'caching.disabled_for'):

docs/source/nitpick-exceptions

@@ -212,8 +212,6 @@ py:class CircusClient
 py:class pgsu.PGSU
 py:meth  pgsu.PGSU.__init__
 
-py:class jsonschema.exceptions._Error
-
 py:class Session
 py:class Query
 py:class importlib_metadata.EntryPoint

environment.yml

@@ -14,11 +14,11 @@ dependencies:
 - click~=8.1
 - disk-objectstore~=0.6.0
 - docstring_parser
+- python-fastjsonschema~=2.17
 - get-annotations~=0.1
 - python-graphviz~=0.19
 - ipython>=7
 - jinja2~=3.0
-- jsonschema~=3.0
 - kiwipy[rmq]~=0.7.7
 - importlib-metadata~=6.0
 - numpy~=1.21

pyproject.toml

@@ -32,11 +32,11 @@ dependencies = [
     "click~=8.1",
     "disk-objectstore~=0.6.0",
     "docstring-parser",
+    "fastjsonschema~=2.17",
     "get-annotations~=0.1;python_version<'3.10'",
     "graphviz~=0.19",
     "ipython>=7",
     "jinja2~=3.0",
-    "jsonschema~=3.0",
     "kiwipy[rmq]~=0.7.7",
     "importlib-metadata~=6.0",
     "numpy~=1.21",
@@ -410,13 +410,13 @@ module = [
     'click_config_file.*',
     'click_spinner.*',
     'docutils.*',
+    'fastjsonschema.*',
     'flask_cors.*',
     'flask_restful.*',
     'get_annotations.*',
     'graphviz.*',
     'importlib._bootstrap.*',
     'IPython.*',
-    'jsonschema.*',
     'kiwipy.*',
     'matplotlib.*',
     'mayavi.*',
@@ -452,6 +452,10 @@ ignore_missing_imports = true
 module = 'plumpy'
 ignore_errors = true
 
+[[tool.mypy.overrides]]
+module = 'aiida.manage.configuration.schema.fastjsonschema_validate_v9'
+ignore_errors = true
+
 [tool.tox]
 legacy_tox_ini = """
 [tox]

requirements/requirements-py-3.10.txt

@@ -66,7 +66,6 @@ ipywidgets==8.0.6
 itsdangerous==2.1.2
 jedi==0.18.2
 jinja2==3.1.2
-jsonschema[format-nongpl]==3.2.0
 jupyter==1.0.0
 jupyter-cache==0.6.1
 jupyter-client==8.2.0

requirements/requirements-py-3.11.txt

@@ -65,7 +65,6 @@ ipywidgets==8.0.6
 itsdangerous==2.1.2
 jedi==0.18.2
 jinja2==3.1.2
-jsonschema[format-nongpl]==3.2.0
 jupyter==1.0.0
 jupyter-cache==0.6.1
 jupyter-client==8.2.0

requirements/requirements-py-3.9.txt

@@ -68,7 +68,6 @@ ipywidgets==8.0.6
 itsdangerous==2.1.2
 jedi==0.18.2
 jinja2==3.1.2
-jsonschema[format-nongpl]==3.2.0
 jupyter==1.0.0
 jupyter-cache==0.6.1
 jupyter-client==8.2.0

utils/dependency_management.py

@@ -27,6 +27,7 @@
 ROOT = Path(__file__).resolve().parent.parent  # repository root
 
 SETUPTOOLS_CONDA_MAPPINGS = {
+    'fastjsonschema': 'python-fastjsonschema',
     'graphviz': 'python-graphviz',
     'docstring-parser': 'docstring_parser',
 }