sqlmodel_sqlite_hero_crud_quickstart_with_session.py

1from typing import Optional
2
3from sqlmodel import Field, Session, SQLModel, create_engine, select
4
5
6class Hero(SQLModel, table=True):
7    id: Optional[int] = Field(default=None, primary_key=True)
8    name: str
9    secret_name: str
10    age: Optional[int] = None
11
12
13hero_1 = Hero(name="Deadpond", secret_name="Dive Wilson")
14hero_2 = Hero(name="Spider-Boy", secret_name="Pedro Parqueador")
15hero_3 = Hero(name="Rusty-Man", secret_name="Tommy Sharp", age=48)
16
17
18engine = create_engine("sqlite:///database.db")
19
20
21SQLModel.metadata.create_all(engine)
22
23with Session(engine) as session:
24    session.add(hero_1)
25    session.add(hero_2)
26    session.add(hero_3)
27    session.commit()
28
29with Session(engine) as session:
30    statement = select(Hero).where(Hero.name == "Spider-Boy")
31    hero = session.exec(statement).first()
32    print(hero)