我们通过sqlalchemy
定义模型创建的数据库表,通常会通过ORM的方式操作,但在开发的过程中难免也会遇到不通过ORM而直接操作数据库的情况。这时我们在定义模型时,给字段设置的默认值就不能安装以前的方法了,这里有些坑我先踩过了,在此记录一下已便他人查阅。
比如我们现在想创建一张user表,根据需求几个字段的默认值设置如下:
-
is_vip
字段设置默认值为0 -
create_time
字段默认值设置为该行插入的时间 -
update_time
字段能够在该行改变时自动更新时间
通常我们会这样创建user表:
from sqlalchemy import Column, String, create_engine, Integer, DateTime
from sqlalchemy.sql import func
from sqlalchemy.ext.declarative import declarative_base
# 创建对象的基类:
Base = declarative_base()
class User(Base):
__tablename__ = 'user'
id = Column(String(36), primary_key=True)
name = Column(String(20))
is_vip = Column(Integer, default=0)
create_time = Column(DateTime, nullable=False, default=datetime.datetime.now)
update_time = Column(DateTime, nullable=False, onupdate=True)
# 初始化数据库连接:
engine = create_engine('mysql+pymysql://root:111111@localhost:3306/testdb')
Base.metadata.create_all(engine)
看起来一切都那么完美,没有任何问题,但是我们查看一下生成的DDL:
CREATE TABLE `user` (
`id` int(11) NOT NULL AUTO_INCREMENT,
`name` varchar(20) DEFAULT NULL,
`is_vip` int(11) DEFAULT NULL,
`create_time` datetime NOT NULL,
`update_time` datetime NOT NULL,
PRIMARY KEY (`id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8;
默认值和自动更新都没有设置成功,想要默认值和自动更新生效就只能通过ORM操作数据库。那该怎么办呢?下面揭晓正确的操作姿势。
from sqlalchemy import Column, String, Integer, DateTime, TIMESTAMP
from sqlalchemy.sql import func
class User1(Base):
# 表的名字:
__tablename__ = 'user1'
id = Column(Integer, primary_key=True)
name = Column(String(20))
# 数据库设置默认值 需是字符串
is_vip = Column(Integer, server_default='0')
# 默认值是创建时间 不自动更新
create_time = Column(DateTime, nullable=False, server_default=func.now())
# 自动更新 需要设置nullable=False
update_time = Column(TIMESTAMP, nullable=False)
使用server_default
在数据库中直接生成默认值,需要注意的是Integer
默认值不能直接设置数字,需要是字符串;DateTime
默认值需要使用func.now()
;自动更新的时间戳可以使用TIMESTAMP
,只需设置nullable=False
即可
下面是生成的DDL:
CREATE TABLE `user1` (
`id` int(11) NOT NULL AUTO_INCREMENT,
`name` varchar(20) DEFAULT NULL,
`is_vip` int(11) DEFAULT '0',
`create_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP,
`update_time` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
PRIMARY KEY (`id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8;