Python C Аргументы API Unicode для std :: wstring - PullRequest
Новая
       25

Python C Аргументы API Unicode для std :: wstring

0 голосов
/ 02 апреля 2020

после https://docs.python.org/3/extending/extending.html#keyword -параметры-функции-расширения Я объявил следующую функцию: 

PyObject* myFunction(PyObject *self, PyObject *args, PyObject *keywds) {
    const wchar_t *query;
    std::size_t query_len;
    PyObject* py_choices;
    double score_cutoff = 0;
    bool preprocess = true;
    static const char *kwlist[] = {"query", "choices", "score_cutoff", "preprocess", NULL};

    if (!PyArg_ParseTupleAndKeywords(args, keywds, "u#O|dp", const_cast<char **>(kwlist),
                                     &query, &query_len, &py_choices, &score_cutoff, &preprocess)) {
        return NULL;
    }

    PyObject* choices = PySequence_Fast(py_choices, "Choices must be a sequence of strings");
    if (!choices) {
        return NULL;
    }

    std::size_t choice_count = PySequence_Fast_GET_SIZE(choices);
    std::wstring query_str(query, query_len);

    for (std::size_t i = 0; i < choice_count; ++i) {
        PyObject* py_choice = PySequence_Fast_GET_ITEM(choices, i);

        const wchar_t *choice;
        std::size_t choice_len;
        if (!PyArg_Parse(py_choice, "u#", &choice, &choice_len)) {
            PyErr_SetString(PyExc_TypeError, "Choices must be a sequence of strings");
            Py_DECREF(choices);
            return NULL;
        }

        std::wstring choice_str(choice, choice_len);

        // do some stuff with both strings
    }

    Py_DECREF(choices);
    Py_RETURN_NONE;
}

целью была функция со следующей сигнатурой в python

def myFunction(query: str, choices: Iterable[str], score_cutoff: float = 0, preprocess : bool = True):

После прочтения аргументов и именованных аргументов я хотел бы поместить строки Юникода в std :: wstring.

Редактировать: Как указано в комментариях, формат p требует использование целого числа, поскольку c имеет тип данных bool

...