From 5ee4cb604141130146756aa727a9245c6126bd3c Mon Sep 17 00:00:00 2001 From: Colin Goutte Date: Fri, 25 Aug 2023 23:14:03 +0200 Subject: [PATCH] Green: Remove custom schema adapter Nice we have the releasse date we missed --- crud.py | 4 +++- dev.py | 11 +---------- 2 files changed, 4 insertions(+), 11 deletions(-) diff --git a/crud.py b/crud.py index f9c20a8..60e8f4b 100644 --- a/crud.py +++ b/crud.py @@ -12,7 +12,8 @@ def create_movie( genres: list[str], description: str = "", vote_average: float | None = None, - vote_count: int | None = None + vote_count: int | None = None, + release_date: str | None = None ): db_movie = models.Movie( title=title, @@ -20,6 +21,7 @@ def create_movie( description=description, vote_average=vote_average, vote_count=vote_count, + release_date=release_date, ) db.add(db_movie) db.commit() diff --git a/dev.py b/dev.py index 8835668..be3bccc 100644 --- a/dev.py +++ b/dev.py @@ -27,16 +27,7 @@ async def root(): @app.post("/pydantic_movies/") async def create_movie(payload: schemas.MoviePayload, db: Session = Depends(get_db)): - data = payload.dict() - crud_params = dict( - genres=data.get("genres", ["Unknown"]), - description=data.get("description", ""), - title=data.get("title", ""), - vote_average=data.get("vote_average"), - vote_count=data.get("vote_count"), - ) - - movie = crud.create_movie(db, **crud_params) + movie = crud.create_movie(db, **payload.dict()) out = {"message": f"Created {movie.title} XX", "id": movie.id} return out