Allow writing bytes when sending escape code to child

This commit is contained in:
Kovid Goyal 2021-09-24 14:06:09 +05:30
parent 2fcfa8d3eb
commit 67cef371dc
No known key found for this signature in database
GPG Key ID: 06BC317B515ACE7C
2 changed files with 3 additions and 2 deletions

View File

@ -993,7 +993,7 @@ class Screen:
def resize(self, width: int, height: int) -> None:
pass
def send_escape_code_to_child(self, code: int, text: str) -> bool:
def send_escape_code_to_child(self, code: int, text: Union[str, bytes]) -> bool:
pass
def reset_callbacks(self) -> None:

View File

@ -3235,7 +3235,8 @@ static PyObject*
send_escape_code_to_child(Screen *self, PyObject *args) {
int code;
char *text;
if (!PyArg_ParseTuple(args, "is", &code, &text)) return NULL;
Py_ssize_t sz;
if (!PyArg_ParseTuple(args, "is#", &code, &text, &sz)) return NULL;
if (write_escape_code_to_child(self, code, text)) Py_RETURN_TRUE;
Py_RETURN_FALSE;
}