From 82b7a7f6976906ec1767f02f4e3d568dd5e2852f Mon Sep 17 00:00:00 2001 From: IgorVolochay Date: Sun, 2 Mar 2025 14:19:09 +0300 Subject: [PATCH] fix: fix get visited cards functional --- app/mongo_worker.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/app/mongo_worker.py b/app/mongo_worker.py index 800f4ba..7f1de2c 100644 --- a/app/mongo_worker.py +++ b/app/mongo_worker.py @@ -57,9 +57,12 @@ class MongoWorker: return counter["counter"] - def get_visited_cards(self, user_id: int) -> Visited: + def get_visited_cards(self, user_id: int) -> Optional[Visited]: document = self.visited_data.find_one({"user_id": user_id}) - return Visited.model_validate(document) + if not document: + return None + else: + return Visited.model_validate(document) def update_visited_cards(self, user_id: int, visited_card_id: int) -> Visited: update_visited = self.visited_data.find_one_and_update({"user_id": user_id}, @@ -156,4 +159,4 @@ if __name__ == "__main__": #print(mongo.like_card(1, 455412573)) #print(mongo.dislike_card(1, 455412573)) print(mongo.update_visited_cards(123, 2)) - print(mongo.get_visited_cards(123)) + print(mongo.get_visited_cards(12))