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