fix(aurora): fixed some broken expiry handlers
Some checks failed
Actions / Build Documentation (MkDocs) (pull_request) Successful in 32s
Actions / Lint Code (Ruff & Pylint) (pull_request) Failing after 48s

This commit is contained in:
Seaswimmer 2024-08-13 01:30:47 -04:00
parent a6124dd9ca
commit 009b406ff6
Signed by: cswimr
GPG key ID: 3813315477F26F82

View file

@ -241,7 +241,12 @@ class AddRole(Type):
@classmethod
async def expiry_handler(cls, moderation: Moderation) -> int:
try:
target = await moderation.get_target()
target = moderation.bot.get_user(moderation.target_id)
if not target:
try:
await moderation.guild.fetch_member(moderation.target_id)
except NotFound:
return 0
await target.remove_roles(
Object(moderation.role_id), reason=f"Automatic role removal from case #{moderation.id}"
@ -409,7 +414,12 @@ class RemoveRole(Type):
@classmethod
async def expiry_handler(cls, moderation: Moderation) -> int:
try:
target = await moderation.get_target()
target = moderation.bot.get_user(moderation.target_id)
if not target:
try:
await moderation.guild.fetch_member(moderation.target_id)
except NotFound:
return 0
await target.add_roles(
Object(moderation.role_id), reason=f"Automatic role addition from case #{moderation.id}"
@ -881,7 +891,12 @@ class Tempban(Ban):
async def expiry_handler(cls, moderation: Moderation) -> int:
reason = f"Automatic {Unban.string} from case #{moderation.id}"
try:
target = await moderation.get_target()
target = moderation.bot.get_user(moderation.target_id)
if not target:
try:
await moderation.bot.fetch_user(moderation.target_id)
except NotFound:
return 0
await moderation.guild.unban(user=target, reason=reason)
if await config.guild(moderation.guild).dm_users() is True: