解决 sqlalchemy 默认设置的当前时间戳不更新问题

Posted on | Post modified: | In | Views

问题背景

最近遇到了一个新数据入库时，时间戳不更新的问题。表结构定义如下：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
import time
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Column, BigInteger, String

Base = declarative_base()


class Task(Base):
    __tablename__ = "task"

    id = Column(BigInteger, primary_key=True)
    status = Column(String, nullable=False)
    message = Column(String)
    create_time = Column(BigInteger, nullable=False, default=int(time.time()))
    update_time = Column(BigInteger, nullable=False, default=int(time.time()))

为了实现 create_time 和 update_time 字段在新增数据时默认写入当前时间戳，给这两个字段设定了默认值 int(time.time()) ,但这种定义方式会导致所有的数据都是同样的时间戳。

解决方法：

查询 sqlalchemy 官方文档，得知 default 除了传入常量外，还可以传入 python 函数，新增数据时，会调用传入的 python 函数。

传入常量的方式：

1
2
3
Table("mytable", meta,
    Column("somecolumn", Integer, default=12)
)

传入函数的方式：

1
2
3
4
5
6
7
8
9
10
# a function which counts upwards
i = 0
def mydefault():
    global i
    i += 1
    return i

t = Table("mytable", meta,
    Column('id', Integer, primary_key=True, default=mydefault),
)

default=int(time.time()) 实际是传入了应用启动时的时间戳，是一个常量，所以每条数据的时间戳是一样的。

对表结构做如下修改，问题解决：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
import time
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Column, BigInteger, String

Base = declarative_base()


def current_timestamp():
    return int(time.time())


class Task(Base):
    __tablename__ = "task"

    id = Column(BigInteger, primary_key=True)
    status = Column(String, nullable=False)
    message = Column(String)
    create_time = Column(BigInteger, nullable=False, default=current_timestamp)
    update_time = Column(BigInteger, nullable=False, default=current_timestamp)

参考：

https://www.osgeo.cn/sqlalchemy/core/defaults.html?highlight=default#python-executed-functions