gf-core/src/runtime/python/transactions.c

#define PY_SSIZE_T_CLEAN
#include <Python.h>
#include <structmember.h>

#include <pgf/pgf.h>
#include "./expr.h"
#include "./ffi.h"
#include "./transactions.h"

TransactionObject *
PGF_newTransaction(PGFObject *self, PyObject *args)
{
    PgfText *name = NULL;
    const char *s = NULL;
    Py_ssize_t size;
    if (!PyArg_ParseTuple(args, "|s#", &s, &size))
        return NULL;

    if (s != NULL) {
        name = (PgfText *)PyMem_Malloc(sizeof(PgfText)+size+1);
        memcpy(name->text, s, size+1);
        name->size = size;
    }

    PgfExn err;
    PgfRevision rev = pgf_clone_revision(self->db, self->revision, name, &err);
    if (name != NULL) {
        PyMem_Free(name);
    }
    if (handleError(err) != PGF_EXN_NONE) {
        return NULL;
    }

    TransactionObject *trans = (TransactionObject *)pgf_TransactionType.tp_alloc(&pgf_TransactionType, 0);
    trans->pgf = self;
    trans->revision = rev;

    return trans;
}

static PyObject *
Transaction_commit(TransactionObject *self, PyObject *args)
{
    PgfExn err;
    pgf_commit_revision(self->pgf->db, self->revision, &err);
    if (handleError(err) != PGF_EXN_NONE) {
        return NULL;
    }

    pgf_free_revision(self->pgf->db, self->pgf->revision);
    self->pgf->revision = self->revision;

    Py_RETURN_TRUE;
}

static PyObject *
Transaction_createFunction(TransactionObject *self, PyObject *args)
{
    const char *s;
    Py_ssize_t size;
    TypeObject *type;
    Py_ssize_t arity = 0;
    float prob = 0.0;
    if (!PyArg_ParseTuple(args, "s#O!nf", &s, &size, &pgf_TypeType, &type, &arity, &prob))
        return NULL;

    PgfText *fname = (PgfText *)PyMem_Malloc(sizeof(PgfText)+size+1);
    memcpy(fname->text, s, size+1);
    fname->size = size;

    PgfExn err;
    pgf_create_function(self->pgf->db, self->revision, fname, (PgfType) type, arity, prob, &marshaller, &err);
    PyMem_Free(fname);
    if (handleError(err) != PGF_EXN_NONE) {
        return NULL;
    }

    Py_RETURN_NONE;
}

static PyObject *
Transaction_dropFunction(TransactionObject *self, PyObject *args)
{
    const char *s;
    Py_ssize_t size;
    if (!PyArg_ParseTuple(args, "s#", &s, &size))
        return NULL;

    PgfText *fname = (PgfText *)PyMem_Malloc(sizeof(PgfText)+size+1);
    memcpy(fname->text, s, size+1);
    fname->size = size;

    PgfExn err;
    pgf_drop_function(self->pgf->db, self->revision, fname, &err);
    PyMem_Free(fname);
    if (handleError(err) != PGF_EXN_NONE) {
        return NULL;
    }

    Py_RETURN_NONE;
}

static PyObject *
Transaction_createCategory(TransactionObject *self, PyObject *args)
{
    const char *s;
    Py_ssize_t size;
    PyObject *hypos;
    float prob = 0.0;
    if (!PyArg_ParseTuple(args, "s#O!f", &s, &size, &PyList_Type, &hypos, prob))
        return NULL;

    PgfText *catname = (PgfText *)PyMem_Malloc(sizeof(PgfText)+size+1);
    memcpy(catname->text, s, size+1);
    catname->size = size;

    Py_ssize_t n_hypos = PyList_Size(hypos);
    // PgfTypeHypo context[n_hypos];
    PgfTypeHypo *context = alloca(sizeof(PgfTypeHypo)*n_hypos);
    for (Py_ssize_t i = 0; i < n_hypos; i++) {
        PyObject *hytup = (PyObject *)PyList_GetItem(hypos, i);
        context[i].bind_type = PyLong_AsLong(PyTuple_GetItem(hytup, 0));
        context[i].cid = PyUnicode_AsPgfText(PyTuple_GetItem(hytup, 1));
        context[i].type = (PgfType) PyTuple_GetItem(hytup, 2);
        Py_INCREF(context[i].type);
    }

    PgfExn err;
    pgf_create_category(self->pgf->db, self->revision, catname, n_hypos, context, prob, &marshaller, &err);
    PyMem_Free(catname);
    if (handleError(err) != PGF_EXN_NONE) {
        return NULL;
    }

    Py_RETURN_NONE;
}

