Coverage for docs_src / tutorial / fastapi / multiple_models / tutorial001_py310.py: 100%

39 statements  

« prev     ^ index     » next       coverage.py v7.13.0, created at 2026-11-09 00:02 +0000

1from fastapi import FastAPI 1abcdefghi

2from sqlmodel import Field, Session, SQLModel, create_engine, select 1abcdefghi

3 

4 

5class Hero(SQLModel, table=True): 1abcdefghi

6 id: int | None = Field(default=None, primary_key=True) 1abcdefghi

7 name: str = Field(index=True) 1abcdefghi

8 secret_name: str 1abcdefghi

9 age: int | None = Field(default=None, index=True) 1abcdefghi

10 

11 

12class HeroCreate(SQLModel): 1abcdefghi

13 name: str 1abcdefghi

14 secret_name: str 1abcdefghi

15 age: int | None = None 1abcdefghi

16 

17 

18class HeroPublic(SQLModel): 1abcdefghi

19 id: int 1abcdefghi

20 name: str 1abcdefghi

21 secret_name: str 1abcdefghi

22 age: int | None = None 1abcdefghi

23 

24 

25sqlite_file_name = "database.db" 1abcdefghi

26sqlite_url = f"sqlite:///{sqlite_file_name}" 1abcdefghi

27 

28connect_args = {"check_same_thread": False} 1abcdefghi

29engine = create_engine(sqlite_url, echo=True, connect_args=connect_args) 1abcdefghi

30 

31 

32def create_db_and_tables(): 1abcdefghi

33 SQLModel.metadata.create_all(engine) 1jklmnopqr

34 

35 

36app = FastAPI() 1abcdefghi

37 

38 

39@app.on_event("startup") 1abcdefghi

40def on_startup(): 1abcdefghi

41 create_db_and_tables() 1jklmnopqr

42 

43 

44@app.post("/heroes/", response_model=HeroPublic) 1abcdefghi

45def create_hero(hero: HeroCreate): 1abcdefghi

46 with Session(engine) as session: 1jklmnopqr

47 db_hero = Hero.model_validate(hero) 1jklmnopqr

48 session.add(db_hero) 1jklmnopqr

49 session.commit() 1jklmnopqr

50 session.refresh(db_hero) 1jklmnopqr

51 return db_hero 1jklmnopqr

52 

53 

54@app.get("/heroes/", response_model=list[HeroPublic]) 1abcdefghi

55def read_heroes(): 1abcdefghi

56 with Session(engine) as session: 1jklmnopqr

57 heroes = session.exec(select(Hero)).all() 1jklmnopqr

58 return heroes 1jklmnopqr