Coverage for pydantic/plugin/_loader.py: 87.50%

1from __future__ import annotations 48 ctx1abcdefghijklmnopqrstuvwxyzMNOPQRSTUVABCDEFGHIJKL

2 

3import importlib.metadata as importlib_metadata 48 ctx1abcdefghijklmnopqrstuvwxyzMNOPQRSTUVABCDEFGHIJKL

4import os 48 ctx1abcdefghijklmnopqrstuvwxyzMNOPQRSTUVABCDEFGHIJKL

5import warnings 48 ctx1abcdefghijklmnopqrstuvwxyzMNOPQRSTUVABCDEFGHIJKL

6from typing import TYPE_CHECKING, Final, Iterable 48 ctx1abcdefghijklmnopqrstuvwxyzMNOPQRSTUVABCDEFGHIJKL

7 

8if TYPE_CHECKING: 48 ctx1abcdefghijklmnopqrstuvwxyzMNOPQRSTUVABCDEFGHIJKL

9 from . import PydanticPluginProtocol 

10 

11 

12PYDANTIC_ENTRY_POINT_GROUP: Final[str] = 'pydantic' 48 ctx1abcdefghijklmnopqrstuvwxyzMNOPQRSTUVABCDEFGHIJKL

13 

14# cache of plugins 

15_plugins: dict[str, PydanticPluginProtocol] | None = None 48 ctx1abcdefghijklmnopqrstuvwxyzMNOPQRSTUVABCDEFGHIJKL

16# return no plugins while loading plugins to avoid recursion and errors while import plugins 

17# this means that if plugins use pydantic 

18_loading_plugins: bool = False 48 ctx1abcdefghijklmnopqrstuvwxyzMNOPQRSTUVABCDEFGHIJKL

19 

20 

21def get_plugins() -> Iterable[PydanticPluginProtocol]: 48 ctx1abcdefghijklmnopqrstuvwxyzMNOPQRSTUVABCDEFGHIJKL

22 """Load plugins for Pydantic. 

23 

24 Inspired by: https://github.com/pytest-dev/pluggy/blob/1.3.0/src/pluggy/_manager.py#L376-L402 

25 """ 

26 disabled_plugins = os.getenv('PYDANTIC_DISABLE_PLUGINS') 48 ctx1abcdefghijklmnopqrstuvwxyzMNOPQRSTUVABCDEFGHIJKL

27 global _plugins, _loading_plugins 

28 if _loading_plugins: 28 ↛ 30line 28 didn't jump to line 30 because the condition on line 28 was never true48 ctx1abcdefghijklmnopqrstuvwxyzMNOPQRSTUVABCDEFGHIJKL

29 # this happens when plugins themselves use pydantic, we return no plugins 

30 return () 

31 elif disabled_plugins in ('__all__', '1', 'true'): 48 ctx1abcdefghijklmnopqrstuvwxyzMNOPQRSTUVABCDEFGHIJKL

32 return () 38 ctx1abcdefghijklmnopqrstuvwxyzABCDEFGHIJKL

33 elif _plugins is None: 48 ctx1abcdefghijklmnopqrstuvwxyzMNOPQRSTUVABCDEFGHIJKL

34 _plugins = {} 48 ctx1abcdefghijklmnopqrstuvwxyzMNOPQRSTUVABCDEFGHIJKL

35 # set _loading_plugins so any plugins that use pydantic don't themselves use plugins 

36 _loading_plugins = True 48 ctx1abcdefghijklmnopqrstuvwxyzMNOPQRSTUVABCDEFGHIJKL

37 try: 48 ctx1abcdefghijklmnopqrstuvwxyzMNOPQRSTUVABCDEFGHIJKL

38 for dist in importlib_metadata.distributions(): 48 ctx1abcdefghijklmnopqrstuvwxyzMNOPQRSTUVABCDEFGHIJKL

39 for entry_point in dist.entry_points: 48 ctx1abcdefghijklmnopqrstuvwxyzMNOPQRSTUVABCDEFGHIJKL

40 if entry_point.group != PYDANTIC_ENTRY_POINT_GROUP: 48 ctx1abcdefghijklmnopqrstuvwxyzMNOPQRSTUVABCDEFGHIJKL

41 continue 48 ctx1abcdefghijklmnopqrstuvwxyzMNOPQRSTUVABCDEFGHIJKL

42 if entry_point.value in _plugins: 42 ↛ 43line 42 didn't jump to line 43 because the condition on line 42 was never true38 ctx1abcdefghijklmnopqrstuvwxyzABCDEFGHIJKL

43 continue 

44 if disabled_plugins is not None and entry_point.name in disabled_plugins.split(','): 38 ctx1abcdefghijklmnopqrstuvwxyzABCDEFGHIJKL

45 continue 38 ctx1abcdefghijklmnopqrstuvwxyzABCDEFGHIJKL

46 try: 38 ctx1abcdefghijklmnopqrstuvwxyzABCDEFGHIJKL

47 _plugins[entry_point.value] = entry_point.load() 38 ctx1abcdefghijklmnopqrstuvwxyzABCDEFGHIJKL

48 except (ImportError, AttributeError) as e: 

49 warnings.warn( 

50 f'{e.__class__.__name__} while loading the `{entry_point.name}` Pydantic plugin, ' 

51 f'this plugin will not be installed.\n\n{e!r}' 

52 ) 

53 finally: 

54 _loading_plugins = False 48 ctx1abcdefghijklmnopqrstuvwxyzMNOPQRSTUVABCDEFGHIJKL

55 

56 return _plugins.values() 48 ctx1abcdefghijklmnopqrstuvwxyzMNOPQRSTUVABCDEFGHIJKL