fix: fix get visited cards functional

This commit is contained in:
IgorVolochay
2025-03-02 14:19:09 +03:00
parent 0d3338b886
commit 82b7a7f697
+5 -2
View File
@@ -57,8 +57,11 @@ class MongoWorker:
return counter["counter"] 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}) document = self.visited_data.find_one({"user_id": user_id})
if not document:
return None
else:
return Visited.model_validate(document) return Visited.model_validate(document)
def update_visited_cards(self, user_id: int, visited_card_id: int) -> Visited: def update_visited_cards(self, user_id: int, visited_card_id: int) -> Visited:
@@ -156,4 +159,4 @@ if __name__ == "__main__":
#print(mongo.like_card(1, 455412573)) #print(mongo.like_card(1, 455412573))
#print(mongo.dislike_card(1, 455412573)) #print(mongo.dislike_card(1, 455412573))
print(mongo.update_visited_cards(123, 2)) print(mongo.update_visited_cards(123, 2))
print(mongo.get_visited_cards(123)) print(mongo.get_visited_cards(12))