64 lines
2.2 KiB
Python
64 lines
2.2 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
|
|||
|
|
from tools.helpers import Parameters, delete_this, get_cmd_error, show_error
|
|||
|
|
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.init_for_speedtest('update')
|
|||
|
|
except asyncio.exceptions.TimeoutError:
|
|||
|
|
await msg.edit_text("⚠️ Update Timeout")
|
|||
|
|
except Exception as e:
|
|||
|
|
await show_error(msg, e)
|
|||
|
|
else:
|
|||
|
|
# TODO：有个未知错误
|
|||
|
|
await msg.edit_text(f"Result\n```{update_res}```")
|
|||
|
|
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 msg.edit_text("⚠️ Speedtest Timeout")
|
|||
|
|
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:
|
|||
|
|
await msg.edit_text("⚠️ Speedtest Timeout")
|
|||
|
|
return
|
|||
|
|
else:
|
|||
|
|
await msg.edit_text(get_cmd_error(cmd))
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
if not link:
|
|||
|
|
await msg.edit_text(text)
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
# 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_error(msg, e)
|
|||
|
|
# delete cmd history
|
|||
|
|
await delete_this(msg)
|