From f2d92a42f87dd4ff4fa194552dd93e4fb9f5f9be Mon Sep 17 00:00:00 2001 From: mhdzumair Date: Fri, 26 Jan 2024 19:42:34 +0530 Subject: [PATCH] Add exception handling for DuplicateKeyError in save_movie_metadata() function --- db/crud.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/db/crud.py b/db/crud.py index d46d561..9cbb29f 100644 --- a/db/crud.py +++ b/db/crud.py @@ -7,6 +7,7 @@ from uuid import uuid4 from beanie import WriteRules from beanie.operators import In from fastapi import BackgroundTasks +from pymongo.errors import DuplicateKeyError from db import schemas, models from db.config import settings @@ -331,7 +332,10 @@ async def save_movie_metadata(metadata: dict): background=background, streams=[new_stream], ) - await movie_data.insert(link_rule=WriteRules.WRITE) + try: + await movie_data.insert(link_rule=WriteRules.WRITE) + except DuplicateKeyError: + logging.warning("Duplicate movie found: %s", movie_data.title) logging.info("Added movie %s", movie_data.title)