source: swig/python/chunk.i @ b1f5833

// bytearrays are new as of 2.6
%{
#if (PY_MAJOR_VERSION == 2 && PY_MINOR_VERSION >= 6) || PY_MAJOR_VERSION > 2
#define MODERN_PYTHON
#endif
%}

// abac_chunk_t is a bytearray or a string
%typemap(in) abac_chunk_t %{
    if (PyString_Check($input))
        PyString_AsStringAndSize(
            $input,
            (char **)&$1.ptr,
            &$1.len
        );
#ifdef MODERN_PYTHON
    else if (PyByteArray_Check($input)) {
        $1.ptr = (unsigned char *)PyByteArray_AS_STRING($input);
        $1.len = PyByteArray_GET_SIZE($input);
    }
#endif
    else
        SWIG_exception(SWIG_TypeError, "Expected string or byte array");
%}

%typemap(out) abac_chunk_t %{
#ifdef MODERN_PYTHON
    $result = PyByteArray_FromStringAndSize(
        (const char *)$1.ptr,
        $1.len
    );
#else
    $result = PyString_FromStringAndSize(
        (const char *)$1.ptr,
        $1.len
    );
#endif
%}