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
|
#include "py_tuple.h"
#include "py_cast.h"
#include "py_errors.h"
#include "py_gil.h"
#include "py_utils.h"
#include <yql/essentials/public/udf/udf_value.h>
#include <yql/essentials/public/udf/udf_value_builder.h>
#include <yql/essentials/public/udf/udf_type_inspection.h>
#include <yql/essentials/public/udf/udf_terminator.h>
using namespace NKikimr;
namespace NPython {
TPyObjectPtr ToPyTuple(const TPyCastContext::TPtr& ctx, const NUdf::TType* type, const NUdf::TUnboxedValuePod& value)
{
const NUdf::TTupleTypeInspector inspector(*ctx->PyCtx->TypeInfoHelper, type);
const auto elementsCount = inspector.GetElementsCount();
const TPyObjectPtr tuple(PyTuple_New(elementsCount));
if (auto ptr = value.GetElements()) {
for (ui32 i = 0U; i < elementsCount; ++i) {
auto item = ToPyObject(ctx, inspector.GetElementType(i), *ptr++);
PyTuple_SET_ITEM(tuple.Get(), i, item.Release());
}
} else {
for (ui32 i = 0U; i < elementsCount; ++i) {
auto item = ToPyObject(ctx, inspector.GetElementType(i), value.GetElement(i));
PyTuple_SET_ITEM(tuple.Get(), i, item.Release());
}
}
return tuple;
}
NUdf::TUnboxedValue FromPyTuple(const TPyCastContext::TPtr& ctx, const NUdf::TType* type, PyObject* value)
{
const NUdf::TTupleTypeInspector inspector(*ctx->PyCtx->TypeInfoHelper, type);
if (const TPyObjectPtr fast = PySequence_Fast(value, "Expected tuple or list.")) {
const Py_ssize_t itemsCount = PySequence_Fast_GET_SIZE(fast.Get());
if (itemsCount < 0 || inspector.GetElementsCount() != itemsCount) {
throw yexception() << "Tuple elements count mismatch.";
}
NUdf::TUnboxedValue* tuple_items = nullptr;
const auto tuple = ctx->ValueBuilder->NewArray(inspector.GetElementsCount(), tuple_items);
for (Py_ssize_t i = 0; i < itemsCount; i++) {
const auto item = PySequence_Fast_GET_ITEM(fast.Get(), i);
*tuple_items++ = FromPyObject(ctx, inspector.GetElementType(i), item);
}
return tuple;
}
throw yexception() << "Expected Tuple or Sequence but got: " << PyObjectRepr(value);
}
}
|
