aboutsummaryrefslogtreecommitdiff
path: root/.venv/lib/python3.12/site-packages/pydantic/plugin/_loader.py
diff options
context:
space:
mode:
authorS. Solomon Darnell2025-03-28 21:52:21 -0500
committerS. Solomon Darnell2025-03-28 21:52:21 -0500
commit4a52a71956a8d46fcb7294ac71734504bb09bcc2 (patch)
treeee3dc5af3b6313e921cd920906356f5d4febc4ed /.venv/lib/python3.12/site-packages/pydantic/plugin/_loader.py
parentcc961e04ba734dd72309fb548a2f97d67d578813 (diff)
downloadgn-ai-master.tar.gz
two version of R2R are hereHEADmaster
Diffstat (limited to '.venv/lib/python3.12/site-packages/pydantic/plugin/_loader.py')
-rw-r--r--.venv/lib/python3.12/site-packages/pydantic/plugin/_loader.py56
1 files changed, 56 insertions, 0 deletions
diff --git a/.venv/lib/python3.12/site-packages/pydantic/plugin/_loader.py b/.venv/lib/python3.12/site-packages/pydantic/plugin/_loader.py
new file mode 100644
index 00000000..2f90dc54
--- /dev/null
+++ b/.venv/lib/python3.12/site-packages/pydantic/plugin/_loader.py
@@ -0,0 +1,56 @@
+from __future__ import annotations
+
+import importlib.metadata as importlib_metadata
+import os
+import warnings
+from typing import TYPE_CHECKING, Final, Iterable
+
+if TYPE_CHECKING:
+ from . import PydanticPluginProtocol
+
+
+PYDANTIC_ENTRY_POINT_GROUP: Final[str] = 'pydantic'
+
+# cache of plugins
+_plugins: dict[str, PydanticPluginProtocol] | None = None
+# return no plugins while loading plugins to avoid recursion and errors while import plugins
+# this means that if plugins use pydantic
+_loading_plugins: bool = False
+
+
+def get_plugins() -> Iterable[PydanticPluginProtocol]:
+ """Load plugins for Pydantic.
+
+ Inspired by: https://github.com/pytest-dev/pluggy/blob/1.3.0/src/pluggy/_manager.py#L376-L402
+ """
+ disabled_plugins = os.getenv('PYDANTIC_DISABLE_PLUGINS')
+ global _plugins, _loading_plugins
+ if _loading_plugins:
+ # this happens when plugins themselves use pydantic, we return no plugins
+ return ()
+ elif disabled_plugins in ('__all__', '1', 'true'):
+ return ()
+ elif _plugins is None:
+ _plugins = {}
+ # set _loading_plugins so any plugins that use pydantic don't themselves use plugins
+ _loading_plugins = True
+ try:
+ for dist in importlib_metadata.distributions():
+ for entry_point in dist.entry_points:
+ if entry_point.group != PYDANTIC_ENTRY_POINT_GROUP:
+ continue
+ if entry_point.value in _plugins:
+ continue
+ if disabled_plugins is not None and entry_point.name in disabled_plugins.split(','):
+ continue
+ try:
+ _plugins[entry_point.value] = entry_point.load()
+ except (ImportError, AttributeError) as e:
+ warnings.warn(
+ f'{e.__class__.__name__} while loading the `{entry_point.name}` Pydantic plugin, '
+ f'this plugin will not be installed.\n\n{e!r}'
+ )
+ finally:
+ _loading_plugins = False
+
+ return _plugins.values()