SQLAlchemy 中的跟踪插入

Question

我正在使用 Flask-SQLAlchemy 在 MySQL 数据库中写入条目。这些条目有一个特定的 id 字段，由 MySQL 自动填充。我想追踪。

让我们看看我的真正意思。我想首先写下我收到的 API 查询。我写了一切，但没有写回复，因为我没有。然后，一旦收到响应，我想获取插入的行并写入响应文件：

模型：

class QueriesDB(db.Model):

    __tablename__ = 'queries'

    id_query = db.Column(INTEGER(unsigned=True), primary_key=True, nullable=False, autoincrement=True)

    id_user = db.Column(INTEGER(unsigned=True), db.ForeignKey(UsersDB.id_user), nullable=True)

    datetime = db.Column(db.TIMESTAMP, nullable=False, default=datetime.utcnow)

    IP = db.Column(db.String(39), nullable=False)

    header = db.Column(LONGTEXT, nullable=False)

    response = db.Column(LONGTEXT, nullable=True)

    answered = db.Column(db.Boolean, nullable=False)

我如何插入它：

valid_query = QueriesDB(id_user=self.id_user, IP=self.request.remote_addr,

                        header=str(dict(self.request.headers)), answered=False)

db.session.add(valid_query)

db.session.commit()

现在我想通过使用 id_query 来获取该特定行，并写入answer = True 以及响应正文。但不确定最好的方法是什么，因为我实际上不知道 id_query。

有没有一种简单的方法可以在执行 commit() 时保留 id_queries？

当然，这可能是一个设计问题。你怎么看？

Answer 1

你应该使用db.session.flush(). https://docs.sqlalchemy.org/en/13/orm/session_api.html#sqlalchemy.orm.session.Session.flushvalid_query = QueriesDB(id_user=self.id_user, IP=self.request.remote_addr,                        header=str(dict(self.request.headers)), answered=False)db.session.add(valid_query)db.session.flush()# you can get autoincremented number from DB print(valid_query.id_query)db.session.commit()