profitroll/TelegramPoster
Archived
1
0
Fork 0
Code Issues Pull Requests Releases 3 Wiki Activity

Migrate to async_pymongo

Browse Source
This commit is contained in:
Profitroll
2023-08-14 14:52:02 +02:00
parent 5b56919b80
commit cd26990b7e
7 changed files with 23 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
plugins/callbacks/submission.py
View File

@@ -25,7 +25,7 @@ async def callback_query_yes(app: PyroClient, clb: CallbackQuery):
user = await app.find_user(clb.from_user)
fullclb = str(clb.data).split("_")
db_entry = col_submitted.find_one({"_id": ObjectId(fullclb[2])})
db_entry = await col_submitted.find_one({"_id": ObjectId(fullclb[2])})
try:
submission = await app.submit_media(fullclb[2])
@@ -127,7 +127,7 @@ async def callback_query_no(app: PyroClient, clb: CallbackQuery):
user = await app.find_user(clb.from_user)
fullclb = str(clb.data).split("_")
db_entry = col_submitted.find_one_and_delete({"_id": ObjectId(fullclb[2])})
db_entry = await col_submitted.find_one_and_delete({"_id": ObjectId(fullclb[2])})
if (
db_entry["temp"]["uuid"] is not None

4
plugins/handlers/submission.py
View File

@@ -152,7 +152,7 @@ async def get_submission(app: PyroClient, msg: Message):
+ sep,
)
inserted = col_submitted.insert_one(
inserted = await col_submitted.insert_one(
{
"user": msg.from_user.id,
"date": datetime.now(),
@@ -165,7 +165,7 @@ async def get_submission(app: PyroClient, msg: Message):
)
else:
inserted = col_submitted.insert_one(
inserted = await col_submitted.insert_one(
{
"user": msg.from_user.id,
"date": datetime.now(),