python - tutorial - Obtener la clave insertada antes de la sesión de confirmación

sqlalchemy delete object

Question

class Parent(db.Model):
    id = db.Column(db.Integer, primary_key=True)

class Child(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    parent_id = db.Column(db.Integer, db.ForeignKey(''parent.id''))

parent = Parent()
db.session.add(parent)

child = Child()
child.parent_id = parent.id
db.session.add(child)

db.session.commit()

Quiero INSERTAR en tablas parent e child dentro de una sesión, considerando que parent_id debe incluirse en la tabla child . En el momento en que creo el objeto child , parent.id es None .

¿Cómo puedo lograr eso?

Answer 1

Puede usar flush () para enjuagar los cambios en la base de datos y así tener su campo de clave primaria actualizado:

parent = Parent()
db.session.add(parent)
db.session.flush()

print parent.id  # after flush(), parent object would be automatically
                 # assigned with a unique primary key to its id field 

child = Child()
child.parent_id = parent.id
db.session.add(child)