Coverage for docs_src / path_params / tutorial005_py310.py: 100%

1from enum import Enum 4 ctx1abcd

2 

3from fastapi import FastAPI 4 ctx1abcd

4 

5 

6class ModelName(str, Enum): 4 ctx1abcd

7 alexnet = "alexnet" 4 ctx1abcd

8 resnet = "resnet" 4 ctx1abcd

9 lenet = "lenet" 4 ctx1abcd

10 

11 

12app = FastAPI() 4 ctx1abcd

13 

14 

15@app.get("/models/{model_name}") 4 ctx1abcd

16async def get_model(model_name: ModelName): 4 ctx1abcd

17 if model_name is ModelName.alexnet: 9 ctx1keflghmij

18 return {"model_name": model_name, "message": "Deep Learning FTW!"} 3 ctx1klm

19 

20 if model_name.value == "lenet": 6 ctx1efghij

21 return {"model_name": model_name, "message": "LeCNN all the images"} 3 ctx1egi

22 

23 return {"model_name": model_name, "message": "Have some residuals"} 3 ctx1fhj