Postgresql date_trunc 通过 Django 的 connection.cursor() 在自定义函数中调用答案

【问题标题】：Postgresql date_trunc called in custom function through Django's connection.cursor()Postgresql date_trunc 通过 Django 的 connection.cursor() 在自定义函数中调用
【发布时间】：2014-09-19 10:15:00
【问题描述】：

我创建了一个自定义 Postgresql 函数，该函数使用 date_trunc 函数执行一些计算并确定月初。在开发自定义函数时，我从使用 psql 建立的数据库连接中检查它，它工作得很好。虽然这样的连接我得到以下输出：

select date_trunc('month', TIMESTAMP WITH TIME ZONE '2014-09-14 04:10:00+02');
       date_trunc       
------------------------
 2014-09-01 00:00:00+02
(1 row)

这正是我想要的：它给了我所在时区的月初。如果我尝试在 django shell 中做同样的事情（我使用的是 psycopg2 连接器），我会得到不同的结果：

>>> from django.db import connection
>>> cursor = connection.cursor()
>>> cursor.execute("select date_trunc('month', TIMESTAMP WITH TIME ZONE '2014-09-14 04:10:00+02');")
>>> unicode(cursor.fetchone()[0])
u'2014-09-01 00:00:00+00:00'

如您所见，数据库以 UTC 时间返回月初，导致与我想要的实际时间戳有两个小时的偏移。

这是 django 将数据库连接的时间戳设置为 UTC 的结果，这导致 date_trunc 在 UTC 时间截断月份。我尝试使用SET TIMEZONE TO 'Europe/Berlin' 手动设置连接的时区，但这会导致AssertionError("database connection isn't set to UTC")。

以下解决方法在我的情况下确实有效，但它并不是一个很好的解决方案。它仅在我不需要函数的任何返回值时才有效。

cursor.execute("SET TIMEZONE to 'Europe/Berlin'; my_function(parameters); SET TIMEZONE to 'UTC';")

我希望有人能帮我找出一个更优雅的方法来解决这个问题。

【问题讨论】：

标签： django postgresql django-database django-postgresql django-timezone

【解决方案1】：

在查询中使用 AT TIME ZONE 'FOO' 或 TIMEZONE('FOO', timestamp) （根据需要使用实际时区名称的参数）；见http://www.postgresql.org/docs/9.2/static/functions-datetime.html#FUNCTIONS-DATETIME-ZONECONVERT
为查询创建上下文管理器，用于保存/设置/恢复特定查询周围的时区

【讨论】：