Coverage for docs_src/query_params_str_validations/tutorial015_an.py: 100%

17 statements  

« prev     ^ index     » next       coverage.py v7.6.1, created at 2025-12-04 08:29 +0000

1import random 1abcdefg

2from typing import Union 1abcdefg

3 

4from fastapi import FastAPI 1abcdefg

5from pydantic import AfterValidator 1abcdefg

6from typing_extensions import Annotated 1abcdefg

7 

8app = FastAPI() 1abcdefg

9 

10data = { 1abcdefg

11 "isbn-9781529046137": "The Hitchhiker's Guide to the Galaxy", 

12 "imdb-tt0371724": "The Hitchhiker's Guide to the Galaxy", 

13 "isbn-9781439512982": "Isaac Asimov: The Complete Stories, Vol. 2", 

14} 

15 

16 

17def check_valid_id(id: str): 1abcdefg

18 if not id.startswith(("isbn-", "imdb-")): 1ChiDjkElmFnoGpqHrsItu

19 raise ValueError('Invalid ID format, it must start with "isbn-" or "imdb-"') 1CDEFGHI

20 return id 1hijklmnopqrstu

21 

22 

23@app.get("/items/") 1abcdefg

24async def read_items( 1abcdefg

25 id: Annotated[Union[str, None], AfterValidator(check_valid_id)] = None, 

26): 

27 if id: 1hivjkwlmxnoypqzrsAtuB

28 item = data.get(id) 1hijklmnopqrstu

29 else: 

30 id, item = random.choice(list(data.items())) 1vwxyzAB

31 return {"id": id, "name": item} 1hivjkwlmxnoypqzrsAtuB