database: use query.message.from_user.id

Itz-fork · Jan 2, 2024 · 1669524 · 1669524
1 parent b57f662
commit 1669524
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 12 deletions.
diff --git 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
@@ -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
@@ -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