|
| 1 | +from sqlmodel import Field, Session, SQLModel, create_engine |
| 2 | + |
| 3 | + |
| 4 | +class Hero(SQLModel, table=True): |
| 5 | + id: int | None = Field(default=None, primary_key=True) |
| 6 | + name: str |
| 7 | + secret_name: str |
| 8 | + age: int | None = None |
| 9 | + |
| 10 | + |
| 11 | +sqlite_file_name = "database.db" |
| 12 | +sqlite_url = f"sqlite:///{sqlite_file_name}" |
| 13 | + |
| 14 | +engine = create_engine(sqlite_url, echo=True) |
| 15 | + |
| 16 | + |
| 17 | +def create_db_and_tables(): |
| 18 | + SQLModel.metadata.create_all(engine) |
| 19 | + |
| 20 | + |
| 21 | +def create_heroes(): |
| 22 | + hero_1 = Hero(name="Deadpond", secret_name="Dive Wilson") # (1)! |
| 23 | + hero_2 = Hero(name="Spider-Boy", secret_name="Pedro Parqueador") # (2)! |
| 24 | + hero_3 = Hero(name="Rusty-Man", secret_name="Tommy Sharp", age=48) # (3)! |
| 25 | + |
| 26 | + print("Before interacting with the database") # (4)! |
| 27 | + print("Hero 1:", hero_1) # (5)! |
| 28 | + print("Hero 2:", hero_2) # (6)! |
| 29 | + print("Hero 3:", hero_3) # (7)! |
| 30 | + |
| 31 | + with Session(engine) as session: # (8)! |
| 32 | + session.add(hero_1) # (9)! |
| 33 | + session.add(hero_2) # (10)! |
| 34 | + session.add(hero_3) # (11)! |
| 35 | + |
| 36 | + print("After adding to the session") # (12)! |
| 37 | + print("Hero 1:", hero_1) # (13)! |
| 38 | + print("Hero 2:", hero_2) # (14)! |
| 39 | + print("Hero 3:", hero_3) # (15)! |
| 40 | + |
| 41 | + session.commit() # (16)! |
| 42 | + |
| 43 | + print("After committing the session") # (17)! |
| 44 | + print("Hero 1:", hero_1) # (18)! |
| 45 | + print("Hero 2:", hero_2) # (19)! |
| 46 | + print("Hero 3:", hero_3) # (20)! |
| 47 | + |
| 48 | + print("After committing the session, show IDs") # (21)! |
| 49 | + print("Hero 1 ID:", hero_1.id) # (22)! |
| 50 | + print("Hero 2 ID:", hero_2.id) # (23)! |
| 51 | + print("Hero 3 ID:", hero_3.id) # (24)! |
| 52 | + |
| 53 | + print("After committing the session, show names") # (25)! |
| 54 | + print("Hero 1 name:", hero_1.name) # (26)! |
| 55 | + print("Hero 2 name:", hero_2.name) # (27)! |
| 56 | + print("Hero 3 name:", hero_3.name) # (28)! |
| 57 | + |
| 58 | + session.refresh(hero_1) # (29)! |
| 59 | + session.refresh(hero_2) # (30)! |
| 60 | + session.refresh(hero_3) # (31)! |
| 61 | + |
| 62 | + print("After refreshing the heroes") # (32)! |
| 63 | + print("Hero 1:", hero_1) # (33)! |
| 64 | + print("Hero 2:", hero_2) # (34)! |
| 65 | + print("Hero 3:", hero_3) # (35)! |
| 66 | + # (36)! |
| 67 | + |
| 68 | + print("After the session closes") # (37)! |
| 69 | + print("Hero 1:", hero_1) # (38)! |
| 70 | + print("Hero 2:", hero_2) # (39)! |
| 71 | + print("Hero 3:", hero_3) # (40)! |
| 72 | + |
| 73 | + |
| 74 | +def main(): |
| 75 | + create_db_and_tables() |
| 76 | + create_heroes() |
| 77 | + |
| 78 | + |
| 79 | +if __name__ == "__main__": |
| 80 | + main() |
0 commit comments