tgbot-collection/ytdlbot
1
1
Fork 0
mirror of https://github.com/tgbot-collection/ytdlbot.git synced 2025-02-24 23:34:44 +08:00
Code Issues Projects Releases Packages Wiki Activity

use async everywhere

Browse source
This commit is contained in:
BennyThink 2021-05-04 13:24:57 +08:00
parent 508a6bfd5f
commit d245780de5
No known key found for this signature in database
GPG key ID: 6CD0DBDA5235D481
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
bot.py
View file

@ -49,7 +49,7 @@ def progress_hook(d: dict, chat_id, message):
threading.Thread(target=go, args=(chat_id, message, msg)).start() threading.Thread(target=go, args=(chat_id, message, msg)).start()
def ytdl_download(url, tempdir, chat_id, message) -> dict: async def ytdl_download(url, tempdir, chat_id, message) -> dict:
response = dict(status=None, error=None, filepath=None) response = dict(status=None, error=None, filepath=None)
logging.info("Downloading for %s", url) logging.info("Downloading for %s", url)
output = os.path.join(tempdir, '%(title)s.%(ext)s') output = os.path.join(tempdir, '%(title)s.%(ext)s')
@ -103,7 +103,7 @@ async def echo_all(event):
temp_dir = tempfile.TemporaryDirectory() temp_dir = tempfile.TemporaryDirectory()
async with bot.action(chat_id, 'video'): async with bot.action(chat_id, 'video'):
result = ytdl_download(url, temp_dir.name, chat_id, message) result = await ytdl_download(url, temp_dir.name, chat_id, message)
if result["status"]: if result["status"]:
async with bot.action(chat_id, 'document'): async with bot.action(chat_id, 'document'):