gnouseu/tuxbot-bot
4
0
Fork 0
Code Issues Pull requests Wiki Activity

fix(commands|Network>getheaders): fix timeout error

Browse source
This commit is contained in:
Romain J 2021-01-27 15:14:53 +01:00
parent 647cc4bd64
commit c5c13506d7
1 changed files with 9 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
tuxbot/cogs/Network/network.py
View file

@ -1,7 +1,9 @@
import asyncio
import functools import functools
import logging import logging
from typing import Union from typing import Union
import aiohttp
import discord import discord
from aiohttp import ClientConnectorError from aiohttp import ClientConnectorError
from discord.ext import commands from discord.ext import commands
@ -150,7 +152,11 @@ class Network(commands.Cog, name="Network"):
"5": 0x343A40, "5": 0x343A40,
} }
async with ctx.session.get(str(ip), headers=headers) as s: async with ctx.session.get(
str(ip),
headers=headers,
timeout=aiohttp.ClientTimeout(total=4),
) as s:
e = discord.Embed( e = discord.Embed(
title=f"Headers : {ip}", title=f"Headers : {ip}",
color=colors.get(str(s.status)[0], 0x6C757D), color=colors.get(str(s.status)[0], 0x6C757D),
@ -171,12 +177,12 @@ class Network(commands.Cog, name="Network"):
"[show all]({})", ctx, self.bot.config "[show all]({})", ctx, self.bot.config
).format(output["link"]) ).format(output["link"])
else: else:
value = f"```{output['text']}```" value = f"```\n{output['text']}```"
e.add_field(name=key, value=value, inline=True) e.add_field(name=key, value=value, inline=True)
await ctx.send(embed=e) await ctx.send(embed=e)
except ClientConnectorError: except (ClientConnectorError, asyncio.exceptions.TimeoutError):
await ctx.send( await ctx.send(
_("Cannot connect to host {}", ctx, self.bot.config).format(ip) _("Cannot connect to host {}", ctx, self.bot.config).format(ip)
) )