Coverage for docs_src / tutorial / fastapi / app_testing / tutorial001_py39 / main.py: 100%
66 statements
« prev ^ index » next coverage.py v7.13.0, created at 2026-01-06 21:09 +0000
1from typing import Optional 1abcdefghi
3from fastapi import Depends, FastAPI, HTTPException, Query 1abcdefghi
4from sqlmodel import Field, Session, SQLModel, create_engine, select 1abcdefghi
7class HeroBase(SQLModel): 1abcdefghi
8 name: str = Field(index=True) 1abcdefghi
9 secret_name: str 1abcdefghi
10 age: Optional[int] = Field(default=None, index=True) 1abcdefghi
13class Hero(HeroBase, table=True): 1abcdefghi
14 id: Optional[int] = Field(default=None, primary_key=True) 1abcdefghi
17class HeroCreate(HeroBase): 1abcdefghi
18 pass 1abcdefghi
21class HeroPublic(HeroBase): 1abcdefghi
22 id: int 1abcdefghi
25class HeroUpdate(SQLModel): 1abcdefghi
26 name: Optional[str] = None 1abcdefghi
27 secret_name: Optional[str] = None 1abcdefghi
28 age: Optional[int] = None 1abcdefghi
31sqlite_file_name = "database.db" 1abcdefghi
32sqlite_url = f"sqlite:///{sqlite_file_name}" 1abcdefghi
34connect_args = {"check_same_thread": False} 1abcdefghi
35engine = create_engine(sqlite_url, echo=True, connect_args=connect_args) 1abcdefghi
38def create_db_and_tables(): 1abcdefghi
39 SQLModel.metadata.create_all(engine) 2ccdcecfcgchcicjckc
42def get_session(): 1abcdefghi
43 with Session(engine) as session: 2lcmcncocpcqcrcsctc
44 yield session 2lcmcncocpcqcrcsctc
47app = FastAPI() 1abcdefghi
50@app.on_event("startup") 1abcdefghi
51def on_startup(): 1abcdefghi
52 create_db_and_tables() 2ccdcecfcgchcicjckc
55@app.post("/heroes/", response_model=HeroPublic) 1abcdefghi
56def create_hero(*, session: Session = Depends(get_session), hero: HeroCreate): 1abcdefghi
57 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
58 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
59 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
60 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
61 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
64@app.get("/heroes/", response_model=list[HeroPublic]) 1abcdefghi
65def read_heroes( 1abcdefghi
66 *,
67 session: Session = Depends(get_session),
68 offset: int = 0,
69 limit: int = Query(default=100, le=100),
70):
71 heroes = session.exec(select(Hero).offset(offset).limit(limit)).all() 2ucvcwcxcyczcAcBcCc
72 return heroes 2ucvcwcxcyczcAcBcCc
75@app.get("/heroes/{hero_id}", response_model=HeroPublic) 1abcdefghi
76def read_hero(*, session: Session = Depends(get_session), hero_id: int): 1abcdefghi
77 hero = session.get(Hero, hero_id) 23b4b5b6b7b8b9b!b#b$b%b'b(b)b*b+b,b-b
78 if not hero: 23b4b5b6b7b8b9b!b#b$b%b'b(b)b*b+b,b-b
79 raise HTTPException(status_code=404, detail="Hero not found") 23b5b7b9b#b%b(b*b,b
80 return hero 24b6b8b!b$b'b)b+b-b
83@app.patch("/heroes/{hero_id}", response_model=HeroPublic) 1abcdefghi
84def update_hero( 1abcdefghi
85 *, session: Session = Depends(get_session), hero_id: int, hero: HeroUpdate
86):
87 db_hero = session.get(Hero, hero_id) 2.bj /bk :bl ;bm =bn ?bo @bp [bq ]br
88 if not db_hero: 2.bj /bk :bl ;bm =bn ?bo @bp [bq ]br
89 raise HTTPException(status_code=404, detail="Hero not found") 2.b/b:b;b=b?b@b[b]b
90 hero_data = hero.model_dump(exclude_unset=True) 1jklmnopqr
91 db_hero.sqlmodel_update(hero_data) 1jklmnopqr
92 session.add(db_hero) 1jklmnopqr
93 session.commit() 1jklmnopqr
94 session.refresh(db_hero) 1jklmnopqr
95 return db_hero 1jklmnopqr
98@app.delete("/heroes/{hero_id}") 1abcdefghi
99def delete_hero(*, session: Session = Depends(get_session), hero_id: int): 1abcdefghi
100 hero = session.get(Hero, hero_id) 2^bUb_bVb`bWb{bXb|bYb}bZb~b0bac1bbc2b
101 if not hero: 2^bUb_bVb`bWb{bXb|bYb}bZb~b0bac1bbc2b
102 raise HTTPException(status_code=404, detail="Hero not found") 2^b_b`b{b|b}b~bacbc
103 session.delete(hero) 2UbVbWbXbYbZb0b1b2b
104 session.commit() 2UbVbWbXbYbZb0b1b2b
105 return {"ok": True} 2UbVbWbXbYbZb0b1b2b