1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
|
import inspect
from typing import Any
from typing import Callable
from typing import Dict
from typing import List
from typing import Set
ACTIVATED_ASYNC_MODE = False
def is_prompt_toolkit_3() -> bool:
from prompt_toolkit import __version__ as ptk_version
return ptk_version.startswith("3.")
def default_values_of(func: Callable[..., Any]) -> List[str]:
"""Return all parameter names of ``func`` with a default value."""
signature = inspect.signature(func)
return [
k
for k, v in signature.parameters.items()
if v.default is not inspect.Parameter.empty
or v.kind != inspect.Parameter.POSITIONAL_OR_KEYWORD
]
def arguments_of(func: Callable[..., Any]) -> List[str]:
"""Return the parameter names of the function ``func``."""
return list(inspect.signature(func).parameters.keys())
def used_kwargs(kwargs: Dict[str, Any], func: Callable[..., Any]) -> Dict[str, Any]:
"""Returns only the kwargs which can be used by a function.
Args:
kwargs: All available kwargs.
func: The function which should be called.
Returns:
Subset of kwargs which are accepted by ``func``.
"""
possible_arguments = arguments_of(func)
return {k: v for k, v in kwargs.items() if k in possible_arguments}
def required_arguments(func: Callable[..., Any]) -> List[str]:
"""Return all arguments of a function that do not have a default value."""
defaults = default_values_of(func)
args = arguments_of(func)
if defaults:
args = args[: -len(defaults)]
return args # all args without default values
def missing_arguments(func: Callable[..., Any], argdict: Dict[str, Any]) -> Set[str]:
"""Return all arguments that are missing to call func."""
return set(required_arguments(func)) - set(argdict.keys())
async def activate_prompt_toolkit_async_mode() -> None:
"""Configure prompt toolkit to use the asyncio event loop.
Needs to be async, so we use the right event loop in py 3.5"""
global ACTIVATED_ASYNC_MODE
if not is_prompt_toolkit_3():
# Tell prompt_toolkit to use asyncio for the event loop.
import prompt_toolkit as pt
pt.eventloop.use_asyncio_event_loop() # type: ignore[attr-defined]
ACTIVATED_ASYNC_MODE = True
|
