catspace-dev/unicheckbot
1
1
Fork 0
mirror of https://github.com/catspace-dev/unicheckbot.git synced 2024-11-22 05:46:33 +03:00
Code Issues Packages Projects Releases Wiki Activity

rewrited commands

Browse Source
This commit is contained in:
kiriharu 2021-01-13 22:24:09 +03:00
parent 352d27f8b2
commit 8afcb2e087
4 changed files with 9 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
apps/tgbot/tgbot/handlers/default/icmp.py
View File

@ -28,7 +28,8 @@ class ICMPCheckerHandler(CheckerBaseHandler):
return await message.answer(icmp_help_message, parse_mode="Markdown") return await message.answer(icmp_help_message, parse_mode="Markdown")
except LocalhostForbidden: except LocalhostForbidden:
return await message.answer(self.localhost_forbidden_message, parse_mode="Markdown") return await message.answer(self.localhost_forbidden_message, parse_mode="Markdown")
await self.check(message.chat.id, message.bot, dict(target=args[0], target_fq=args[0])) else:
await self.check(message.chat.id, message.bot, dict(target=args[0], target_fq=args[0]))
async def process_args(self, text: str) -> list: async def process_args(self, text: str) -> list:
args = text.split() args = text.split()

10
apps/tgbot/tgbot/handlers/default/minecraft.py
View File

@ -1,10 +1,8 @@
from aiogram.types import Message
from core.coretypes import ResponseStatus, ErrorPayload, MinecraftResponse from core.coretypes import ResponseStatus, ErrorPayload, MinecraftResponse
from httpx import Response from httpx import Response
from tgbot.handlers.base import CheckerBaseHandler, process_args_for_host_port from tgbot.handlers.base import CheckerTargetPortHandler, process_args_for_host_port
from tgbot.handlers.metrics import push_status_metric from tgbot.handlers.metrics import push_status_metric
from tgbot.middlewares.throttling import rate_limit
minecraft_help_message = """ minecraft_help_message = """
❓ Получает статистику о Minecraft сервере ❓ Получает статистику о Minecraft сервере
@ -18,17 +16,13 @@ minecraft_help_message = """
invalid_port = """❗Неправильный порт. Напишите /minecraft чтобы увидеть справку к данному способу проверки.""" invalid_port = """❗Неправильный порт. Напишите /minecraft чтобы увидеть справку к данному способу проверки."""
class MinecraftCheckerHandler(CheckerBaseHandler): class MinecraftCheckerHandler(CheckerTargetPortHandler):
help_message = minecraft_help_message help_message = minecraft_help_message
api_endpoint = "minecraft" api_endpoint = "minecraft"
def __init__(self): def __init__(self):
super().__init__() super().__init__()
@rate_limit
async def handler(self, message: Message):
await self.target_port_handler(message)
async def process_args(self, text: str) -> list: async def process_args(self, text: str) -> list:
return process_args_for_host_port(text, 25565) return process_args_for_host_port(text, 25565)

6
apps/tgbot/tgbot/handlers/default/tcp.py
View File

@ -2,7 +2,7 @@ from aiogram.types import Message
from core.coretypes import ResponseStatus, ErrorPayload, PortResponse from core.coretypes import ResponseStatus, ErrorPayload, PortResponse
from httpx import Response from httpx import Response
from tgbot.handlers.base import CheckerBaseHandler, NotEnoughArgs, InvalidPort from tgbot.handlers.base import CheckerTargetPortHandler, NotEnoughArgs, InvalidPort
from tgbot.handlers.helpers import check_int from tgbot.handlers.helpers import check_int
from tgbot.handlers.metrics import push_status_metric from tgbot.handlers.metrics import push_status_metric
from tgbot.middlewares.throttling import rate_limit from tgbot.middlewares.throttling import rate_limit
@ -17,7 +17,7 @@ tcp_help_message = """
invalid_port = """❗Неправильный порт. Напишите /tcp чтобы увидеть справку к данному способу проверки.""" invalid_port = """❗Неправильный порт. Напишите /tcp чтобы увидеть справку к данному способу проверки."""
class TCPCheckerHandler(CheckerBaseHandler): class TCPCheckerHandler(CheckerTargetPortHandler):
help_message = tcp_help_message help_message = tcp_help_message
api_endpoint = "tcp_port" api_endpoint = "tcp_port"
@ -26,7 +26,7 @@ class TCPCheckerHandler(CheckerBaseHandler):
@rate_limit @rate_limit
async def handler(self, message: Message): async def handler(self, message: Message):
await self.target_port_handler(message) await super(TCPCheckerHandler, self).handler(message)
async def process_args(self, text: str) -> list: async def process_args(self, text: str) -> list:
port = None port = None

10
apps/tgbot/tgbot/handlers/default/web.py
View File

@ -1,9 +1,7 @@
from aiogram.types import Message
from httpx import Response from httpx import Response
from core.coretypes import ResponseStatus, HTTP_EMOJI, HttpCheckerResponse, ErrorPayload from core.coretypes import ResponseStatus, HTTP_EMOJI, HttpCheckerResponse, ErrorPayload
from ..base import CheckerBaseHandler, process_args_for_host_port from ..base import CheckerTargetPortHandler, process_args_for_host_port
from ..metrics import push_status_metric from ..metrics import push_status_metric
from tgbot.middlewares.throttling import rate_limit
web_help_message = """ web_help_message = """
❓ Производит проверку хоста по протоколу HTTP. ❓ Производит проверку хоста по протоколу HTTP.
@ -16,17 +14,13 @@ web_help_message = """
invalid_port = """❗Неправильный порт. Напишите /web чтобы увидеть справку к данному способу проверки.""" invalid_port = """❗Неправильный порт. Напишите /web чтобы увидеть справку к данному способу проверки."""
class WebCheckerHandler(CheckerBaseHandler): class WebCheckerHandler(CheckerTargetPortHandler):
help_message = web_help_message help_message = web_help_message
api_endpoint = "http" api_endpoint = "http"
def __init__(self): def __init__(self):
super().__init__() super().__init__()
@rate_limit
async def handler(self, message: Message):
await self.target_port_handler(message)
async def process_args(self, text: str) -> list: async def process_args(self, text: str) -> list:
return process_args_for_host_port(text, 80) return process_args_for_host_port(text, 80)