Speedtest & Usage

README.md

@@ -67,6 +67,8 @@ This project is heavily inspired from @out386 's telegram bot which is written i
 - Shortener support
 - No Duplicate mirrors
 - Extract password protected files (It's not hack, you have to enter password for extracting. LOL)
+- Speedtest
+- Heroku dyno usage
 
 - For extracting password protected files, using custom filename and download from password protected index links see these examples :-
 > https://telegra.ph/Magneto-Python-Aria---Custom-Filename-Examples-01-20
@@ -91,6 +93,7 @@ stats - Bot Usage Stats
 help - Get Detailed Help
 speedtest - Check Speed of the host (Not working currently!)
 log - Bot Log [owner only]
+usage - check heroku dyno usage [owner only]
 ```
 
 <details>

bot/__init__.py

@@ -142,6 +142,16 @@ def getConfig(name: str):
     MAX_TORRENT_SIZE = int(getConfig("MAX_TORRENT_SIZE"))
 except KeyError:
     MAX_TORRENT_SIZE = None
+try:
+    HEROKU_API_KEY = getConfig('HEROKU_API_KEY')
+except KeyError:
+    logging.warning('HEROKU API KEY not provided!')
+    HEROKU_API_KEY = None
+try:
+    HEROKU_APP_NAME = getConfig('HEROKU_APP_NAME')
+except KeyError:
+    logging.warning('HEROKU APP NAME not provided!')
+    HEROKU_APP_NAME = None
 try:
    ENABLE_FILESIZE_LIMIT = getConfig('ENABLE_FILESIZE_LIMIT')
    if ENABLE_FILESIZE_LIMIT.lower() == 'true':

bot/__main__.py

@@ -12,7 +12,7 @@
 from bot.helper.telegram_helper.message_utils import LOGGER, editMessage, sendLogFile, sendMessage
 from .helper.ext_utils.bot_utils import get_readable_file_size, get_readable_time
 from .helper.telegram_helper.filters import CustomFilters
-from .modules import authorize, list, cancel_mirror, mirror_status, mirror, clone, watch, delete # noqa
+from .modules import authorize, list, cancel_mirror, mirror_status, mirror, clone, watch, delete, speedtest, usage # noqa
 
 from pyrogram import idle
 from bot import app
@@ -100,6 +100,7 @@ def bot_help(update, context):
 /{BotCommands.LogCommand}: Get a log file of the bot. Handy for getting crash reports
 
 /{BotCommands.SpeedCommand}: Check Internet Speed of the Host
+/{BotCommands.UsageCommand}: To see Heroku Dyno Stats (Owner only).
 '''
     sendMessage(help_string, context.bot, update)
 

bot/helper/telegram_helper/bot_commands.py

@@ -20,5 +20,7 @@ def __init__(self):
         self.WatchCommand = 'watch'
         self.TarWatchCommand = 'tarwatch'
         self.deleteCommand = 'del'
+        self.UsageCommand = 'usage'
+        self.SpeedCommand = 'speedtest'
 
 BotCommands = _BotCommands()

bot/modules/speedtest.py

@@ -1,13 +1,13 @@
+from speedtest import Speedtest
 from bot.helper.telegram_helper.filters import CustomFilters
-from bot import dispatcher, AUTHORIZED_CHATS
+from bot import dispatcher
 from bot.helper.telegram_helper.bot_commands import BotCommands
-from telegram import Update, InlineKeyboardButton, InlineKeyboardMarkup, ParseMode
-from telegram.ext import CallbackContext, Filters, run_async, CommandHandler
+from telegram import ParseMode
+from bot.helper.telegram_helper.message_utils import *
+from telegram.ext import CommandHandler
 
-run_async
 def speedtest(update, context):
-    message = update.effective_message
-    ed_msg = message.reply_text("𝓡𝓾𝓷𝓷𝓲𝓷𝓰 𝓢𝓹𝓮𝓮𝓭 𝓣𝓮𝓼𝓽 . . . ")
+    speed = sendMessage("𝓡𝓾𝓷𝓷𝓲𝓷𝓰 𝓢𝓹𝓮𝓮𝓭 𝓣𝓮𝓼𝓽 . . . ", context.bot, update)
     test = Speedtest()
     test.get_best_server()
     test.download()
@@ -26,7 +26,7 @@ def speedtest(update, context):
 <b>📶 Ping:</b> <code>{result['ping']} ms</code>
 <b>🏬 ISP:</b> <code>{result['client']['isp']}</code>
 '''
-    ed_msg.delete()
+    speed.delete()
     try:
         update.effective_message.reply_photo(path, string_speed, parse_mode=ParseMode.HTML)
     except:
@@ -44,6 +44,6 @@ def speed_convert(size):
 
 
 SPEED_HANDLER = CommandHandler(BotCommands.SpeedCommand, speedtest, 
-                                                  filters=CustomFilters.authorized_chat | CustomFilters.authorized_user)
+                                                  filters=CustomFilters.owner_filter | CustomFilters.authorized_user, run_async=True)
 
 dispatcher.add_handler(SPEED_HANDLER)

requirements.txt

@@ -8,11 +8,13 @@ js2py
 lxml
 megasdkrestclient
 psutil
+heroku3
 Pyrogram
 python-dotenv
 python-magic
 python-telegram-bot
 requests
+speedtest-cli
 telegraph
 tenacity
 TgCrypto