mirror of
https://github.com/kennethreitz/simplemind.git
synced 2026-06-05 22:50:18 +00:00
38 lines
1.1 KiB
Python
38 lines
1.1 KiB
Python
import difflib
|
|
from typing import Optional, Type
|
|
|
|
from .providers import providers, BaseProvider
|
|
|
|
_PROVIDER_NAMES = [provider.NAME.lower() for provider in providers]
|
|
|
|
|
|
def find_provider(provider_name: Optional[str]) -> Type[BaseProvider]:
|
|
"""
|
|
Find and instantiate a provider by name.
|
|
|
|
Parameters:
|
|
provider_name (Union[str, None]): The name of the provider to find.
|
|
|
|
Returns:
|
|
An instance of the provider class if found.
|
|
|
|
Raises:
|
|
ValueError: If the provider is not found, with a suggestion for the closest match.
|
|
"""
|
|
if provider_name:
|
|
for provider_class in providers:
|
|
if provider_class.NAME.lower() == provider_name.lower():
|
|
# Instantiate the provider
|
|
return provider_class()
|
|
|
|
provider_found = difflib.get_close_matches(
|
|
provider_name.lower(), _PROVIDER_NAMES, n=1
|
|
) # Show only one suggestion
|
|
|
|
if provider_found:
|
|
raise ValueError(
|
|
f"Provider {provider_name!r} not found. Did you mean {provider_found[0]!r}?"
|
|
)
|
|
else:
|
|
raise ValueError(f"Provider {provider_name} not found.")
|