psycopg/psycopg2
1
1
Fork 0
mirror of https://github.com/psycopg/psycopg2.git synced 2024-11-13 04:26:33 +03:00
Code Issues Packages Projects Releases Wiki Activity

Encodings from Karsten (part 2.)

Browse Source
This commit is contained in:
Federico Di Gregorio 2007-01-19 14:47:21 +00:00
parent 5642a68adb
commit 3a94e747d7
1 changed files with 15 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
psycopg/connection_type.c
View File

@ -180,21 +180,30 @@ psyco_conn_set_isolation_level(connectionObject *self, PyObject *args)
static PyObject * static PyObject *
psyco_conn_set_client_encoding(connectionObject *self, PyObject *args) psyco_conn_set_client_encoding(connectionObject *self, PyObject *args)
{ {
char *pos, *enc = NULL; char *buffer, *enc = NULL;
int i, j;
EXC_IF_CONN_CLOSED(self); EXC_IF_CONN_CLOSED(self);
if (!PyArg_ParseTuple(args, "s", &enc)) return NULL; if (!PyArg_ParseTuple(args, "s", &enc)) return NULL;
/* convert to upper case */ /* convert to upper case and remove '-' and '_' from string */
for (pos = enc ; *pos != '\0' ; pos++) buffer = PyMem_Malloc(strlen(enc));
*pos = toupper(*pos); for (i=j=0 ; i < strlen(enc) ; i++) {
if (enc[i] == '_' || enc[i] == '-')
continue;
else
buffer[j++] = toupper(enc[i]);
}
buffer[j] = '\0';
if (conn_set_client_encoding(self, enc) == 0) { if (conn_set_client_encoding(self, buffer) == 0) {
PyMem_Free(buffer);
Py_INCREF(Py_None); Py_INCREF(Py_None);
return Py_None; return Py_None;
} }
else { else {
PyMem_Free(buffer);
return NULL; return NULL;
} }
} }