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

1from enum import Enum 5 ctx1abcde

2 

3from fastapi import FastAPI 5 ctx1abcde

4 

5 

6class ModelName(str, Enum): 5 ctx1abcde

7 alexnet = "alexnet" 5 ctx1abcde

8 resnet = "resnet" 5 ctx1abcde

9 lenet = "lenet" 5 ctx1abcde

10 

11 

12app = FastAPI() 5 ctx1abcde

13 

14 

15@app.get("/models/{model_name}") 5 ctx1abcde

16async def get_model(model_name: ModelName): 5 ctx1abcde

17 if model_name is ModelName.alexnet: 5 ctx1abcde

18 return {"model_name": model_name, "message": "Deep Learning FTW!"} 5 ctx1abcde

19 

20 if model_name.value == "lenet": 5 ctx1abcde

21 return {"model_name": model_name, "message": "LeCNN all the images"} 5 ctx1abcde

22 

23 return {"model_name": model_name, "message": "Have some residuals"} 5 ctx1abcde