161 lines
6.4 KiB
Python
161 lines
6.4 KiB
Python
from telegram import InlineKeyboardButton, InlineKeyboardMarkup
|
|
from telegram.ext import CallbackContext, CallbackQueryHandler
|
|
from telegram.message import Message
|
|
from telegram.update import Update
|
|
import psutil, shutil
|
|
import time
|
|
from bot import AUTO_DELETE_MESSAGE_DURATION, LOGGER, bot, \
|
|
status_reply_dict, status_reply_dict_lock, download_dict, download_dict_lock, botStartTime, dispatcher
|
|
from bot.helper.ext_utils.bot_utils import get_readable_message, get_readable_file_size, get_readable_time, MirrorStatus
|
|
from telegram.error import TimedOut, BadRequest
|
|
|
|
|
|
def sendMessage(text: str, bot, update: Update):
|
|
try:
|
|
return bot.send_message(update.message.chat_id,
|
|
reply_to_message_id=update.message.message_id,
|
|
text=text, allow_sending_without_reply=True, parse_mode='HTMl')
|
|
except Exception as e:
|
|
LOGGER.error(str(e))
|
|
def sendMarkup(text: str, bot, update: Update, reply_markup: InlineKeyboardMarkup):
|
|
return bot.send_message(update.message.chat_id,
|
|
reply_to_message_id=update.message.message_id,
|
|
text=text, reply_markup=reply_markup, allow_sending_without_reply=True, parse_mode='HTMl')
|
|
|
|
def editMessage(text: str, message: Message, reply_markup=None):
|
|
try:
|
|
bot.edit_message_text(text=text, message_id=message.message_id,
|
|
chat_id=message.chat.id,reply_markup=reply_markup,
|
|
parse_mode='HTMl')
|
|
except Exception as e:
|
|
LOGGER.error(str(e))
|
|
|
|
|
|
def deleteMessage(bot, message: Message):
|
|
try:
|
|
bot.delete_message(chat_id=message.chat.id,
|
|
message_id=message.message_id)
|
|
except Exception as e:
|
|
LOGGER.error(str(e))
|
|
|
|
|
|
def sendLogFile(bot, update: Update):
|
|
with open('log.txt', 'rb') as f:
|
|
bot.send_document(document=f, filename=f.name,
|
|
reply_to_message_id=update.message.message_id,
|
|
chat_id=update.message.chat_id)
|
|
|
|
|
|
def auto_delete_message(bot, cmd_message: Message, bot_message: Message):
|
|
if AUTO_DELETE_MESSAGE_DURATION != -1:
|
|
time.sleep(AUTO_DELETE_MESSAGE_DURATION)
|
|
try:
|
|
# Skip if None is passed meaning we don't want to delete bot xor cmd message
|
|
deleteMessage(bot, cmd_message)
|
|
deleteMessage(bot, bot_message)
|
|
except AttributeError:
|
|
pass
|
|
|
|
|
|
def delete_all_messages():
|
|
with status_reply_dict_lock:
|
|
for message in list(status_reply_dict.values()):
|
|
try:
|
|
deleteMessage(bot, message)
|
|
del status_reply_dict[message.chat.id]
|
|
except Exception as e:
|
|
LOGGER.error(str(e))
|
|
|
|
|
|
def update_all_messages():
|
|
msg = get_readable_message()
|
|
with download_dict_lock:
|
|
dlspeed_bytes = 0
|
|
uldl_bytes = 0
|
|
for download in list(download_dict.values()):
|
|
speedy = download.speed()
|
|
if download.status() == MirrorStatus.STATUS_DOWNLOADING:
|
|
if 'K' in speedy:
|
|
dlspeed_bytes += float(speedy.split('K')[0]) * 1024
|
|
elif 'M' in speedy:
|
|
dlspeed_bytes += float(speedy.split('M')[0]) * 1048576
|
|
if download.status() == MirrorStatus.STATUS_UPLOADING:
|
|
if 'KB/s' in speedy:
|
|
uldl_bytes += float(speedy.split('K')[0]) * 1024
|
|
elif 'MB/s' in speedy:
|
|
uldl_bytes += float(speedy.split('M')[0]) * 1048576
|
|
dlspeed = get_readable_file_size(dlspeed_bytes)
|
|
ulspeed = get_readable_file_size(uldl_bytes)
|
|
msg += f"<b>DL:</b> {dlspeed}ps 🔻 | <b>UL:</b> {ulspeed}ps 🔺"
|
|
with status_reply_dict_lock:
|
|
for chat_id in list(status_reply_dict.keys()):
|
|
if status_reply_dict[chat_id] and msg != status_reply_dict[chat_id].text:
|
|
if len(msg) == 0:
|
|
msg = "Starting DL"
|
|
try:
|
|
keyboard = [[InlineKeyboardButton("♻️ Stats", callback_data="stats_")]]
|
|
editMessage(msg, status_reply_dict[chat_id], reply_markup=InlineKeyboardMarkup(keyboard))
|
|
except Exception as e:
|
|
LOGGER.error(str(e))
|
|
status_reply_dict[chat_id].text = msg
|
|
|
|
|
|
def sendStatusMessage(msg, bot):
|
|
progress = get_readable_message()
|
|
with download_dict_lock:
|
|
dlspeed_bytes = 0
|
|
uldl_bytes = 0
|
|
for download in list(download_dict.values()):
|
|
speedy = download.speed()
|
|
if download.status() == MirrorStatus.STATUS_DOWNLOADING:
|
|
if 'K' in speedy:
|
|
dlspeed_bytes += float(speedy.split('K')[0]) * 1024
|
|
elif 'M' in speedy:
|
|
dlspeed_bytes += float(speedy.split('M')[0]) * 1048576
|
|
if download.status() == MirrorStatus.STATUS_UPLOADING:
|
|
if 'KB/s' in speedy:
|
|
uldl_bytes += float(speedy.split('K')[0]) * 1024
|
|
elif 'MB/s' in speedy:
|
|
uldl_bytes += float(speedy.split('M')[0]) * 1048576
|
|
dlspeed = get_readable_file_size(dlspeed_bytes)
|
|
ulspeed = get_readable_file_size(uldl_bytes)
|
|
progress += f"<b>DL:</b> {dlspeed}ps 🔻 | <b>UL:</b> {ulspeed}ps 🔺"
|
|
with status_reply_dict_lock:
|
|
if msg.message.chat.id in list(status_reply_dict.keys()):
|
|
try:
|
|
message = status_reply_dict[msg.message.chat.id]
|
|
deleteMessage(bot, message)
|
|
del status_reply_dict[msg.message.chat.id]
|
|
except Exception as e:
|
|
LOGGER.error(str(e))
|
|
del status_reply_dict[msg.message.chat.id]
|
|
pass
|
|
if len(progress) == 0:
|
|
progress = "Starting DL"
|
|
message = sendMessage(progress, bot, msg)
|
|
status_reply_dict[msg.message.chat.id] = message
|
|
|
|
|
|
def pop_up_stats(update, context):
|
|
query = update.callback_query
|
|
stats = bot_sys_stats()
|
|
query.answer(text=stats, show_alert=True)
|
|
|
|
def bot_sys_stats():
|
|
currentTime = get_readable_time(time.time() - botStartTime)
|
|
cpu = psutil.cpu_percent()
|
|
mem = psutil.virtual_memory().percent
|
|
disk = psutil.disk_usage("/").percent
|
|
total, used, free = shutil.disk_usage('.')
|
|
free = get_readable_file_size(free)
|
|
stats = f"""
|
|
Bot Uptime: {currentTime}
|
|
Free Disk: {free}
|
|
CPU: {cpu}%
|
|
RAM: {mem}%
|
|
DISK: {disk}%
|
|
"""
|
|
return stats
|
|
|
|
dispatcher.add_handler(CallbackQueryHandler(pop_up_stats, pattern="stats_"))
|