technocloud-public/nginx-unit
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Python: bytearray body support for ASGI module.

Browse Source 
@filiphanes requested support for bytearray
and memoryview in the request body here:
<https://github.com/nginx/unit/issues/648>

This patch implements bytearray body support only.
Memoryview body still need to be implemented.
This commit is contained in:
Andrei Zeliankou
2024-01-26 14:58:43 +00:00
committed by andrey-zelenkov
parent f71ead5fa5
commit 697a585062
4 changed files with 53 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
src/python/nxt_python_asgi_http.c
View File

@@ -362,16 +362,6 @@ nxt_py_asgi_http_response_body(nxt_py_asgi_http_t *http, PyObject *dict)
Py_ssize_t body_len, body_off;
nxt_py_asgi_ctx_data_t *ctx_data;
body = PyDict_GetItem(dict, nxt_py_body_str);
if (nxt_slow_path(body != NULL && !PyBytes_Check(body))) {
return PyErr_Format(PyExc_TypeError, "'body' is not a byte string");
}
more_body = PyDict_GetItem(dict, nxt_py_more_body_str);
if (nxt_slow_path(more_body != NULL && !PyBool_Check(more_body))) {
return PyErr_Format(PyExc_TypeError, "'more_body' is not a bool");
}
if (nxt_slow_path(http->complete)) {
return PyErr_Format(PyExc_RuntimeError,
"Unexpected ASGI message 'http.response.body' "
@@ -382,9 +372,26 @@ nxt_py_asgi_http_response_body(nxt_py_asgi_http_t *http, PyObject *dict)
return PyErr_Format(PyExc_RuntimeError, "Concurrent send");
}
more_body = PyDict_GetItem(dict, nxt_py_more_body_str);
if (nxt_slow_path(more_body != NULL && !PyBool_Check(more_body))) {
return PyErr_Format(PyExc_TypeError, "'more_body' is not a bool");
}
body = PyDict_GetItem(dict, nxt_py_body_str);
if (body != NULL) {
body_str = PyBytes_AS_STRING(body);
body_len = PyBytes_GET_SIZE(body);
if (PyBytes_Check(body)) {
body_str = PyBytes_AS_STRING(body);
body_len = PyBytes_GET_SIZE(body);
} else if (PyByteArray_Check(body)) {
body_str = PyByteArray_AS_STRING(body);
body_len = PyByteArray_GET_SIZE(body);
} else {
return PyErr_Format(PyExc_TypeError,
"'body' is not a byte string or bytearray");
}
nxt_unit_req_debug(http->req, "asgi_http_response_body: %d, %d",
(int) body_len, (more_body == Py_True) );