By setting 'status_code' parameter to the decorator method (@app.get(), @app.post(), @app.put(), @app.delete()), we can explicilty set the return code.
Example
@app.post("/emps", status_code=201)
def createEmployee(emp: Employee):
    noOfEmps = len(emps)
    newId = noOfEmps + 1
    emps[newId] = emp
    return {"id" : newId, "name" : emp.name, "age": emp.age, "country": emp.country}
Find the below working application.
statusCode.py
from fastapi import FastAPI, Path
from typing import Optional
from pydantic import BaseModel
from enum import Enum
app = FastAPI()
# model classes
class Employee(BaseModel):
    name: str
    age: int
    country: str
# employees information
emps = {
    1 : {
        "name" : "Krishna",
        "age": 32,
        "country": "India"
    },
    2 : {
        "name" : "Ram",
        "age": 33,
        "country": "China"
    },
    3 : {
        "name" : "Bomma",
        "age": 38,
        "country": "China"
    }
}
# Create home endpoint
@app.get("/")
def home():
    return {"name" : "Hello World app", "version": "2.0.0"}
# Employees REST APIs
@app.get("/emps")
def allEmployees():
    return emps
@app.post("/emps", status_code=201)
def createEmployee(emp: Employee):
    noOfEmps = len(emps)
    newId = noOfEmps + 1
    emps[newId] = emp
    return {"id" : newId, "name" : emp.name, "age": emp.age, "country": emp.country}
Open terminal and execute the command ‘uvicorn statusCode:app --reload’.
$uvicorn statusCode:app --reload
INFO:     Uvicorn running on http://127.0.0.1:8000 (Press CTRL+C to quit)
INFO:     Started reloader process [46766] using statreload
INFO:     Started server process [46768]
INFO:     Waiting for application startup.
INFO:     Application startup complete.
Open the url ‘http://localhost:8000/docs’ in browser and experiment with the api ‘POST /emps’.
 
 
Previous Next Home
No comments:
Post a Comment