如何让适配PostgreSQL的SQLAlchemy实体兼容SQLite dialect？

问题场景 #

你编写了适配PostgreSQL的SQLAlchemy实体，使用server_default=func.clock_timestamp()设置创建时间：

row_created = sa.Column('row_created_', sa.DateTime(timezone=True), server_default=func.clock_timestamp(), nullable=False)

但切换到SQLite时触发报错：

sqlalchemy.exc.OperationalError: (sqlite3.OperationalError) unknown function: clock_timestamp()

可行解决方案 #

1. 方言适配改写函数（推荐）

利用SQLAlchemy的编译扩展，为SQLite单独改写clock_timestamp()的生成逻辑，保持模型定义统一：

from sqlalchemy import func
from sqlalchemy.ext.compiler import compiles

# 为SQLite编译func.clock_timestamp时替换为CURRENT_TIMESTAMP
@compiles(func.clock_timestamp, "sqlite")
def compile_clock_timestamp_sqlite(element, compiler, **kw):
    return "CURRENT_TIMESTAMP"

# 原列定义无需修改
row_created = sa.Column('row_created_', sa.DateTime(timezone=True), server_default=func.clock_timestamp(), nullable=False)

PostgreSQL会正常生成clock_timestamp()，SQLite则自动替换为支持的CURRENT_TIMESTAMP，完美兼容。

2. 条件式定义列属性

如果需要针对不同数据库做更灵活的列配置，可以根据方言名称判断后定义列：

from sqlalchemy import create_engine, Column, DateTime, Integer, func
from sqlalchemy.ext.declarative import declarative_base

Base = declarative_base()
# 替换为实际使用的数据库连接串
engine = create_engine("sqlite:///your_db.db")

class YourModel(Base):
    __tablename__ = "your_table"
    id = Column(Integer, primary_key=True)
    
    if engine.dialect.name == "postgresql":
        row_created = Column('row_created_', DateTime(timezone=True), server_default=func.clock_timestamp(), nullable=False)
    else:
        # SQLite使用CURRENT_TIMESTAMP，或func.current_timestamp()
        row_created = Column('row_created_', DateTime(timezone=True), server_default=func.current_timestamp(), nullable=False)

这种方式适合列属性差异较大的场景，但需要提前确定使用的方言。

3. 为SQLite注册自定义函数

直接让SQLite识别clock_timestamp()函数，无需修改模型：

import sqlite3
from datetime import datetime, timezone
from sqlalchemy import create_engine

# 定义clock_timestamp的实现，对齐PostgreSQL的带时区返回值
def sqlite_clock_timestamp():
    return datetime.now(timezone.utc).isoformat()

# 创建引擎后注册函数
engine = create_engine("sqlite:///your_db.db")
with engine.connect() as conn:
    # 注册0参数的clock_timestamp函数
    conn.connection.create_function("clock_timestamp", 0, sqlite_clock_timestamp)

这样SQLite执行时会调用自定义的函数实现，避免报错，返回值尽量对齐PostgreSQL的带时区时间格式。

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