def session_scope() -> Session:
    """Provide a transactional scope around a series of operations."""
    db = None
    try:
        db = Session(autocommit=False, autoflush=False,
                     bind=engine)  # create session from SQLModel session
        yield db
    finally:
        db.close()
Ejemplo n.º 2
0
def create_plants():
    plant_1 = Plant(name="Hebe")
    plant_2 = Plant(name="Astilbe")
    plant_3 = Plant(name="Sedum")
    plant_4 = Plant(name="Helenium")
    plant_5 = Plant(name="Heather")

    session = Session(engine)

    session.add(plant_1)
    session.add(plant_2)
    session.add(plant_3)
    session.add(plant_4)
    session.add(plant_5)

    session.commit()

    session.close()