nex/college-bot-v1
1
0
Fork 0
mirror of https://github.com/nexy7574/LCC-bot.git synced 2024-09-19 18:16:34 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add error handling to uptime checker

Browse source
This commit is contained in:
nex 2023-01-23 16:35:29 +00:00
parent fd9074f89e
commit 07e10f0eb5
1 changed files with 11 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
cogs/uptime.py
View file

@ -64,6 +64,7 @@ class UptimeCompetition(commands.Cog):
self.bot = bot
self.http = AsyncClient()
self._warning_posted = False
self.test_uptimes.add_exception_type(Exception)
self.test_uptimes.start()
self.last_result: list[UptimeEntry] = []
self.task_lock = asyncio.Lock()
@ -158,7 +159,11 @@ class UptimeCompetition(commands.Cog):
if max_retries := target.get("max_retries"):
request_kwargs["max_retries"] = max_retries
kwargs: Dict[str, str | int | None] = {"target_id": target["id"], "target": target["uri"], "notes": ""}
attempts, response = await self._test_url(target["uri"], **request_kwargs)
try:
attempts, response = await self._test_url(target["uri"], **request_kwargs)
except Exception as e:
response = e
attempts = 1
if isinstance(response, Exception):
kwargs["is_up"] = False
kwargs["response_time"] = None
@ -240,7 +245,11 @@ class UptimeCompetition(commands.Cog):
async def test_uptimes(self):
self.task_event.clear()
async with self.task_lock:
self.last_result = await self.do_test_uptimes()
try:
self.last_result = await self.do_test_uptimes()
except (Exception, ConnectionError):
print()
console.print_exception()
self.task_event.set()
uptime = discord.SlashCommandGroup("uptime", "Commands for the uptime competition.")