SQLAlchemy操作PostgreSQL时datetime列删除报错：时区时间比较异常

我最近在Docker的Python3.10环境（镜像python:3.10-slim）里，用psycopg2连接PostgreSQL，通过SQLAlchemy做数据删除时遇到了一个头疼的偶发问题：有时候会抛出TypeError: can't compare offset-naive and offset-aware datetimes错误，但有时候又完全正常。

我的操作流程 #

首先在模型里定义了一个无时区的DateTime字段：

class VolumeInfo(Base):
    ...
    date: datetime.datetime = Column(DateTime, nullable=False)

然后写了删除指定天数前数据的逻辑：

days_interval = 10
to_date = datetime.datetime.combine(
            datetime.datetime.utcnow().date(),
            datetime.time(0, 0, 0, 0),
        ).replace(tzinfo=None)
from_date = to_date - datetime.timedelta(days=days_interval)

query = delete(VolumeInfo).where(VolumeInfo.date < from_date)
db.execute(query)

偶发的错误栈 #

Traceback (most recent call last):
  ...
  File "script.py", line 381, in delete_volumes
    db.execute(query)
  File "/usr/local/lib/python3.10/site-packages/sqlalchemy/orm/session.py", line 1660, in execute
    ) = compile_state_cls.orm_pre_session_exec(
  File "/usr/local/lib/python3.10/site-packages/sqlalchemy/orm/persistence.py", line 1843, in orm_pre_session_exec
    update_options = cls._do_pre_synchronize_evaluate(
  File "/usr/local/lib/python3.10/site-packages/sqlalchemy/orm/persistence.py", line 2007, in _do_pre_synchronize_evaluate
    matched_objects = [
  File "/usr/local/lib/python3.10/site-packages/sqlalchemy/orm/persistence.py", line 2012, in <listcomp>
    and eval_condition(state.obj())
  File "/usr/local/lib/python3.10/site-packages/sqlalchemy/orm/evaluator.py", line 211, in evaluate
    return operator(eval_left(obj), eval_right(obj))
TypeError: can't compare offset-naive and offset-aware datetimes

问题根源分析 #

这个错误偶发的关键在于SQLAlchemy的ORM预评估机制：当执行delete/update操作时，如果当前会话中已经加载了符合过滤条件的VolumeInfo对象，SQLAlchemy会先在Python内存里对这些对象做条件校验；如果会话里没有这些对象，就直接把SQL发给数据库执行。

问题就出在内存校验这一步：如果数据库返回的date字段被psycopg2转换成了**带时区（offset-aware）的datetime对象，而我们代码里的from_date是无时区（offset-naive）**的，两者在Python层面无法直接比较，就会触发这个错误。而当会话里没有加载对应对象时，数据库本身的datetime比较逻辑是正常的，所以不会报错。

几个可行的解决方案 #

方案1：直接禁用ORM预评估（最快解决）

在执行delete时，添加synchronize_session=False参数，告诉SQLAlchemy跳过内存中的对象校验，直接把SQL发送给数据库处理：

# 方式1：在execute时指定
db.execute(query, synchronize_session=False)

# 方式2：在构建delete语句时指定执行选项
query = delete(VolumeInfo).where(VolumeInfo.date < from_date).execution_options(synchronize_session=False)
db.execute(query)

这样所有的条件判断都交给PostgreSQL处理，数据库本身能正确处理无时区datetime的比较，彻底避开Python层面的类型冲突。

方案2：统一时区处理（最规范）

从根源上避免时区不一致的问题，建议统一使用带时区的datetime：

1. 修改模型字段为带时区的类型：

from sqlalchemy import DateTime, func

class VolumeInfo(Base):
    ...
    date: datetime.datetime = Column(DateTime(timezone=True), nullable=False, default=func.now())

2. 代码中使用带UTC时区的datetime对象：

import pytz

days_interval = 10
utc_now = datetime.datetime.now(pytz.utc)
to_date = utc_now.replace(hour=0, minute=0, second=0, microsecond=0)
from_date = to_date - datetime.timedelta(days=days_interval)

query = delete(VolumeInfo).where(VolumeInfo.date < from_date)
db.execute(query)

这样不管是数据库存储还是代码中的datetime，都是带UTC时区的对象，就不会出现类型不匹配的比较问题了。

方案3：强制psycopg2返回无时区datetime

如果你的数据库字段确实是无时区类型，但psycopg2偶尔会返回带时区的对象，可以通过连接参数强制psycopg2返回无时区的UTC时间：

from sqlalchemy import create_engine

engine = create_engine(
    "postgresql+psycopg2://user:password@host/dbname",
    connect_args={"options": "-c timezone=UTC"}
)

这样psycopg2会把数据库中的datetime转换成无时区的UTC时间，和代码里的from_date类型一致，就不会触发比较错误了。

内容的提问来源于stack exchange，提问作者not_alex