volochay/thisORthat
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 1 Wiki Activity

fix: fix get visited cards functional

Browse Source
This commit is contained in:
IgorVolochay
2025-03-02 14:19:09 +03:00
parent 0d3338b886
commit 82b7a7f697
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
app/mongo_worker.py
+6 -3
View File
@@ -57,9 +57,12 @@ 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})
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: 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}, 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.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))