speedtest.py

import speedtest
from main_startup.core.decorators import friday_on_cmd
from main_startup.helper_func.basic_helpers import edit_or_reply, get_text, edit_or_send_as_file, get_user


@friday_on_cmd(['speedtest', 'st'],
               cmd_help={
                'help': 'Test Your Server Speed.',
                'example': '{ch}speedtest'})
async def spee_test(client, message):
    ms_g = await edit_or_reply(message, "`Please Wait, Calculating Server Speed.`")
    s = speedtest.Speedtest()
    s.get_best_server()
    s.download()
    s.upload()
    response = s.results.dict()
    download_speed = response.get("download")
    upload_speed = response.get("upload")
    ping_time = response.get("ping")
    client_infos = response.get("client")
    img_ = s.results.share()
    _neat_test = f"""<b><u>SPEED TEST</b></u>
    
▶ <b>IP :</b> <code>{client_infos['ip']}</code>
▶ <b>ISP :</b> <code>{client_infos['isp']}</code>
▶ <b>Country :</b> <code>{client_infos['country']}</code>
▶ <b>Download Speed :</b> <code>{convert_from_bytes_to_human_formats(download_speed)}</code>
▶ <b>Upload Speed :</b> <code>{convert_from_bytes_to_human_formats(upload_speed)}</code>
▶ <b>ISP RATING :</b> <code>{client_infos['isprating']}</code>
▶ <b>PING TIME :</b> <code>{ping_time}</code>
"""
    if img_:
      if message.reply_to_message:
        await client.send_photo(
            message.chat.id,
            img_,
            reply_to_message_id=message.reply_to_message.message_id,
            caption=_neat_test
        )
      else:
        await client.send_photo(
            message.chat.id,
            img_,
            caption=_neat_test
            )
      await ms_g.delete()
    else:
      await ms_g.edit(_neat_test)


def convert_from_bytes_to_human_formats(size):
    power = 2**10
    n = 0
    units = {
        0: "",
        1: "KB",
        2: "MB",
        3: "GB",
        4: "TB"
    }
    while size > power:
        size /= power
        n += 1
    return f"{round(size, 2)} {units[n]}"