美文网首页
解决 sqlalchemy 默认设置的当前时间戳不更新问题

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

作者: richardlee | 来源:发表于2021-07-29 22:55 被阅读0次

问题背景

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

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_timeupdate_time 字段在新增数据时默认写入当前时间戳,给这两个字段设定了默认值 int(time.time()) ,但这种定义方式会导致所有的数据都是同样的时间戳。

解决方法:

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

传入常量的方式:

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

传入函数的方式:

# 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()) 实际是传入了应用启动时的时间戳,是一个常量,所以每条数据的时间戳是一样的。

对表结构做如下修改,问题解决:

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

相关文章

网友评论

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

      本文链接:https://www.haomeiwen.com/subject/hyzfvltx.html