mirror of
https://github.com/fastapi/fastapi.git
synced 2026-04-03 23:05:52 -04:00
49 lines
1.6 KiB
Python
49 lines
1.6 KiB
Python
from typing import List, Set
|
|
|
|
from fastapi import Body, FastAPI, Path, Query, Depends, Cookie
|
|
from pydantic import BaseModel
|
|
from pydantic.types import UrlStr
|
|
from starlette.status import HTTP_201_CREATED
|
|
from starlette.responses import HTMLResponse
|
|
from random import choice
|
|
|
|
app = FastAPI()
|
|
|
|
|
|
class InterestsTracker(BaseModel):
|
|
track_code: str
|
|
interests: List[str]
|
|
|
|
|
|
fake_tracked_users_db = {
|
|
"Foo": {"track_code": "Foo", "interests": ["sports", "movies"]},
|
|
"Bar": {"track_code": "Bar", "interests": ["food", "shows"]},
|
|
"Baz": {"track_code": "Baz", "interests": ["gaming", "virtual reality"]},
|
|
}
|
|
|
|
async def get_tracked_interests(track_code: str = Cookie(None)):
|
|
if track_code in fake_tracked_users_db:
|
|
track_dict = fake_tracked_users_db[track_code]
|
|
track = InterestsTracker(**track_dict)
|
|
return track
|
|
return None
|
|
|
|
class ComplexTracker:
|
|
def __init__(self, tracker: InterestsTracker = Depends(get_tracked_interests)):
|
|
self.tracker = tracker
|
|
|
|
def random_interest(self):
|
|
"""
|
|
Get a random interest from the tracked ones for the current user.
|
|
If the user doesn't have tracked interests, return a random one from the ones available.
|
|
"""
|
|
if self.tracker.interests:
|
|
return choice(self.tracker.interests)
|
|
return choice(["sports", "movies", "food", "shows", "gaming", "virtual reality"])
|
|
|
|
|
|
@app.get("/suggested-category")
|
|
async def read_suggested_category(tracker: ComplexTracker = Depends(None)):
|
|
response = {"category": tracker.random_interest()}
|
|
return response
|