Add PyUnicode_AsStringAndSize(), which is like PyUnicode_AsString() but
has an extra (optional) output parameter through which it returns the size. Use this in a few places where I used PyUnicode_AsString() + strlen(), and in one new place (which fixes test_pep263).
This commit is contained in:
parent
9befa93b04
commit
7d1df6c9b1
|
@ -641,20 +641,25 @@ PyAPI_FUNC(PyObject*) PyUnicode_FromOrdinal(int ordinal);
|
||||||
PyAPI_FUNC(PyObject *) _PyUnicode_AsDefaultEncodedString(
|
PyAPI_FUNC(PyObject *) _PyUnicode_AsDefaultEncodedString(
|
||||||
PyObject *, const char *);
|
PyObject *, const char *);
|
||||||
|
|
||||||
/* Return a char* holding the default encoded value of the
|
/* Return a char* holding the UTF-8 encoded value of the
|
||||||
Unicode object.
|
Unicode object.
|
||||||
|
|
||||||
|
DEPRECATED: use PyUnicode_AsStringAndSize() instead.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
|
PyAPI_FUNC(char *) PyUnicode_AsStringAndSize(PyObject*, Py_ssize_t *);
|
||||||
|
|
||||||
|
/* Returns the UTF-8 encoding, and its size.
|
||||||
|
|
||||||
|
If the output argument is NULL, no size is stored.
|
||||||
|
*/
|
||||||
|
|
||||||
PyAPI_FUNC(char *) PyUnicode_AsString(PyObject*);
|
PyAPI_FUNC(char *) PyUnicode_AsString(PyObject*);
|
||||||
|
|
||||||
|
/* Returns the UTF-8 encoding.
|
||||||
|
|
||||||
/* Returns the currently active default encoding.
|
This is equivalent to PyUnicode_AsStringAndSize(x, NULL).
|
||||||
|
|
||||||
The default encoding is currently implemented as run-time settable
|
|
||||||
process global. This may change in future versions of the
|
|
||||||
interpreter to become a parameter which is managed on a per-thread
|
|
||||||
basis.
|
|
||||||
|
|
||||||
*/
|
*/
|
||||||
|
|
||||||
PyAPI_FUNC(const char*) PyUnicode_GetDefaultEncoding(void);
|
PyAPI_FUNC(const char*) PyUnicode_GetDefaultEncoding(void);
|
||||||
|
|
|
@ -50,12 +50,11 @@ int pysqlite_statement_create(pysqlite_Statement* self, pysqlite_Connection* con
|
||||||
self->st = NULL;
|
self->st = NULL;
|
||||||
self->in_use = 0;
|
self->in_use = 0;
|
||||||
|
|
||||||
sql_cstr = PyUnicode_AsString(sql);
|
sql_cstr = PyUnicode_AsStringAndSize(sql, &sql_cstr_len);
|
||||||
if (sql_cstr == NULL) {
|
if (sql_cstr == NULL) {
|
||||||
rc = PYSQLITE_SQL_WRONG_TYPE;
|
rc = PYSQLITE_SQL_WRONG_TYPE;
|
||||||
return rc;
|
return rc;
|
||||||
}
|
}
|
||||||
sql_cstr_len = strlen(sql_cstr); /* XXX */
|
|
||||||
|
|
||||||
self->in_weakreflist = NULL;
|
self->in_weakreflist = NULL;
|
||||||
Py_INCREF(sql);
|
Py_INCREF(sql);
|
||||||
|
@ -216,12 +215,11 @@ int pysqlite_statement_recompile(pysqlite_Statement* self, PyObject* params)
|
||||||
Py_ssize_t sql_len;
|
Py_ssize_t sql_len;
|
||||||
sqlite3_stmt* new_st;
|
sqlite3_stmt* new_st;
|
||||||
|
|
||||||
sql_cstr = PyUnicode_AsString(self->sql);
|
sql_cstr = PyUnicode_AsStringAndSize(self->sql, &sql_len);
|
||||||
if (sql_cstr == NULL) {
|
if (sql_cstr == NULL) {
|
||||||
rc = PYSQLITE_SQL_WRONG_TYPE;
|
rc = PYSQLITE_SQL_WRONG_TYPE;
|
||||||
return rc;
|
return rc;
|
||||||
}
|
}
|
||||||
sql_len = strlen(sql_cstr); /* XXXX */
|
|
||||||
|
|
||||||
rc = sqlite3_prepare(self->db,
|
rc = sqlite3_prepare(self->db,
|
||||||
sql_cstr,
|
sql_cstr,
|
||||||
|
|
|
@ -1184,16 +1184,25 @@ PyObject *_PyUnicode_AsDefaultEncodedString(PyObject *unicode,
|
||||||
}
|
}
|
||||||
|
|
||||||
char*
|
char*
|
||||||
PyUnicode_AsString(PyObject *unicode)
|
PyUnicode_AsStringAndSize(PyObject *unicode, Py_ssize_t *psize)
|
||||||
{
|
{
|
||||||
|
PyObject *str8;
|
||||||
if (!PyUnicode_Check(unicode)) {
|
if (!PyUnicode_Check(unicode)) {
|
||||||
PyErr_BadArgument();
|
PyErr_BadArgument();
|
||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
unicode = _PyUnicode_AsDefaultEncodedString(unicode, NULL);
|
str8 = _PyUnicode_AsDefaultEncodedString(unicode, NULL);
|
||||||
if (!unicode)
|
if (str8 == NULL)
|
||||||
return NULL;
|
return NULL;
|
||||||
return PyString_AsString(unicode);
|
if (psize != NULL)
|
||||||
|
*psize = PyString_GET_SIZE(str8);
|
||||||
|
return PyString_AS_STRING(str8);
|
||||||
|
}
|
||||||
|
|
||||||
|
char*
|
||||||
|
PyUnicode_AsString(PyObject *unicode)
|
||||||
|
{
|
||||||
|
return PyUnicode_AsStringAndSize(unicode, NULL);
|
||||||
}
|
}
|
||||||
|
|
||||||
Py_UNICODE *PyUnicode_AsUnicode(PyObject *unicode)
|
Py_UNICODE *PyUnicode_AsUnicode(PyObject *unicode)
|
||||||
|
@ -8098,6 +8107,7 @@ unicode_buffer_getbuffer(PyUnicodeObject *self, PyBuffer *view, int flags)
|
||||||
|
|
||||||
if (flags & PyBUF_CHARACTER) {
|
if (flags & PyBUF_CHARACTER) {
|
||||||
PyErr_SetString(PyExc_SystemError, "can't use str as char buffer");
|
PyErr_SetString(PyExc_SystemError, "can't use str as char buffer");
|
||||||
|
abort();
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
return PyBuffer_FillInfo(view, (void *)self->str,
|
return PyBuffer_FillInfo(view, (void *)self->str,
|
||||||
|
|
|
@ -383,7 +383,8 @@ fp_readl(char *s, int size, struct tok_state *tok)
|
||||||
goto error;
|
goto error;
|
||||||
allocated = 1;
|
allocated = 1;
|
||||||
}
|
}
|
||||||
if (PyObject_AsCharBuffer(bufobj, &buf, &buflen) < 0) {
|
buf = PyUnicode_AsStringAndSize(bufobj, &buflen);
|
||||||
|
if (buf == NULL) {
|
||||||
goto error;
|
goto error;
|
||||||
}
|
}
|
||||||
if (buflen > size) {
|
if (buflen > size) {
|
||||||
|
|
Loading…
Reference in New Issue