xtaodada/sycgram
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add notice of successful download and upload

Browse Source
This commit is contained in:
iwumingz 2022-04-09 17:03:07 +08:00
parent a10fc3247e
commit d5489a7ef2
1 changed files with 16 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
plugins/updown.py
View File

@ -1,3 +1,4 @@
import asyncio
from datetime import datetime from datetime import datetime
import os import os
@ -6,7 +7,7 @@ from loguru import logger
from pyrogram import Client from pyrogram import Client
from pyrogram.types import Message from pyrogram.types import Message
from tools.constants import DOWNLOAD_PATH from tools.constants import DOWNLOAD_PATH
from tools.helpers import Parameters, show_cmd_tip, show_exception from tools.helpers import Parameters, delete_this, show_cmd_tip, show_exception
@Client.on_message(command("upload")) @Client.on_message(command("upload"))
@ -19,7 +20,7 @@ async def upload(cli: Client, msg: Message):
if msg.reply_to_message else None if msg.reply_to_message else None
_, filename = os.path.split(where) _, filename = os.path.split(where)
try: try:
await cli.send_document( res = await cli.send_document(
chat_id=msg.chat.id, chat_id=msg.chat.id,
document=where, document=where,
file_name=filename, file_name=filename,
@ -27,6 +28,11 @@ async def upload(cli: Client, msg: Message):
) )
except Exception as e: except Exception as e:
return await show_exception(msg, e) return await show_exception(msg, e)
else:
if res:
await delete_this(msg)
else:
await msg.edit_text("⚠️ Maybe fail to upload ...")
@Client.on_message(command("download")) @Client.on_message(command("download"))
@ -42,9 +48,16 @@ async def download(_: Client, msg: Message):
else: else:
file_name = DOWNLOAD_PATH if not where else where file_name = DOWNLOAD_PATH if not where else where
try: try:
await replied_msg.download(file_name=file_name) res = await replied_msg.download(file_name=file_name)
except ValueError: except ValueError:
return await show_cmd_tip(msg, cmd) return await show_cmd_tip(msg, cmd)
except Exception as e: except Exception as e:
logger.error(e) logger.error(e)
return await show_exception(msg, e) return await show_exception(msg, e)
else:
if res:
await msg.edit_text("✅ Download this successfully.")
await asyncio.sleep(3)
await delete_this(msg)
else:
await msg.edit_text("⚠️ Maybe fail to download ...")