static PyObject *
Transaction_dropCategory(TransactionObject *self, PyObject *args)
{
    const char *s;
    Py_ssize_t size;
    if (!PyArg_ParseTuple(args, "s#", &s, &size))
        return NULL;

    PgfText *catname = (PgfText *)PyMem_Malloc(sizeof(PgfText)+size+1);
    memcpy(catname->text, s, size+1);
    catname->size = size;

    PgfExn err;
    pgf_drop_category(self->pgf->db, self->revision, catname, &err);
    PyMem_Free(catname);
    if (handleError(err) != PGF_EXN_NONE) {
        return NULL;
    }

    Py_RETURN_NONE;
}

static TransactionObject *
Transaction_enter(TransactionObject *self, PyObject *Py_UNUSED(ignored))
{
    return self;
}

static PyObject *
Transaction_exit(TransactionObject *self, PyObject *const *args, Py_ssize_t nargs)
{
    // PyObject *exc_type = Py_None;
    // PyObject *exc_value = Py_None;
    // PyObject *exc_tb = Py_None;

    // if (!_PyArg_CheckPositional("__exit__", nargs, 0, 3)) {
    //     Py_RETURN_FALSE;
    // }
    if (nargs < 1) {
        goto skip_optional;
    }
    // exc_type = args[0];
    // if (nargs < 2) {
    //     goto skip_optional;
    // }
    // exc_value = args[1];
    // if (nargs < 3) {
    //     goto skip_optional;
    // }
    // exc_tb = args[2];
skip_optional:
    // TODO check exception

    return Transaction_commit(self, NULL);
}

// static void
// Transaction_dealloc(PGFObject* self)
// {
//     Py_TYPE(self)->tp_free((PyObject*)self);
// }

static PyGetSetDef Transaction_getseters[] = {
    {NULL}  /* Sentinel */
};

static PyMemberDef Transaction_members[] = {
    {NULL}  /* Sentinel */
};

static PyMethodDef Transaction_methods[] = {
    {"commit", (PyCFunction)Transaction_commit, METH_VARARGS,
     "Commit transaction"
    },

    {"__enter__", (PyCFunction)Transaction_enter, METH_NOARGS,
     ""
    },

    {"__exit__", (PyCFunction)(void(*)(void))Transaction_exit, METH_FASTCALL,
     ""
    },

    {"createFunction", (PyCFunction)Transaction_createFunction, METH_VARARGS,
     "Create function"
    },
    {"dropFunction", (PyCFunction)Transaction_dropFunction, METH_VARARGS,
     "Drop function"
    },
    {"createCategory", (PyCFunction)Transaction_createCategory, METH_VARARGS,
     "Create category"
    },
    {"dropCategory", (PyCFunction)Transaction_dropCategory, METH_VARARGS,
     "Drop category"
    },
    {NULL}  /* Sentinel */
};

PyTypeObject pgf_TransactionType = {
    PyVarObject_HEAD_INIT(NULL, 0)
    //0,                         /*ob_size*/
    "pgf.Transaction",                 /*tp_name*/
    sizeof(TransactionObject),         /*tp_basicsize*/
    0,                         /*tp_itemsize*/
    0, //(destructor)Transaction_dealloc,   /*tp_dealloc*/
    0,                         /*tp_print*/
    0,                         /*tp_getattr*/
    0,                         /*tp_setattr*/
    0,                         /*tp_compare*/
    0,                         /*tp_repr*/
    0,                         /*tp_as_number*/
    0,                         /*tp_as_sequence*/
    0,                         /*tp_as_mapping*/
    0,                         /*tp_hash */
    0,                         /*tp_call*/
    0, // (reprfunc) Transaction_str,        /*tp_str*/
    0,                         /*tp_getattro*/
    0,                         /*tp_setattro*/
    0,                         /*tp_as_buffer*/
    Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /*tp_flags*/
    "Transaction object",              /*tp_doc*/
    0,                           /*tp_traverse */
    0,                           /*tp_clear */
    0,                           /*tp_richcompare */
    0,                           /*tp_weaklistoffset */
    0,                           /*tp_iter */
    0,                           /*tp_iternext */
    Transaction_methods,               /*tp_methods */
    Transaction_members,               /*tp_members */
    Transaction_getseters,             /*tp_getset */
    0,                         /*tp_base */
    0,                         /*tp_dict */
    0,                         /*tp_descr_get */
    0,                         /*tp_descr_set */
    0,                         /*tp_dictoffset */
    0,                         /*tp_init */
    0,                         /*tp_alloc */
    0,                         /*tp_new */
};