65 lines
2.3 KiB
Python
65 lines
2.3 KiB
Python
import asyncio
|
||
import re
|
||
|
||
from core import command
|
||
from pyrogram import Client
|
||
from pyrogram.errors import FloodWait
|
||
from pyrogram.types import Message
|
||
from tools.constants import SPEEDTEST_RUN, SYCGRAM_INFO
|
||
from tools.helpers import Parameters, delete_this, show_cmd_tip, show_exception
|
||
from tools.speedtests import Speedtester
|
||
|
||
|
||
@Client.on_message(command('speedtest'))
|
||
async def speedtest(_: Client, msg: Message):
|
||
"""服务器测速，用法：-speedtest <节点ID|list|update>"""
|
||
cmd, opt = Parameters.get(msg)
|
||
|
||
await msg.edit_text("⚡️ Speedtest is running.")
|
||
async with Speedtester() as tester:
|
||
if opt == 'update':
|
||
try:
|
||
update_res = await tester.install_speedtest_cli('update')
|
||
except asyncio.exceptions.TimeoutError:
|
||
await show_exception(msg, "Update Timeout！")
|
||
except Exception as e:
|
||
await show_exception(msg, e)
|
||
else:
|
||
await msg.edit_text(
|
||
f"**{SYCGRAM_INFO}**\n> # `{update_res}`",
|
||
parse_mode='md'
|
||
)
|
||
return
|
||
elif opt == 'list':
|
||
try:
|
||
text = await tester.list_servers_ids(f"{SPEEDTEST_RUN} -L")
|
||
await msg.edit_text(text, parse_mode='md')
|
||
except asyncio.exceptions.TimeoutError:
|
||
await show_exception(msg, "Speedtest Timeout")
|
||
except Exception as e:
|
||
await show_exception(msg, e)
|
||
return
|
||
elif bool(re.match(r'[0-9]+', opt)) or not opt:
|
||
try:
|
||
text, link = await tester.running(
|
||
f"""{SPEEDTEST_RUN}{'' if not opt else f' -s {opt}'}"""
|
||
)
|
||
except asyncio.exceptions.TimeoutError:
|
||
return await show_exception(msg, "Speedtest Timeout")
|
||
else:
|
||
return await show_cmd_tip(msg, cmd)
|
||
|
||
if not link:
|
||
return await msg.edit_text(text)
|
||
|
||
# send speed report
|
||
try:
|
||
await msg.reply_photo(photo=link, caption=text, parse_mode='md')
|
||
except FloodWait as e:
|
||
await asyncio.sleep(e.x)
|
||
await msg.reply_photo(photo=link, caption=text, parse_mode='md')
|
||
except Exception as e:
|
||
await show_exception(msg, e)
|
||
# delete cmd history
|
||
await delete_this(msg)
|