Building Coupon Reminders - Part 2 Serverless and Secured Image Storage and Processing - app.py

coupon_reminders_part2_apis.py

from pydantic import BaseModel
from typing import List
from fastapi import FastAPI
from mangum import Mangum

import Coupon

class AddCouponRequest(BaseModel):
    user_id: str

class GetCouponsResponse(BaseModel):
    coupons: List[Coupon]

stage = os.environ.get('STAGE', None)
openapi_prefix = f"/{stage}" if stage else "/"

app = FastAPI(title="Coupon Reminders App", openapi_prefix=openapi_prefix)

@app.post("/addCoupon")
def add_coupon(request: AddCouponRequest):
    Coupon.process_and_store(request.user_id)
    return

@app.get("/getCoupons/{user_id}")
def get_coupon(user_id: str) -> GetCouponsResponse:
    return GetCouponsResponse(coupons=Coupon.retrieve_latest_all(user_id))

handler = Mangum(app)