Как использовать декоратор, чтобы отметить обязательный аргумент только для ключевого слова как необязательный в подписи ⇐ Python

[Anonymous]

У меня есть куча функций, которые имеют определенный аргумент ключевого слова, давайте назовем его kw , который является обязательным и имеет определенный тип t . Тем не менее, я хочу иметь возможность вызвать функции, не обязательно предоставляя этот аргумент ключевого слова, и если его нет, я хочу, чтобы они были созданы, вероятно, через декоратор. Кроме того, я хочу, чтобы мой IDE показал свою подпись, как если бы KW были необязательными.
Другими словами, у меня есть функции, которые имеют подпись, как это:

Код: Выделить всё

def f(*args, kw: T, **kwargs): ...
< /code>
Я хочу, чтобы их подпись подсказки, как если бы она была определена так: < /p>
def f(*args, kw: T | None=None, **kwargs): ...

Примечание, что каждая функция может иметь другие позиционные и/или ключевые аргументы. Декоратор вроде так: < /p>

Код: Выделить всё

def decorator(func):
sig = inspect.signature(func)
hints = get_type_hints(func)
orig_ann = hints.get("kw")
new_ann = Optional[orig_ann]

new_params = []
for param in sig.parameters.values():
if param.name == "kw":
new_params.append(
param.replace(
annotation=new_ann,
default=None,
)
)
else:
new_params.append(param)

new_sig = sig.replace(parameters=new_params)

@wraps(func)
def wrapper(*args, **kwargs):
if kwargs.get("kw") is None:
kwargs["kw"] = some_function_or_something()
return func(*args, **kwargs)

"do something to update the signature"
return wrapper

Но я не могу понять что мне нужно было сделать, чтобы обновить статическую подпись так, чтобы подсказка изменилась. Является ли это возможным? Как бы я это сделал, если да?>

Подробнее здесь: https://stackoverflow.com/questions/796 ... ptional-in