41 lines
1.2 KiB
Python
41 lines
1.2 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from .base import PluginLoader
|
||
|
|
|
||
|
|
|
||
|
|
class Discovery(PluginLoader):
|
||
|
|
"""Discovery plugins."""
|
||
|
|
|
||
|
|
|
||
|
|
def get_discover(parser, args):
|
||
|
|
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)
|
||
|
|
discovery_parser.add_argument(
|
||
|
|
"--discovery",
|
||
|
|
choices=choices,
|
||
|
|
default=next(iter(choices)),
|
||
|
|
required=False,
|
||
|
|
help="interpreter discovery method",
|
||
|
|
)
|
||
|
|
options, _ = parser.parse_known_args(args)
|
||
|
|
discover_class = discover_types[options.discovery]
|
||
|
|
discover_class.add_parser_arguments(discovery_parser)
|
||
|
|
options, _ = parser.parse_known_args(args, namespace=options)
|
||
|
|
return discover_class(options)
|
||
|
|
|
||
|
|
|
||
|
|
def _get_default_discovery(discover_types):
|
||
|
|
return list(discover_types.keys())
|
||
|
|
|
||
|
|
|
||
|
|
__all__ = [
|
||
|
|
"Discovery",
|
||
|
|
"get_discover",
|
||
|
|
]
|