From 16695245496747d41f8266c032d0aeec19e69e1d Mon Sep 17 00:00:00 2001 From: Itz-fork Date: Tue, 2 Jan 2024 15:29:47 +0530 Subject: [PATCH] database: use `query.message.from_user.id` --- megadl/helpers/database.py | 12 ++---------- megadl/modules/mega_dl.py | 2 +- megadl/modules/mega_up.py | 2 +- 3 files changed, 4 insertions(+), 12 deletions(-) diff --git a/megadl/helpers/database.py b/megadl/helpers/database.py index 388e579e..cb75aa0e 100644 --- a/megadl/helpers/database.py +++ b/megadl/helpers/database.py @@ -42,22 +42,14 @@ async def plus_fl_count( await self.mongoc.update_async( self.coll_users, {"_id": user_id}, - [ - {"$match": {"_id": user_id}}, - {"$inc": {"total_downloads": downloads}}, - {"$replaceWith": {"$mergeObjects": ["$$ROOT", "$$CURRENT"]}}, - ], + {"$inc": {"total_downloads": downloads}}, use_given=True, ) elif uploads: await self.mongoc.update_async( self.coll_users, {"_id": user_id}, - [ - {"$match": {"_id": user_id}}, - {"$inc": {"total_uploads": uploads}}, - {"$replaceWith": {"$mergeObjects": ["$$ROOT", "$$CURRENT"]}}, - ], + {"$inc": {"total_uploads": uploads}}, use_given=True, ) diff --git a/megadl/modules/mega_dl.py b/megadl/modules/mega_dl.py index 95957db6..d344b0b4 100644 --- a/megadl/modules/mega_dl.py +++ b/megadl/modules/mega_dl.py @@ -49,7 +49,7 @@ async def dl_from_cb(client: CypherClient, query: CallbackQuery): # Access saved info _mid = int(query.data.split("-")[1]) qcid = query.message.chat.id - qusr = query.from_user.id + qusr = query.message.from_user.id dtmp = client.glob_tmp.get(qusr) url = dtmp[0] dlid = dtmp[1] diff --git a/megadl/modules/mega_up.py b/megadl/modules/mega_up.py index d61cee90..9c4a9894 100644 --- a/megadl/modules/mega_up.py +++ b/megadl/modules/mega_up.py @@ -54,7 +54,7 @@ async def to_up_cb(client: CypherClient, query: CallbackQuery): _mid = int(query.data.split("-")[1]) qmid = query.message.id qcid = query.message.chat.id - qusr = query.from_user.id + qusr = query.message.from_user.id # weird workaround to add support for private mode conf = None