Coverage for docs_src/advanced/uuid/tutorial001.py: 100%

1import uuid 10 ctx1abcdefghij

2from typing import Union 10 ctx1abcdefghij

3 

4from sqlmodel import Field, Session, SQLModel, create_engine, select 10 ctx1abcdefghij

5 

6 

7class Hero(SQLModel, table=True): 10 ctx1abcdefghij

8 id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True) 10 ctx1abcdefghij

9 name: str = Field(index=True) 10 ctx1abcdefghij

10 secret_name: str 10 ctx1abcdefghij

11 age: Union[int, None] = Field(default=None, index=True) 10 ctx1abcdefghij

12 

13 

14sqlite_file_name = "database.db" 10 ctx1abcdefghij

15sqlite_url = f"sqlite:///{sqlite_file_name}" 10 ctx1abcdefghij

16 

17engine = create_engine(sqlite_url, echo=True) 10 ctx1abcdefghij

18 

19 

20def create_db_and_tables(): 10 ctx1abcdefghij

21 SQLModel.metadata.create_all(engine) 10 ctx1abcdefghij

22 

23 

24def create_hero(): 10 ctx1abcdefghij

25 with Session(engine) as session: 10 ctx1abcdefghij

26 hero = Hero(name="Deadpond", secret_name="Dive Wilson") 10 ctx1abcdefghij

27 print("The hero before saving in the DB") 10 ctx1abcdefghij

28 print(hero) 10 ctx1abcdefghij

29 print("The hero ID was already set") 10 ctx1abcdefghij

30 print(hero.id) 10 ctx1abcdefghij

31 session.add(hero) 10 ctx1abcdefghij

32 session.commit() 10 ctx1abcdefghij

33 session.refresh(hero) 10 ctx1abcdefghij

34 print("After saving in the DB") 10 ctx1abcdefghij

35 print(hero) 10 ctx1abcdefghij

36 

37 

38def select_hero(): 10 ctx1abcdefghij

39 with Session(engine) as session: 10 ctx1abcdefghij

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

41 session.add(hero_2) 10 ctx1abcdefghij

42 session.commit() 10 ctx1abcdefghij

43 session.refresh(hero_2) 10 ctx1abcdefghij

44 hero_id = hero_2.id 10 ctx1abcdefghij

45 print("Created hero:") 10 ctx1abcdefghij

46 print(hero_2) 10 ctx1abcdefghij

47 print("Created hero ID:") 10 ctx1abcdefghij

48 print(hero_id) 10 ctx1abcdefghij

49 

50 statement = select(Hero).where(Hero.id == hero_id) 10 ctx1abcdefghij

51 selected_hero = session.exec(statement).one() 10 ctx1abcdefghij

52 print("Selected hero:") 10 ctx1abcdefghij

53 print(selected_hero) 10 ctx1abcdefghij

54 print("Selected hero ID:") 10 ctx1abcdefghij

55 print(selected_hero.id) 10 ctx1abcdefghij

56 

57 

58def main() -> None: 10 ctx1abcdefghij

59 create_db_and_tables() 10 ctx1abcdefghij

60 create_hero() 10 ctx1abcdefghij

61 select_hero() 10 ctx1abcdefghij

62 

63 

64if __name__ == "__main__": 10 ctx1abcdefghij

65 main()