Канонизировать арги и кварги в кварго-каноническую форму - PullRequest
0 голосов
/ 29 апреля 2019

Я задал похожий вопрос здесь для позарг-канонической формы.

Любые возможные аргументы должны быть преобразованы в форму ключевого слова, при этом значения по умолчанию опускаются на основе результата оператора is, id или аналогичного.

Например:

def myfunc(a, b=None, **kwargs):
    pass

def canonicalize(func, *args, **kwargs):
    something = inspect.signature(func)
    # Do something with args/kwargs here
    return new_args, new_kwargs

Пример вывода:

>>> canonicalize(myfunc, 1, 2, g=3)
(1,), {'b': 2, 'g': 3}
>>> canonicalize(myfunc, 1)
(1,), {}
>>> canonicalize(myfunc, 1, b=2)
(1,), {'b': 2}
>>> canonicalize(myfunc, 1, g=3, b=None)
(1,), {'g': 3}

1 Ответ

1 голос
/ 29 апреля 2019

Вы можете использовать следующую функцию - вы в основном перемещаете аргументы, которые имеют значение по умолчанию, во все, что было в ** kwargs (если оно на самом деле не присвоено своему собственному значению по умолчанию):

import inspect

def canonicalize(f, *args, **kwargs):
    sig = inspect.signature(f)
    bargs = sig.bind(*args, **kwargs)
    # Pop out the named kwargs variable defaulting to {}
    ret_kwargs = bargs.arguments.pop(inspect.getfullargspec(f).varkw, {})
    # For all possible signature values
    for k, v in sig.parameters.items():
        # If the name exists in the bound arguments and has a default value
        if k in bargs.arguments and v.default is not inspect.Parameter.empty:
            # Remove from the bound arguments dict
            val = bargs.arguments.pop(k)
            # If the value isn't the same as the default value add it to ret_kwargs
            if val is not v.default:
                ret_kwargs[k] = val

    # bargs.args here will be made up of what's left in bargs.arguments
    return bargs.args, ret_kwargs
...