xtaodada/ytdlbot
1
0
Fork 0
mirror of https://github.com/omg-xtao/ytdlbot.git synced 2024-11-16 03:45:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

add worker info for every user

Browse Source
This commit is contained in:
BennyThink 2022-02-05 18:23:28 +08:00
parent d6fff5033b
commit 3f0f31fc68
No known key found for this signature in database
GPG Key ID: 6CD0DBDA5235D481
1 changed files with 3 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
ytdlbot/tasks.py
View File

@ -210,16 +210,9 @@ def normal_audio(bot_msg):
os.unlink(flac_tmp)
def get_worker_status(username):
def get_worker_status():
worker_name = os.getenv("WORKER_NAME")
try:
me = celery_client.get_me()
mention = me.mention()
except Exception:
mention = "YouTube Downloader"
if worker_name and username == OWNER:
return f"Downloaded by {mention}-{worker_name}"
return f"Downloaded by {mention}"
return f"Downloaded by {worker_name}"
def ytdl_normal_download(bot_msg, client, url):
@ -248,7 +241,7 @@ def ytdl_normal_download(bot_msg, client, url):
remain = bot_text.remaining_quota_caption(chat_id)
size = sizeof_fmt(os.stat(video_path).st_size)
meta = get_metadata(video_path)
worker = get_worker_status(bot_msg.chat.username)
worker = "Downloaded by {}".format(os.getenv("WORKER_NAME", "Unknown"))
cap = f"`{filename}`\n\n{url}\n\nInfo: {meta['width']}x{meta['height']} {size} {meta['duration']}s" \
f"\n{remain}\n{worker}"
settings = get_user_settings(str(chat_id))