Coverage for docs_src / tutorial / fastapi / app_testing / tutorial001_py310 / main.py: 100%
65 statements
« prev ^ index » next coverage.py v7.13.0, created at 2026-11-09 00:02 +0000
1from fastapi import Depends, FastAPI, HTTPException, Query 1abcdefghi
2from sqlmodel import Field, Session, SQLModel, create_engine, select 1abcdefghi
5class HeroBase(SQLModel): 1abcdefghi
6 name: str = Field(index=True) 1abcdefghi
7 secret_name: str 1abcdefghi
8 age: int | None = Field(default=None, index=True) 1abcdefghi
11class Hero(HeroBase, table=True): 1abcdefghi
12 id: int | None = Field(default=None, primary_key=True) 1abcdefghi
15class HeroCreate(HeroBase): 1abcdefghi
16 pass 1abcdefghi
19class HeroPublic(HeroBase): 1abcdefghi
20 id: int 1abcdefghi
23class HeroUpdate(SQLModel): 1abcdefghi
24 name: str | None = None 1abcdefghi
25 secret_name: str | None = None 1abcdefghi
26 age: int | None = None 1abcdefghi
29sqlite_file_name = "database.db" 1abcdefghi
30sqlite_url = f"sqlite:///{sqlite_file_name}" 1abcdefghi
32connect_args = {"check_same_thread": False} 1abcdefghi
33engine = create_engine(sqlite_url, echo=True, connect_args=connect_args) 1abcdefghi
36def create_db_and_tables(): 1abcdefghi
37 SQLModel.metadata.create_all(engine) 2ccdcecfcgchcicjckc
40def get_session(): 1abcdefghi
41 with Session(engine) as session: 2lcmcncocpcqcrcsctc
42 yield session 2lcmcncocpcqcrcsctc
45app = FastAPI() 1abcdefghi
48@app.on_event("startup") 1abcdefghi
49def on_startup(): 1abcdefghi
50 create_db_and_tables() 2ccdcecfcgchcicjckc
53@app.post("/heroes/", response_model=HeroPublic) 1abcdefghi
54def create_hero(*, session: Session = Depends(get_session), hero: HeroCreate): 1abcdefghi
55 db_hero = Hero.model_validate(hero) 2s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpbqbrbsbtbubvbwbxbybzbAbBbCbDbEbFbGbHbIbJbKbLbMbNbObPbQbRbSbTb
56 session.add(db_hero) 2s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpbqbrbsbtbubvbwbxbybzbAbBbCbDbEbFbGbHbIbJbKbLbMbNbObPbQbRbSbTb
57 session.commit() 2s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpbqbrbsbtbubvbwbxbybzbAbBbCbDbEbFbGbHbIbJbKbLbMbNbObPbQbRbSbTb
58 session.refresh(db_hero) 2s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpbqbrbsbtbubvbwbxbybzbAbBbCbDbEbFbGbHbIbJbKbLbMbNbObPbQbRbSbTb
59 return db_hero 2s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpbqbrbsbtbubvbwbxbybzbAbBbCbDbEbFbGbHbIbJbKbLbMbNbObPbQbRbSbTb
62@app.get("/heroes/", response_model=list[HeroPublic]) 1abcdefghi
63def read_heroes( 1abcdefghi
64 *,
65 session: Session = Depends(get_session),
66 offset: int = 0,
67 limit: int = Query(default=100, le=100),
68):
69 heroes = session.exec(select(Hero).offset(offset).limit(limit)).all() 2ucvcwcxcyczcAcBcCc
70 return heroes 2ucvcwcxcyczcAcBcCc
73@app.get("/heroes/{hero_id}", response_model=HeroPublic) 1abcdefghi
74def read_hero(*, session: Session = Depends(get_session), hero_id: int): 1abcdefghi
75 hero = session.get(Hero, hero_id) 23b4b5b6b7b8b9b!b#b$b%b'b(b)b*b+b,b-b
76 if not hero: 23b4b5b6b7b8b9b!b#b$b%b'b(b)b*b+b,b-b
77 raise HTTPException(status_code=404, detail="Hero not found") 23b5b7b9b#b%b(b*b,b
78 return hero 24b6b8b!b$b'b)b+b-b
81@app.patch("/heroes/{hero_id}", response_model=HeroPublic) 1abcdefghi
82def update_hero( 1abcdefghi
83 *, session: Session = Depends(get_session), hero_id: int, hero: HeroUpdate
84):
85 db_hero = session.get(Hero, hero_id) 2.bj /bk :bl ;bm =bn ?bo @bp [bq ]br
86 if not db_hero: 2.bj /bk :bl ;bm =bn ?bo @bp [bq ]br
87 raise HTTPException(status_code=404, detail="Hero not found") 2.b/b:b;b=b?b@b[b]b
88 hero_data = hero.model_dump(exclude_unset=True) 1jklmnopqr
89 db_hero.sqlmodel_update(hero_data) 1jklmnopqr
90 session.add(db_hero) 1jklmnopqr
91 session.commit() 1jklmnopqr
92 session.refresh(db_hero) 1jklmnopqr
93 return db_hero 1jklmnopqr
96@app.delete("/heroes/{hero_id}") 1abcdefghi
97def delete_hero(*, session: Session = Depends(get_session), hero_id: int): 1abcdefghi
98 hero = session.get(Hero, hero_id) 2^bUb_bVb`bWb{bXb|bYb}bZb~b0bac1bbc2b
99 if not hero: 2^bUb_bVb`bWb{bXb|bYb}bZb~b0bac1bbc2b
100 raise HTTPException(status_code=404, detail="Hero not found") 2^b_b`b{b|b}b~bacbc
101 session.delete(hero) 2UbVbWbXbYbZb0b1b2b
102 session.commit() 2UbVbWbXbYbZb0b1b2b
103 return {"ok": True} 2UbVbWbXbYbZb0b1b2b