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

filter forwarded messages, set api timeout to 100

Browse Source
This commit is contained in:
kiriharu 2021-01-04 22:03:04 +03:00
parent a050dcaf08
commit 0efd433363
2 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
apps/tgbot/tgbot/handlers/default/__init__.py
View File

@ -7,7 +7,7 @@ from .icmp import icmp_cmd
def setup(dp: Dispatcher): def setup(dp: Dispatcher):
dp.register_message_handler(start_cmd, commands=['start']) dp.register_message_handler(start_cmd, is_forwarded=False, commands=['start'])
dp.register_message_handler(web_cmd, commands=['web', 'http']) dp.register_message_handler(web_cmd, is_forwarded=False, commands=['web', 'http'])
dp.register_message_handler(whois_cmd, commands=['whois']) dp.register_message_handler(whois_cmd, is_forwarded=False, commands=['whois'])
dp.register_message_handler(icmp_cmd, commands=['icmp']) dp.register_message_handler(icmp_cmd, is_forwarded=False, commands=['icmp', 'ping'])

4
apps/tgbot/tgbot/handlers/helpers.py
View File

@ -1,4 +1,4 @@
from httpx import AsyncClient from httpx import AsyncClient, Timeout
from typing import List from typing import List
from core.coretypes import APINode from core.coretypes import APINode
@ -15,7 +15,7 @@ def check_int(value) -> bool:
async def send_api_requests(endpoint: str, data: dict, nodes: List[APINode]): async def send_api_requests(endpoint: str, data: dict, nodes: List[APINode]):
for node in nodes: for node in nodes:
data.update(dict(token=node.token)) data.update(dict(token=node.token))
async with AsyncClient() as client: async with AsyncClient(timeout=Timeout(timeout=100.0)) as client:
result = await client.get( result = await client.get(
f"{node.address}/{endpoint}", params=data f"{node.address}/{endpoint}", params=data
) )