Coverage for docs_src/tutorial/select/tutorial002.py: 100%

1from typing import Optional 6 ctx1abcdef

2 

3from sqlmodel import Field, Session, SQLModel, create_engine, select # (1)! 6 ctx1abcdef

4 

5 

6class Hero(SQLModel, table=True): # (2)! 6 ctx1abcdef

7 id: Optional[int] = Field(default=None, primary_key=True) 6 ctx1abcdef

8 name: str 6 ctx1abcdef

9 secret_name: str 6 ctx1abcdef

10 age: Optional[int] = None 6 ctx1abcdef

11 

12 

13sqlite_file_name = "database.db" 6 ctx1abcdef

14sqlite_url = f"sqlite:///{sqlite_file_name}" 6 ctx1abcdef

15 

16engine = create_engine(sqlite_url, echo=True) # (3)! 6 ctx1abcdef

17 

18 

19def create_db_and_tables(): 6 ctx1abcdef

20 SQLModel.metadata.create_all(engine) # (4)! 6 ctx1abcdef

21 

22 

23def create_heroes(): 6 ctx1abcdef

24 hero_1 = Hero(name="Deadpond", secret_name="Dive Wilson") # (5)! 6 ctx1abcdef

25 hero_2 = Hero(name="Spider-Boy", secret_name="Pedro Parqueador") 6 ctx1abcdef

26 hero_3 = Hero(name="Rusty-Man", secret_name="Tommy Sharp", age=48) 6 ctx1abcdef

27 

28 with Session(engine) as session: # (6)! 6 ctx1abcdef

29 session.add(hero_1) 6 ctx1abcdef

30 session.add(hero_2) 6 ctx1abcdef

31 session.add(hero_3) 6 ctx1abcdef

32 

33 session.commit() 6 ctx1abcdef

34 

35 

36def select_heroes(): 6 ctx1abcdef

37 with Session(engine) as session: # (7)! 6 ctx1abcdef

38 statement = select(Hero) # (8)! 6 ctx1abcdef

39 results = session.exec(statement) # (9)! 6 ctx1abcdef

40 for hero in results: # (10)! 6 ctx1abcdef

41 print(hero) # (11)! 6 ctx1abcdef

42 # (12)! 

43 

44 

45def main(): 6 ctx1abcdef

46 create_db_and_tables() 6 ctx1abcdef

47 create_heroes() 6 ctx1abcdef

48 select_heroes() # (13)! 6 ctx1abcdef

49 

50 

51if __name__ == "__main__": 6 ctx1abcdef

52 main()