Macbook
61 lines
2.0 KiB
Python
61 lines
2.0 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
from .base import PluginLoader
|
|
|
|
if TYPE_CHECKING:
|
|
from virtualenv.config.cli.parser import VirtualEnvConfigParser
|
|
from virtualenv.discovery.discover import Discover
|
|
|
|
|
|
class Discovery(PluginLoader):
|
|
"""Discovery plugins."""
|
|
|
|
|
|
def get_discover(parser: VirtualEnvConfigParser, args: list[str] | None) -> Discover:
|
|
discover_types = Discovery.entry_points_for("virtualenv.discovery")
|
|
discovery_parser = parser.add_argument_group(
|
|
title="discovery",
|
|
description="discover and provide a target interpreter",
|
|
)
|
|
choices = _get_default_discovery(discover_types)
|
|
# prefer the builtin if present, otherwise fallback to first defined type
|
|
choices = sorted(choices, key=lambda a: 0 if a == "builtin" else 1)
|
|
try:
|
|
default_discovery = next(iter(choices))
|
|
except StopIteration as e:
|
|
msg = "No discovery plugin found. Try reinstalling virtualenv to fix this issue."
|
|
raise RuntimeError(msg) from e
|
|
discovery_parser.add_argument(
|
|
"--discovery",
|
|
choices=choices,
|
|
default=default_discovery,
|
|
required=False,
|
|
help="interpreter discovery method",
|
|
)
|
|
options, _ = parser.parse_known_args(args)
|
|
discovery = options.discovery
|
|
if discovery not in discover_types:
|
|
available = ", ".join(sorted(discover_types))
|
|
msg = (
|
|
f"discovery {discovery!r} is not available. "
|
|
f"Available discovery methods: {available}. "
|
|
f"Is the plugin installed?"
|
|
)
|
|
raise RuntimeError(msg)
|
|
discover_class = discover_types[discovery]
|
|
discover_class.add_parser_arguments(discovery_parser) # ty: ignore[unresolved-attribute]
|
|
options, _ = parser.parse_known_args(args, namespace=options)
|
|
return discover_class(options)
|
|
|
|
|
|
def _get_default_discovery(discover_types: dict[str, type]) -> list[str]:
|
|
return list(discover_types.keys())
|
|
|
|
|
|
__all__ = [
|
|
"Discovery",
|
|
"get_discover",
|
|
]
|