jarvis/jarvis-tasks
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add error handling to unban

Browse source
This commit is contained in:
Zeva Rose 2022-02-19 17:19:37 -07:00
parent 07975106ef
commit 23a5ac0849
1 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
jarvis_tasks/tasks/ban.py
View file

@ -4,6 +4,7 @@ from datetime import datetime, timedelta
from logging import Logger from logging import Logger
from dis_snek import Snake from dis_snek import Snake
from dis_snek.client.errors import NotFound
from jarvis_core.db import q from jarvis_core.db import q
from jarvis_core.db.models import Ban, Unban from jarvis_core.db.models import Ban, Unban
@ -25,14 +26,17 @@ async def unban(bot: Snake, logger: Logger) -> None:
user = await bot.fetch_user(ban.user) user = await bot.fetch_user(ban.user)
if guild and user: if guild and user:
logger.debug(f"Unbanned user {user.id} from guild {guild.id}") logger.debug(f"Unbanned user {user.id} from guild {guild.id}")
await guild.unban(user=user, reason="JARVIS tempban expired") try:
await guild.unban(user=user, reason="JARVIS tempban expired")
except NotFound:
logger.debug(f"User {user.id} not banned from guild {guild.id}")
ban.update(q(active=False)) ban.update(q(active=False))
await ban.commit() await ban.commit()
u = Unban( u = Unban(
user=user.id, user=user.id,
guild=guild.id, guild=guild.id,
username=user.name, username=user.username,
discrim=user.discriminator, discrim=user.discriminator,
admin=bot.user.id, admin=bot.user.id,
reason="Ban expired", reason="Ban expired",