cswimr/SeaCogs
cswimr/SeaCogs
1
1
Fork 1
Code Issues 10 Pull requests 4 Projects 1 Packages 1 Wiki Activity Actions

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

Browse source
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
1 changed files with 18 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
aurora/models/moderation_types.py
View file

@ -241,7 +241,12 @@ class AddRole(Type):
@classmethod @classmethod
async def expiry_handler(cls, moderation: Moderation) -> int: async def expiry_handler(cls, moderation: Moderation) -> int:
try: 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( await target.remove_roles(
Object(moderation.role_id), reason=f"Automatic role removal from case #{moderation.id}" Object(moderation.role_id), reason=f"Automatic role removal from case #{moderation.id}"
@ -409,7 +414,12 @@ class RemoveRole(Type):
@classmethod @classmethod
async def expiry_handler(cls, moderation: Moderation) -> int: async def expiry_handler(cls, moderation: Moderation) -> int:
try: 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( await target.add_roles(
Object(moderation.role_id), reason=f"Automatic role addition from case #{moderation.id}" 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: async def expiry_handler(cls, moderation: Moderation) -> int:
reason = f"Automatic {Unban.string} from case #{moderation.id}" reason = f"Automatic {Unban.string} from case #{moderation.id}"
try: 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) await moderation.guild.unban(user=target, reason=reason)
if await config.guild(moderation.guild).dm_users() is True: if await config.guild(moderation.guild).dm_users() is True: