Coverage for docs_src/tutorial/automatic_id_none_refresh/tutorial001.py: 100%

1from typing import Optional 10 ctx1klmnopqrst

2 

3from sqlmodel import Field, Session, SQLModel, create_engine 10 ctx1klmnopqrst

4 

5 

6class Hero(SQLModel, table=True): 10 ctx1klmnopqrst

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

8 name: str 10 ctx1klmnopqrst

9 secret_name: str 10 ctx1klmnopqrst

10 age: Optional[int] = None 10 ctx1klmnopqrst

11 

12 

13sqlite_file_name = "database.db" 10 ctx1klmnopqrst

14sqlite_url = f"sqlite:///{sqlite_file_name}" 10 ctx1klmnopqrst

15 

16engine = create_engine(sqlite_url, echo=True) 10 ctx1klmnopqrst

17 

18 

19def create_db_and_tables(): 10 ctx1klmnopqrst

20 SQLModel.metadata.create_all(engine) 10 ctx1abcdefghij

21 

22 

23def create_heroes(): 10 ctx1klmnopqrst

24 hero_1 = Hero(name="Deadpond", secret_name="Dive Wilson") 10 ctx1abcdefghij

25 hero_2 = Hero(name="Spider-Boy", secret_name="Pedro Parqueador") 10 ctx1abcdefghij

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

27 

28 print("Before interacting with the database") 10 ctx1abcdefghij

29 print("Hero 1:", hero_1) 10 ctx1abcdefghij

30 print("Hero 2:", hero_2) 10 ctx1abcdefghij

31 print("Hero 3:", hero_3) 10 ctx1abcdefghij

32 

33 with Session(engine) as session: 10 ctx1abcdefghij

34 session.add(hero_1) 10 ctx1abcdefghij

35 session.add(hero_2) 10 ctx1abcdefghij

36 session.add(hero_3) 10 ctx1abcdefghij

37 

38 print("After adding to the session") 10 ctx1abcdefghij

39 print("Hero 1:", hero_1) 10 ctx1abcdefghij

40 print("Hero 2:", hero_2) 10 ctx1abcdefghij

41 print("Hero 3:", hero_3) 10 ctx1abcdefghij

42 

43 session.commit() 10 ctx1abcdefghij

44 

45 print("After committing the session") 10 ctx1abcdefghij

46 print("Hero 1:", hero_1) 10 ctx1abcdefghij

47 print("Hero 2:", hero_2) 10 ctx1abcdefghij

48 print("Hero 3:", hero_3) 10 ctx1abcdefghij

49 

50 print("After committing the session, show IDs") 10 ctx1abcdefghij

51 print("Hero 1 ID:", hero_1.id) 10 ctx1abcdefghij

52 print("Hero 2 ID:", hero_2.id) 10 ctx1abcdefghij

53 print("Hero 3 ID:", hero_3.id) 10 ctx1abcdefghij

54 

55 print("After committing the session, show names") 10 ctx1abcdefghij

56 print("Hero 1 name:", hero_1.name) 10 ctx1abcdefghij

57 print("Hero 2 name:", hero_2.name) 10 ctx1abcdefghij

58 print("Hero 3 name:", hero_3.name) 10 ctx1abcdefghij

59 

60 session.refresh(hero_1) 10 ctx1abcdefghij

61 session.refresh(hero_2) 10 ctx1abcdefghij

62 session.refresh(hero_3) 10 ctx1abcdefghij

63 

64 print("After refreshing the heroes") 10 ctx1abcdefghij

65 print("Hero 1:", hero_1) 10 ctx1abcdefghij

66 print("Hero 2:", hero_2) 10 ctx1abcdefghij

67 print("Hero 3:", hero_3) 10 ctx1abcdefghij

68 

69 print("After the session closes") 10 ctx1abcdefghij

70 print("Hero 1:", hero_1) 10 ctx1abcdefghij

71 print("Hero 2:", hero_2) 10 ctx1abcdefghij

72 print("Hero 3:", hero_3) 10 ctx1abcdefghij

73 

74 

75def main(): 10 ctx1klmnopqrst

76 create_db_and_tables() 10 ctx1abcdefghij

77 create_heroes() 10 ctx1abcdefghij

78 

79 

80if __name__ == "__main__": 10 ctx1klmnopqrst

81 main()