mirror of
https://github.com/PyMySQL/mysqlclient.git
synced 2025-08-16 03:50:43 +08:00
A home-made mysql interface for Python so I can learn the API. It's almost
sorta functional. The problem with the existing interface (minor) is that it's for 3.21 mysql. Not a huge problem, but it doesn't seem to have a stable maintainer at the moment either. Tommorrow I work on result objects so I can get queries done and stuff. Python 1.5.2b2 (#1, Feb 18 1999, 18:20:08) [GCC egcs-2.90.29 980515 (egcs-1.0.3 re on linux2 Copyright 1991-1995 Stichting Mathematisch Centrum, Amsterdam >>> import _mysql >>> dir(_mysql) ['__doc__', '__file__', '__name__', 'connect', 'error', 'escape', 'get_client_info'] >>> _mysql.get_client_info() '3.22.19a' >>> c=_mysql.connect(passwd='foobar') >>> dir(c) ['affected_rows', 'closed', 'dump_debug_info', 'errno', 'error', 'get_host_info', 'get_proto_info', 'get_server_info', 'info', 'open'] >>> c <open connection to 'localhost' at 80ddf50> >>> c.get_server_info() '3.22.19a' >>>
This commit is contained in:
adustman
453
mysql/_mysqlmodule.c
Normal file
453
mysql/_mysqlmodule.c
Normal file
@ -0,0 +1,453 @@
|
||||
#include "Python.h"
|
||||
#include "structmember.h"
|
||||
#include "mysql.h"
|
||||
|
||||
static PyObject *_mysql_Error;
|
||||
|
||||
typedef struct {
|
||||
PyObject_HEAD
|
||||
MYSQL connection;
|
||||
int open;
|
||||
} _mysql_ConnectionObject;
|
||||
|
||||
extern PyTypeObject _mysql_ConnectionObject_Type;
|
||||
|
||||
#define MySQL_Exception(e,c) PyErr_SetString(e, mysql_error(&(c->connection)));
|
||||
|
||||
static PyObject *
|
||||
_mysql_connect(self, args, kwargs)
|
||||
PyObject *self;
|
||||
PyObject *args;
|
||||
PyObject *kwargs;
|
||||
{
|
||||
MYSQL *conn;
|
||||
char *host = NULL, *user = NULL, *passwd = NULL,
|
||||
*db = NULL, *unix_socket = NULL;
|
||||
uint port = 3306;
|
||||
uint client_flag = 0;
|
||||
static char *kwlist[] = { "host", "user", "passwd", "db", "port",
|
||||
"unix_socket", "client_flag",
|
||||
NULL } ;
|
||||
_mysql_ConnectionObject *c = PyObject_NEW(_mysql_ConnectionObject,
|
||||
&_mysql_ConnectionObject_Type);
|
||||
if (c == NULL) return NULL;
|
||||
c->open = 0;
|
||||
if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|ssssisi:connect",
|
||||
kwlist,
|
||||
&host, &user, &passwd, &db,
|
||||
&port, &unix_socket, &client_flag))
|
||||
return NULL;
|
||||
Py_BEGIN_ALLOW_THREADS ;
|
||||
conn = mysql_real_connect(&(c->connection), host, user, passwd, db,
|
||||
port, unix_socket, client_flag);
|
||||
Py_END_ALLOW_THREADS ;
|
||||
if (!conn) {
|
||||
MySQL_Exception(_mysql_Error, c);
|
||||
Py_DECREF(c);
|
||||
return NULL;
|
||||
}
|
||||
c->open = 1;
|
||||
return (PyObject *) c;
|
||||
}
|
||||
|
||||
static PyObject *
|
||||
_mysql_ConnectionObject_close(c, args)
|
||||
_mysql_ConnectionObject *c;
|
||||
PyObject *args;
|
||||
{
|
||||
if (!PyArg_NoArgs(args)) return NULL;
|
||||
Py_BEGIN_ALLOW_THREADS
|
||||
mysql_close(&(c->connection));
|
||||
Py_END_ALLOW_THREADS
|
||||
c->open = 0;
|
||||
Py_INCREF(Py_None);
|
||||
return Py_None;
|
||||
}
|
||||
|
||||
static PyObject *
|
||||
_mysql_ConnectionObject_affected_rows(c, args)
|
||||
_mysql_ConnectionObject *c;
|
||||
PyObject *args;
|
||||
{
|
||||
if (!PyArg_NoArgs(args)) return NULL;
|
||||
return PyLong_FromUnsignedLongLong(mysql_affected_rows(&(c->connection)));
|
||||
}
|
||||
|
||||
/* static PyObject * */
|
||||
/* _mysql_ConnectionObject_seek(c, args) */
|
||||
/* _mysql_ConnectionObject *c; */
|
||||
/* PyObject *args; */
|
||||
/* { */
|
||||
/* int offset; */
|
||||
/* if (!PyArg_ParseTuple(args, "u", &offset)) return NULL; */
|
||||
/* mysql_data_seek(&(c->connection), (unsigned int)offset); */
|
||||
/* Py_INCREF(Py_None); */
|
||||
/* return Py_None; */
|
||||
/* } */
|
||||
|
||||
static PyObject *
|
||||
_mysql_debug(self, args)
|
||||
PyObject *self;
|
||||
PyObject *args;
|
||||
{
|
||||
char *debug;
|
||||
if (!PyArg_ParseTuple(args, "s", &debug)) return NULL;
|
||||
mysql_debug(debug);
|
||||
Py_INCREF(Py_None);
|
||||
return Py_None;
|
||||
}
|
||||
|
||||
static PyObject *
|
||||
_mysql_ConnectionObject_dump_debug_info(c, args)
|
||||
_mysql_ConnectionObject *c;
|
||||
PyObject *args;
|
||||
{
|
||||
int err;
|
||||
if (!PyArg_NoArgs(args)) return NULL;
|
||||
Py_BEGIN_ALLOW_THREADS
|
||||
err = mysql_dump_debug_info(&(c->connection));
|
||||
Py_END_ALLOW_THREADS
|
||||
if (err) {
|
||||
MySQL_Exception(_mysql_Error, c);
|
||||
return NULL;
|
||||
}
|
||||
Py_INCREF(Py_None);
|
||||
return Py_None;
|
||||
}
|
||||
|
||||
static PyObject *
|
||||
_mysql_ConnectionObject_errno(c, args)
|
||||
_mysql_ConnectionObject *c;
|
||||
PyObject *args;
|
||||
{
|
||||
if (!PyArg_NoArgs(args)) return NULL;
|
||||
return PyInt_FromLong((long)mysql_errno(&(c->connection)));
|
||||
}
|
||||
|
||||
static PyObject *
|
||||
_mysql_ConnectionObject_error(c, args)
|
||||
_mysql_ConnectionObject *c;
|
||||
PyObject *args;
|
||||
{
|
||||
if (!PyArg_NoArgs(args)) return NULL;
|
||||
return PyString_FromString(mysql_error(&(c->connection)));
|
||||
}
|
||||
|
||||
static PyObject *
|
||||
_mysql_escape(self, args)
|
||||
PyObject *self;
|
||||
PyObject *args;
|
||||
{
|
||||
PyObject *str;
|
||||
char *in, *out;
|
||||
int len, size;
|
||||
PyObject *o;
|
||||
if (!PyArg_ParseTuple(args, "s#:escape", &in, &size)) return NULL;
|
||||
str = PyString_FromStringAndSize((char *) NULL, size*2+1);
|
||||
if (!str) return PyErr_NoMemory();
|
||||
out = PyString_AS_STRING(str);
|
||||
len = mysql_escape_string(out, in, size);
|
||||
if (_PyString_Resize(&str, len) < 0) return NULL;
|
||||
return (str);
|
||||
}
|
||||
|
||||
static PyObject *
|
||||
_mysql_get_client_info(self, args)
|
||||
PyObject *self;
|
||||
PyObject *args;
|
||||
{
|
||||
if (!PyArg_ParseTuple(args, ":get_client_info")) return NULL;
|
||||
return PyString_FromString(mysql_get_client_info());
|
||||
}
|
||||
|
||||
static PyObject *
|
||||
_mysql_ConnectionObject_get_host_info(c)
|
||||
_mysql_ConnectionObject *c;
|
||||
{
|
||||
return PyString_FromString(mysql_get_host_info(&(c->connection)));
|
||||
}
|
||||
|
||||
static PyObject *
|
||||
_mysql_ConnectionObject_get_proto_info(c)
|
||||
_mysql_ConnectionObject *c;
|
||||
{
|
||||
return PyInt_FromLong((long)mysql_get_proto_info(&(c->connection)));
|
||||
}
|
||||
|
||||
static PyObject *
|
||||
_mysql_ConnectionObject_get_server_info(c)
|
||||
_mysql_ConnectionObject *c;
|
||||
{
|
||||
return PyString_FromString(mysql_get_server_info(&(c->connection)));
|
||||
}
|
||||
|
||||
static PyObject *
|
||||
_mysql_ConnectionObject_info(c)
|
||||
_mysql_ConnectionObject *c;
|
||||
{
|
||||
char *s;
|
||||
s = mysql_info(&(c->connection));
|
||||
if (s) return PyString_FromString(s);
|
||||
Py_INCREF(Py_None);
|
||||
return Py_None;
|
||||
}
|
||||
|
||||
static PyObject *
|
||||
_mysql_ConnectionObject_insert_id(c, args)
|
||||
_mysql_ConnectionObject *c;
|
||||
PyObject *args;
|
||||
{
|
||||
my_ulonglong r;
|
||||
if (!PyArg_NoArgs(args)) return NULL;
|
||||
Py_BEGIN_ALLOW_THREADS
|
||||
r = mysql_insert_id(&(c->connection));
|
||||
Py_END_ALLOW_THREADS
|
||||
return PyLong_FromUnsignedLongLong(r);
|
||||
}
|
||||
|
||||
static PyObject *
|
||||
_mysql_ConnectionObject_kill(c, args)
|
||||
_mysql_ConnectionObject *c;
|
||||
PyObject *args;
|
||||
{
|
||||
unsigned long pid;
|
||||
int r;
|
||||
if (!PyArg_ParseTuple(args, "i:kill", &pid)) return NULL;
|
||||
Py_BEGIN_ALLOW_THREADS
|
||||
r = mysql_kill(&(c->connection), pid);
|
||||
Py_END_ALLOW_THREADS
|
||||
if (r) {
|
||||
MySQL_Exception(_mysql_Error, c);
|
||||
return NULL;
|
||||
}
|
||||
Py_INCREF(Py_None);
|
||||
return Py_None;
|
||||
}
|
||||
|
||||
static PyObject *
|
||||
_mysql_ConnectionObject_ping(c, args)
|
||||
_mysql_ConnectionObject *c;
|
||||
PyObject *args;
|
||||
{
|
||||
int r;
|
||||
if (!PyArg_ParseTuple(args, ":ping")) return NULL;
|
||||
Py_BEGIN_ALLOW_THREADS
|
||||
r = mysql_ping(&(c->connection));
|
||||
Py_END_ALLOW_THREADS
|
||||
if (r) {
|
||||
MySQL_Exception(_mysql_Error, c);
|
||||
return NULL;
|
||||
}
|
||||
Py_INCREF(Py_None);
|
||||
return Py_None;
|
||||
}
|
||||
|
||||
static PyObject *
|
||||
_mysql_ConnectionObject_query(c, args)
|
||||
_mysql_ConnectionObject *c;
|
||||
PyObject *args;
|
||||
{
|
||||
char *query;
|
||||
int len, r;
|
||||
if (!PyArg_ParseTuple(args, "s#:query", &query, &len)) return NULL;
|
||||
Py_BEGIN_ALLOW_THREADS
|
||||
r = mysql_real_query(&(c->connection), query, len);
|
||||
Py_END_ALLOW_THREADS
|
||||
if (r) {
|
||||
MySQL_Exception(_mysql_Error, c);
|
||||
return NULL;
|
||||
}
|
||||
Py_INCREF(Py_None);
|
||||
return Py_None;
|
||||
}
|
||||
|
||||
static PyObject *
|
||||
_mysql_ConnectionObject_row_tell(c, args)
|
||||
_mysql_ConnectionObject *c;
|
||||
PyObject *args;
|
||||
{
|
||||
unsigned int r;
|
||||
if (!PyArg_NoArgs(args)) return NULL;
|
||||
Py_BEGIN_ALLOW_THREADS
|
||||
r = mysql_insert_id(&(c->connection));
|
||||
Py_END_ALLOW_THREADS
|
||||
return PyInt_FromLong((long)r);
|
||||
}
|
||||
|
||||
static PyObject *
|
||||
_mysql_ConnectionObject_select_db(c, args)
|
||||
_mysql_ConnectionObject *c;
|
||||
PyObject *args;
|
||||
{
|
||||
char *db;
|
||||
int r;
|
||||
if (!PyArg_ParseTuple(args, "s:select_db", &db)) return NULL;
|
||||
Py_BEGIN_ALLOW_THREADS
|
||||
r = mysql_select_db(&(c->connection), db);
|
||||
Py_END_ALLOW_THREADS
|
||||
if (r) {
|
||||
MySQL_Exception(_mysql_Error, c);
|
||||
return NULL;
|
||||
}
|
||||
Py_INCREF(Py_None);
|
||||
return Py_None;
|
||||
}
|
||||
|
||||
static PyObject *
|
||||
_mysql_ConnectionObject_shutdown(c, args)
|
||||
_mysql_ConnectionObject *c;
|
||||
PyObject *args;
|
||||
{
|
||||
int r;
|
||||
if (!PyArg_ParseTuple(args, ":shutdown")) return NULL;
|
||||
Py_BEGIN_ALLOW_THREADS
|
||||
r = mysql_shutdown(&(c->connection));
|
||||
Py_END_ALLOW_THREADS
|
||||
if (r) {
|
||||
MySQL_Exception(_mysql_Error, c);
|
||||
return NULL;
|
||||
}
|
||||
Py_INCREF(Py_None);
|
||||
return Py_None;
|
||||
}
|
||||
|
||||
static PyObject *
|
||||
_mysql_ConnectionObject_thread_id(c, args)
|
||||
_mysql_ConnectionObject *c;
|
||||
PyObject *args;
|
||||
{
|
||||
unsigned long pid;
|
||||
if (!PyArg_NoArgs(args)) return NULL;
|
||||
Py_BEGIN_ALLOW_THREADS
|
||||
pid = mysql_thread_id(&(c->connection));
|
||||
Py_END_ALLOW_THREADS
|
||||
return PyLong_FromUnsignedLong(pid);
|
||||
}
|
||||
|
||||
static void
|
||||
_mysql_ConnectionObject_dealloc(c)
|
||||
_mysql_ConnectionObject *c;
|
||||
{
|
||||
if (c->open) {
|
||||
Py_BEGIN_ALLOW_THREADS
|
||||
mysql_close(&(c->connection));
|
||||
Py_END_ALLOW_THREADS
|
||||
}
|
||||
PyMem_Free((char *) c);
|
||||
}
|
||||
|
||||
static PyObject *
|
||||
_mysql_ConnectionObject_repr(c)
|
||||
_mysql_ConnectionObject *c;
|
||||
{
|
||||
char buf[300];
|
||||
sprintf(buf, "<%s connection to '%.256s' at %lx>",
|
||||
c->open ? "open" : "closed",
|
||||
c->connection.host,
|
||||
(long)c);
|
||||
return PyString_FromString(buf);
|
||||
}
|
||||
|
||||
static PyMethodDef _mysql_ConnectionObject_methods[] = {
|
||||
{"affected_rows", (PyCFunction)_mysql_ConnectionObject_affected_rows, 0},
|
||||
{"dump_debug_info", (PyCFunction)_mysql_ConnectionObject_dump_debug_info, 0},
|
||||
{"error", (PyCFunction)_mysql_ConnectionObject_error, 0},
|
||||
{"errno", (PyCFunction)_mysql_ConnectionObject_errno, 0},
|
||||
{"get_host_info", (PyCFunction)_mysql_ConnectionObject_get_host_info, 0},
|
||||
{"get_proto_info", (PyCFunction)_mysql_ConnectionObject_get_proto_info, 0},
|
||||
{"get_server_info", (PyCFunction)_mysql_ConnectionObject_get_server_info, 0},
|
||||
{"info", (PyCFunction)_mysql_ConnectionObject_info, 0},
|
||||
{NULL, NULL} /* sentinel */
|
||||
};
|
||||
|
||||
static struct memberlist _mysql_ConnectionObject_memberlist[] = {
|
||||
{"open", T_INT, 0, RO},
|
||||
{"closed", T_INT, 0, RO},
|
||||
{NULL} /* Sentinel */
|
||||
};
|
||||
|
||||
static PyObject *
|
||||
_mysql_ConnectionObject_getattr(c, name)
|
||||
_mysql_ConnectionObject *c;
|
||||
char *name;
|
||||
{
|
||||
PyObject *res;
|
||||
|
||||
res = Py_FindMethod(_mysql_ConnectionObject_methods, (PyObject *)c, name);
|
||||
if (res != NULL)
|
||||
return res;
|
||||
PyErr_Clear();
|
||||
if (strcmp(name, "open") == 0)
|
||||
return PyInt_FromLong((long)(c->open));
|
||||
if (strcmp(name, "closed") == 0)
|
||||
return PyInt_FromLong((long)!(c->open));
|
||||
return PyMember_Get((char *)c, _mysql_ConnectionObject_memberlist, name);
|
||||
}
|
||||
|
||||
static int
|
||||
_mysql_ConnectionObject_setattr(c, name, v)
|
||||
_mysql_ConnectionObject *c;
|
||||
char *name;
|
||||
PyObject *v;
|
||||
{
|
||||
if (v == NULL) {
|
||||
PyErr_SetString(PyExc_AttributeError,
|
||||
"can't delete file attributes");
|
||||
return -1;
|
||||
}
|
||||
return PyMember_Set((char *)c, _mysql_ConnectionObject_memberlist, name, v);
|
||||
}
|
||||
|
||||
PyTypeObject _mysql_ConnectionObject_Type = {
|
||||
PyObject_HEAD_INIT(&PyType_Type)
|
||||
0,
|
||||
"connection",
|
||||
sizeof(_mysql_ConnectionObject),
|
||||
0,
|
||||
(destructor)_mysql_ConnectionObject_dealloc, /* tp_dealloc */
|
||||
0, /*tp_print*/
|
||||
(getattrfunc)_mysql_ConnectionObject_getattr, /* tp_getattr */
|
||||
0, /* tp_setattr */
|
||||
0, /*tp_compare*/
|
||||
(reprfunc)_mysql_ConnectionObject_repr, /* tp_repr */
|
||||
};
|
||||
|
||||
static PyMethodDef
|
||||
_mysql_methods[] = {
|
||||
{ "connect", _mysql_connect, METH_VARARGS | METH_KEYWORDS },
|
||||
{ "escape", _mysql_escape, METH_VARARGS },
|
||||
{ "get_client_info", _mysql_get_client_info, METH_VARARGS },
|
||||
{NULL, NULL} /* sentinel */
|
||||
};
|
||||
|
||||
static PyObject *
|
||||
_mysql_NewException(PyObject *dict,
|
||||
char *name,
|
||||
PyObject *base)
|
||||
{
|
||||
PyObject *v;
|
||||
char longname[256];
|
||||
|
||||
sprintf(longname, "_mysql.%s", name);
|
||||
if ((v = PyErr_NewException(longname, base, NULL)) == NULL)
|
||||
return NULL;
|
||||
if (PyDict_SetItemString(dict, name, v)) return NULL;
|
||||
return v;
|
||||
}
|
||||
|
||||
void
|
||||
init_mysql()
|
||||
{
|
||||
PyObject *dict, *module;
|
||||
module = Py_InitModule("_mysql", _mysql_methods);
|
||||
|
||||
dict = PyModule_GetDict(module);
|
||||
if ((_mysql_Error = _mysql_NewException(dict, "error", NULL)) == NULL) goto error;
|
||||
error:
|
||||
if (PyErr_Occurred())
|
||||
PyErr_SetString(PyExc_ImportError,
|
||||
"_mysql: init failed");
|
||||
return;
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user