diff options
Diffstat (limited to 'psycopg/adapter_datetime.c')
| -rw-r--r-- | psycopg/adapter_datetime.c | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/psycopg/adapter_datetime.c b/psycopg/adapter_datetime.c index deaf59f..24d158c 100644 --- a/psycopg/adapter_datetime.c +++ b/psycopg/adapter_datetime.c @@ -57,6 +57,7 @@ static PyObject * pydatetime_str(pydatetimeObject *self) { if (self->type <= PSYCO_DATETIME_TIMESTAMP) { + PyObject *tz; /* Select the right PG type to cast into. */ char *fmt = NULL; @@ -68,7 +69,10 @@ pydatetime_str(pydatetimeObject *self) fmt = "'%s'::date"; break; case PSYCO_DATETIME_TIMESTAMP: - fmt = "'%s'::timestamp"; + tz = PyObject_GetAttrString(self->wrapped, "tzinfo"); + if (!tz) { return NULL; } + fmt = (tz == Py_None) ? "'%s'::timestamp" : "'%s'::timestamptz"; + Py_DECREF(tz); break; } |