fastapi response models

tinylambda · tinylambda · commit f411a82399ee · 2022-03-10T13:29:23.000+08:00
diff --git a/module_fastapi/fastapi_extra_models.py b/module_fastapi/fastapi_extra_models.py
@@ -0,0 +1,116 @@
+from typing import Optional, Union, List, Dict
+
+from fastapi import FastAPI
+from pydantic import BaseModel, EmailStr
+
+
+app = FastAPI()
+
+
+# class UserIn(BaseModel):
+#     username: str
+#     password: str
+#     email: EmailStr
+#     full_name: Optional[str] = None
+#
+#
+# class UserOut(BaseModel):
+#     username: str
+#     email: EmailStr
+#     full_name: Optional[str] = None
+#
+#
+# class UserInDB(BaseModel):
+#     username: str
+#     hashed_password: str
+#     email: EmailStr
+#     full_name: Optional[str] = None
+
+
+class UserBase(BaseModel):
+    username: str
+    email: EmailStr
+    full_name: Optional[str] = None
+
+
+class UserIn(UserBase):
+    password: str
+
+
+class UserOut(UserBase):
+    pass
+
+
+class UserInDB(UserBase):
+    hashed_password: str
+
+
+class BaseItem(BaseModel):
+    description: str
+    type: str
+
+
+class CarItem(BaseItem):
+    type = 'car'
+
+
+class PlaneItem(BaseItem):
+    type = 'plane'
+    size: int
+
+
+items = {
+    "item1": {"description": "All my friends drive a low rider", "type": "car"},
+    "item2": {
+        "description": "Music is my aeroplane, it's my aeroplane",
+        "type": "plane",
+        "size": 5,
+    },
+}
+
+
+class Item(BaseModel):
+    name: str
+    description: str
+
+
+items2 = [
+    {"name": "Foo", "description": "There comes my hero"},
+    {"name": "Red", "description": "It's my aeroplane"},
+]
+
+
+def fake_password_hasher(raw_password: str):
+    return 'supersecret' + raw_password
+
+
+def fake_save_user(user_in: UserIn):
+    hashed_password = fake_password_hasher(user_in.password)
+    user_in_db = UserInDB(**user_in.dict(), hashed_password=hashed_password)
+    print('User saved! ..not really')
+    return user_in_db
+
+
+@app.post('/user/', response_model=UserOut)
+async def create_user(user_in: UserIn):
+    user_saved = fake_save_user(user_in)
+    return user_saved
+
+
+@app.get('/items/{item_id}', response_model=Union[PlaneItem, CarItem])
+async def read_item(item_id: str):
+    return items[item_id]
+
+
+@app.get('/items/', response_model=List[Item])
+async def read_items():
+    return items2
+
+
+@app.get('/keyword-weights/', response_model=Dict[str, float])
+async def read_keyword_weights():
+    return {'foo': 2.3, 'bar': 4.5}
+
+
+# PYTHONPATH=module_fastapi uvicorn fastapi_extra_models:app --reload
+
diff --git a/module_fastapi/fastapi_response_model.py b/module_fastapi/fastapi_response_model.py
@@ -0,0 +1,64 @@
+from typing import Optional, List
+
+from fastapi import FastAPI
+from pydantic import BaseModel, EmailStr
+
+app = FastAPI()
+
+
+class Item(BaseModel):
+    name: str
+    description: Optional[str] = None
+    price: float
+    tax: float = 10.5
+    tags: List[str] = []
+
+
+items = {
+    "foo": {"name": "Foo", "price": 50.2},
+    "bar": {"name": "Bar", "description": "The bartenders", "price": 62, "tax": 20.2},
+    "baz": {"name": "Baz", "description": None, "price": 50.2, "tax": 10.5, "tags": []},
+}
+
+
+class UserIn(BaseModel):
+    username: str
+    password: str
+    email: EmailStr
+    full_name: Optional[str] = None
+
+
+class UserOut(BaseModel):
+    username: str
+    email: EmailStr
+    full_name: Optional[str] = None
+
+
+@app.post('/items/', response_model=Item)
+async def create_item(item: Item):
+    return item
+
+
+@app.get(
+    '/items/{item_id}',
+    response_model=Item,
+    response_model_exclude_unset=True,
+    # response_model_exclude_defaults=True,
+    # response_model_exclude_none=True,
+    # response_model_include={'name'},
+    # response_model_exclude={'name'},
+)
+async def read_item(item_id: str):
+    return items[item_id]
+
+
+@app.post('/user/', response_model=UserIn)
+async def create_user(user: UserIn):
+    return user
+
+
+@app.post('/user2/', response_model=UserOut)
+async def create_user2(user: UserIn):
+    return user
+
+# PYTHONPATH=module_fastapi uvicorn fastapi_response_model:app --reload
