86 lines
2.3 KiB
Python
86 lines
2.3 KiB
Python
import inspect
|
|
from typing import Any, Dict
|
|
from typing import Callable
|
|
|
|
|
|
def supports_argument(func: Callable, arg_name: str) -> bool:
|
|
"""Check whether a function has a specific argument
|
|
|
|
### Args:
|
|
* func (`Callable`): Function to be inspected
|
|
* arg_name (`str`): Argument to be checked
|
|
|
|
### Returns:
|
|
* `bool`: `True` if argument is supported and `False` if not
|
|
"""
|
|
if hasattr(func, "__code__"):
|
|
return arg_name in inspect.signature(func).parameters
|
|
|
|
if hasattr(func, "__doc__"):
|
|
if doc := func.__doc__:
|
|
first_line = doc.splitlines()[0]
|
|
return arg_name in first_line
|
|
|
|
return False
|
|
|
|
|
|
def nested_set(target: dict, value: Any, *path: str, create_missing=True) -> Dict[str, Any]:
|
|
"""Set the key by its path to the value
|
|
|
|
### Args:
|
|
* target (`dict`): Dictionary to perform modifications on
|
|
* value (`Any`): Any data
|
|
* *path (`str`): Path to the key of the target
|
|
* create_missing (`bool`, *optional*): Create keys on the way if they're missing. Defaults to `True`
|
|
|
|
### Raises:
|
|
* `KeyError`: Key is not found under path provided
|
|
|
|
### Returns:
|
|
* `Dict[str, Any]`: Changed dictionary
|
|
"""
|
|
d = target
|
|
|
|
for key in path[:-1]:
|
|
if key in d:
|
|
d = d[key]
|
|
elif create_missing:
|
|
d = d.setdefault(key, {})
|
|
else:
|
|
raise KeyError(
|
|
f"Key '{key}' is not found under path provided ({path}) and create_missing is False"
|
|
)
|
|
|
|
if path[-1] in d or create_missing:
|
|
d[path[-1]] = value
|
|
|
|
return target
|
|
|
|
|
|
def nested_delete(target: dict, *path: str) -> Dict[str, Any]:
|
|
"""Delete the key by its path
|
|
|
|
### Args:
|
|
* target (`dict`): Dictionary to perform modifications on
|
|
|
|
### Raises:
|
|
* `KeyError`: Key is not found under path provided
|
|
|
|
### Returns:
|
|
`Dict[str, Any]`: Changed dictionary
|
|
"""
|
|
d = target
|
|
|
|
for key in path[:-1]:
|
|
if key in d:
|
|
d = d[key]
|
|
else:
|
|
raise KeyError(f"Key '{key}' is not found under path provided ({path})")
|
|
|
|
if path[-1] in d:
|
|
del d[path[-1]]
|
|
else:
|
|
raise KeyError(f"Key '{path[-1]}' is not found under path provided ({path})")
|
|
|
|
return target
|