+++ /dev/null
-/* adapter_mxdatetime.c - mx date/time objects
- *
- * Copyright (C) 2003-2004 Federico Di Gregorio <fog@debian.org>
- *
- * This file is part of psycopg.
- *
- * This program is free software; you can redistribute it and/or
- * modify it under the terms of the GNU General Public License
- * as published by the Free Software Foundation; either version 2,
- * or (at your option) any later version.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- * GNU General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
- */
-
-#include <Python.h>
-#include <structmember.h>
-#include <stringobject.h>
-#include <mxDateTime.h>
-#include <string.h>
-
-#define PSYCOPG_MODULE
-#include "psycopg/config.h"
-#include "psycopg/python.h"
-#include "psycopg/psycopg.h"
-#include "psycopg/adapter_mxdatetime.h"
-#include "psycopg/microprotocols_proto.h"
-
-/* the pointer to the mxDateTime API is initialized by the module init code,
- we just need to grab it */
-extern mxDateTimeModule_APIObject *mxDateTimeP;
-
-
-/* mxdatetime_str, mxdatetime_getquoted - return result of quoting */
-
-static PyObject *
-mxdatetime_str(mxdatetimeObject *self)
-{
- PyObject *str = NULL, *res = NULL;
-
- switch (self->type) {
-
- case PSYCO_MXDATETIME_DATE:
- case PSYCO_MXDATETIME_TIMESTAMP:
- str = PyObject_Str(self->wrapped);
-
- /* given the limitation of the mx.DateTime module that uses the same
- type for both date and timestamp values we need to do some black
- magic and make sure we're not using an adapt()ed timestamp as a
- simple date */
- if (strncmp(&(PyString_AsString(str)[11]), "00:00:00.000", 12) == 0) {
- PyObject *tmp =
- PyString_FromStringAndSize(PyString_AsString(str), 10);
- Py_DECREF(str);
- str = tmp;
- }
- break;
-
- case PSYCO_MXDATETIME_TIME:
- case PSYCO_MXDATETIME_INTERVAL:
- str = PyObject_Str(self->wrapped);
-
- /* given the limitation of the mx.DateTime module that uses the same
- type for both time and delta values we need to do some black magic
- and make sure we're not using an adapt()ed interval as a simple
- time */
- if (PyString_Size(str) > 8 && PyString_AsString(str)[8] == ':') {
- mxDateTimeDeltaObject *obj = (mxDateTimeDeltaObject*)self->wrapped;
-
- char buffer[8];
- int i, j, x;
-
- double ss = obj->hour*3600.0 + obj->minute*60.0 + obj->second;
- int us = (int)((ss - floor(ss))*1000000);
-
- for (i=1000000, j=0; i > 0 ; i /= 10) {
- x = us/i;
- us -= x*i;
- buffer[j++] = '0'+x;
- }
- buffer[j] = '\0';
-
- res = PyString_FromFormat("'%ld days %d.%s seconds'",
- obj->day, (int)round(ss), buffer);
- }
- break;
- }
-
- if (str != NULL && res == NULL) {
- res = PyString_FromFormat("'%s'", PyString_AsString(str));
- }
- Py_XDECREF(str);
-
- return res;
-}
-
-PyObject *
-mxdatetime_getquoted(mxdatetimeObject *self, PyObject *args)
-{
- if (!PyArg_ParseTuple(args, "")) return NULL;
- return mxdatetime_str(self);
-}
-
-PyObject *
-mxdatetime_conform(mxdatetimeObject *self, PyObject *args)
-{
- PyObject *res, *proto;
-
- if (!PyArg_ParseTuple(args, "O", &proto)) return NULL;
-
- if (proto == (PyObject*)&isqlquoteType)
- res = (PyObject*)self;
- else
- res = Py_None;
-
- Py_INCREF(res);
- return res;
-}
-
-/** the MxDateTime object **/
-
-/* object member list */
-
-static struct PyMemberDef mxdatetimeObject_members[] = {
- {"adapted", T_OBJECT, offsetof(mxdatetimeObject, wrapped), RO},
- {"type", T_INT, offsetof(mxdatetimeObject, type), RO},
- {NULL}
-};
-
-/* object method table */
-
-static PyMethodDef mxdatetimeObject_methods[] = {
- {"getquoted", (PyCFunction)mxdatetime_getquoted, METH_VARARGS,
- "getquoted() -> wrapped object value as SQL date/time"},
- {"__conform__", (PyCFunction)mxdatetime_conform, METH_VARARGS, NULL},
- {NULL} /* Sentinel */
-};
-
-/* initialization and finalization methods */
-
-static int
-mxdatetime_setup(mxdatetimeObject *self, PyObject *obj, int type)
-{
- Dprintf("mxdatetime_setup: init mxdatetime object at %p, refcnt = %d",
- self, ((PyObject *)self)->ob_refcnt);
-
- self->type = type;
- self->wrapped = obj;
- Py_INCREF(self->wrapped);
-
- Dprintf("mxdatetime_setup: good mxdatetime object at %p, refcnt = %d",
- self, ((PyObject *)self)->ob_refcnt);
- return 0;
-}
-
-static void
-mxdatetime_dealloc(PyObject* obj)
-{
- mxdatetimeObject *self = (mxdatetimeObject *)obj;
-
- Py_XDECREF(self->wrapped);
-
- Dprintf("mxdatetime_dealloc: deleted mxdatetime object at %p, refcnt = %d",
- obj, obj->ob_refcnt);
-
- obj->ob_type->tp_free(obj);
-}
-
-static int
-mxdatetime_init(PyObject *obj, PyObject *args, PyObject *kwds)
-{
- PyObject *mx;
- int type = -1; /* raise an error if type was not passed! */
-
- if (!PyArg_ParseTuple(args, "O|i", &mx, &type))
- return -1;
-
- return mxdatetime_setup((mxdatetimeObject *)obj, mx, type);
-}
-
-static PyObject *
-mxdatetime_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
-{
- return type->tp_alloc(type, 0);
-}
-
-static void
-mxdatetime_del(PyObject* self)
-{
- PyObject_Del(self);
-}
-
-static PyObject *
-mxdatetime_repr(mxdatetimeObject *self)
-{
- return PyString_FromFormat("<psycopg2._psycopg.MxDateTime object at %p>",
- self);
-}
-
-/* object type */
-
-#define mxdatetimeType_doc \
-"MxDateTime(mx, type) -> new mx.DateTime wrapper object"
-
-PyTypeObject mxdatetimeType = {
- PyObject_HEAD_INIT(NULL)
- 0,
- "psycopg2._psycopg.MxDateTime",
- sizeof(mxdatetimeObject),
- 0,
- mxdatetime_dealloc, /*tp_dealloc*/
- 0, /*tp_print*/
- 0, /*tp_getattr*/
- 0, /*tp_setattr*/
-
- 0, /*tp_compare*/
- (reprfunc)mxdatetime_repr, /*tp_repr*/
- 0, /*tp_as_number*/
- 0, /*tp_as_sequence*/
- 0, /*tp_as_mapping*/
- 0, /*tp_hash */
-
- 0, /*tp_call*/
- (reprfunc)mxdatetime_str, /*tp_str*/
- 0, /*tp_getattro*/
- 0, /*tp_setattro*/
- 0, /*tp_as_buffer*/
-
- Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE, /*tp_flags*/
-
- mxdatetimeType_doc, /*tp_doc*/
-
- 0, /*tp_traverse*/
- 0, /*tp_clear*/
-
- 0, /*tp_richcompare*/
- 0, /*tp_weaklistoffset*/
-
- 0, /*tp_iter*/
- 0, /*tp_iternext*/
-
- /* Attribute descriptor and subclassing stuff */
-
- mxdatetimeObject_methods, /*tp_methods*/
- mxdatetimeObject_members, /*tp_members*/
- 0, /*tp_getset*/
- 0, /*tp_base*/
- 0, /*tp_dict*/
-
- 0, /*tp_descr_get*/
- 0, /*tp_descr_set*/
- 0, /*tp_dictoffset*/
-
- mxdatetime_init, /*tp_init*/
- PyType_GenericAlloc, /*tp_alloc*/
- mxdatetime_new, /*tp_new*/
- (freefunc)mxdatetime_del, /*tp_free Low-level free-memory routine */
- 0, /*tp_is_gc For PyObject_IS_GC */
- 0, /*tp_bases*/
- 0, /*tp_mro method resolution order */
- 0, /*tp_cache*/
- 0, /*tp_subclasses*/
- 0 /*tp_weaklist*/
-};
-
-
-/** module-level functions **/
-
-#ifdef PSYCOPG_DEFAULT_MXDATETIME
-
-PyObject *
-psyco_Date(PyObject *self, PyObject *args)
-{
- PyObject *res, *mx;
- int year, month, day;
-
- if (!PyArg_ParseTuple(args, "iii", &year, &month, &day))
- return NULL;
-
- mx = mxDateTimeP->DateTime_FromDateAndTime(year, month, day, 0, 0, 0.0);
- if (mx == NULL) return NULL;
-
- res = PyObject_CallFunction((PyObject *)&mxdatetimeType, "Oi", mx,
- PSYCO_MXDATETIME_DATE);
- Py_DECREF(mx);
- return res;
-}
-
-PyObject *
-psyco_Time(PyObject *self, PyObject *args)
-{
- PyObject *res, *mx;
- int hours, minutes=0;
- double seconds=0.0;
-
- if (!PyArg_ParseTuple(args, "iid", &hours, &minutes, &seconds))
- return NULL;
-
- mx = mxDateTimeP->DateTimeDelta_FromTime(hours, minutes, seconds);
- if (mx == NULL) return NULL;
-
- res = PyObject_CallFunction((PyObject *)&mxdatetimeType, "Oi", mx,
- PSYCO_MXDATETIME_TIME);
- Py_DECREF(mx);
- return res;
-}
-
-PyObject *
-psyco_Timestamp(PyObject *self, PyObject *args)
-{
- PyObject *res, *mx;
- int year, month, day;
- int hour=0, minute=0; /* default to midnight */
- double second=0.0;
-
- if (!PyArg_ParseTuple(args, "lii|iid", &year, &month, &day,
- &hour, &minute, &second))
- return NULL;
-
- mx = mxDateTimeP->DateTime_FromDateAndTime(year, month, day,
- hour, minute, second);
- if (mx == NULL) return NULL;
-
- res = PyObject_CallFunction((PyObject *)&mxdatetimeType, "Oi", mx,
- PSYCO_MXDATETIME_TIMESTAMP);
- Py_DECREF(mx);
- return res;
-}
-
-PyObject *
-psyco_DateFromTicks(PyObject *self, PyObject *args)
-{
- PyObject *res, *mx;
- double ticks;
-
- if (!PyArg_ParseTuple(args,"d", &ticks))
- return NULL;
-
- if (!(mx = mxDateTimeP->DateTime_FromTicks(ticks)))
- return NULL;
-
- res = PyObject_CallFunction((PyObject *)&mxdatetimeType, "Oi", mx,
- PSYCO_MXDATETIME_DATE);
- Py_DECREF(mx);
- return res;
-}
-
-PyObject *
-psyco_TimeFromTicks(PyObject *self, PyObject *args)
-{
- PyObject *res, *mx, *dt;
- double ticks;
-
- if (!PyArg_ParseTuple(args,"d", &ticks))
- return NULL;
-
- if (!(dt = mxDateTimeP->DateTime_FromTicks(ticks)))
- return NULL;
-
- if (!(mx = mxDateTimeP->DateTimeDelta_FromDaysAndSeconds(
- 0, ((mxDateTimeObject*)dt)->abstime)))
- {
- Py_DECREF(dt);
- return NULL;
- }
-
- Py_DECREF(dt);
- res = PyObject_CallFunction((PyObject *)&mxdatetimeType, "Oi", mx,
- PSYCO_MXDATETIME_TIME);
- Py_DECREF(mx);
- return res;
-}
-
-PyObject *
-psyco_TimestampFromTicks(PyObject *self, PyObject *args)
-{
- PyObject *mx, *res;
- double ticks;
-
- if (!PyArg_ParseTuple(args, "d", &ticks))
- return NULL;
-
- if (!(mx = mxDateTimeP->DateTime_FromTicks(ticks)))
- return NULL;
-
- res = PyObject_CallFunction((PyObject *)&mxdatetimeType, "Oi", mx,
- PSYCO_MXDATETIME_TIMESTAMP);
- Py_DECREF(mx);
- return res;
-}
-
-#endif
-
-PyObject *
-psyco_DateFromMx(PyObject *self, PyObject *args)
-{
- PyObject *mx;
-
- if (!PyArg_ParseTuple(args, "O!", mxDateTimeP->DateTime_Type, &mx))
- return NULL;
-
- return PyObject_CallFunction((PyObject *)&mxdatetimeType, "Oi", mx,
- PSYCO_MXDATETIME_DATE);
-}
-
-PyObject *
-psyco_TimeFromMx(PyObject *self, PyObject *args)
-{
- PyObject *mx;
-
- if (!PyArg_ParseTuple(args, "O!", mxDateTimeP->DateTimeDelta_Type, &mx))
- return NULL;
-
- return PyObject_CallFunction((PyObject *)&mxdatetimeType, "Oi", mx,
- PSYCO_MXDATETIME_TIME);
-}
-
-PyObject *
-psyco_TimestampFromMx(PyObject *self, PyObject *args)
-{
- PyObject *mx;
-
- if (!PyArg_ParseTuple(args, "O!", mxDateTimeP->DateTime_Type, &mx))
- return NULL;
-
- return PyObject_CallFunction((PyObject *)&mxdatetimeType, "Oi", mx,
- PSYCO_MXDATETIME_TIMESTAMP);
-}
-
-PyObject *
-psyco_IntervalFromMx(PyObject *self, PyObject *args)
-{
- PyObject *mx;
-
- if (!PyArg_ParseTuple(args, "O!", mxDateTimeP->DateTime_Type, &mx))
- return NULL;
-
- return PyObject_CallFunction((PyObject *)&mxdatetimeType, "Oi", mx,
- PSYCO_MXDATETIME_INTERVAL);
-}