python - 在提交对象之前从 SqlAlchemy 获取主键

Question

我一直在将 sqlalchemy 与 postgres 一起使用，但不知道如何解决下面的问题。

我想在提交到数据库之前获取主键的值。

我有一个名为 process 的类，它具有 proc_id 和另一个名为 proc_num 的元素。

class process(db.Model):
    __tablename__ = 'tb_processos'
    proc_id = db.Column(db.Integer, primary_key=True, nullable=False,
                        autoincrement=True, unique=True)
    proc_num = db.Column(db.String(20))

出于我的原因，proc_num 必须是 String 类型。

当我创建对象时，我希望 proc_num 接收 proc_id。

最好的方法是什么？

对不起我的英语:/

谢谢

Answer 1

您可以使用flush() 方法，更改不会永久保存到磁盘，直到您调用commit():

p = process(proc_num=45)
db.session.flush(p)
p = process.query.filter_by(proc_num=45)
p.proc_id

# finally you can call commit
db.session.commit()