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()
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()