cswimr/SeaCogs
cswimr/SeaCogs
1
1
Fork 1
Code Issues 11 Pull requests 9 Projects 1 Wiki Activity Actions

feat(aurora): boilerplate for all currently added moderation types
Some checks failed
Actions / Build Documentation (MkDocs) (pull_request) Failing after 26s
Details
Actions / Lint Code (Ruff & Pylint) (pull_request) Failing after 40s
Details

Browse source
This commit is contained in:
Seaswimmer 2024-07-06 13:03:59 -04:00
parent a05e957dde
commit 8f0425456c
Signed by: cswimr
GPG key ID: 3813315477F26F82
4 changed files with 201 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

19
aurora/importers/aurora.py
View file

@ -9,7 +9,9 @@ from redbot.core import commands, data_manager
from redbot.core.utils.chat_formatting import warning
from ..models.moderation import Moderation
from ..models.type import Type
from ..utilities.json import dump
from ..utilities.registry import type_registry
from ..utilities.utils import create_guild_table, timedelta_from_string
@ -40,23 +42,18 @@ class ImportAuroraView(ui.View):
file = await self.ctx.message.attachments[0].read()
data: list[dict] = sorted(json.loads(file), key=lambda x: x["moderation_id"])
user_mod_types = ["NOTE", "WARN", "ADDROLE", "REMOVEROLE", "MUTE", "UNMUTE", "KICK", "TEMPBAN", "BAN", "UNBAN"]
channel_mod_types = ["SLOWMODE", "LOCKDOWN"]
failed_cases = []
for case in data:
if case["moderation_id"] == 0:
continue
moderation_type: Type = type_registry[str.lower(case["moderation_type"])]
if "target_type" not in case or not case["target_type"]:
if case["moderation_type"] in user_mod_types:
case["target_type"] = "USER"
elif case["moderation_type"] in channel_mod_types:
case["target_type"] = "CHANNEL"
if moderation_type.channel:
case["target_type"] = "channel"
else:
case["target_type"] = "USER"
case["target_type"] = "user"
if "role_id" not in case or not case["role_id"]:
case["role_id"] = None
@ -95,8 +92,8 @@ class ImportAuroraView(ui.View):
bot=interaction.client,
guild_id=self.ctx.guild.id,
moderator_id=case["moderator_id"],
moderation_type=case["moderation_type"],
target_type=case["target_type"],
moderation_type=moderation_type.key,
target_type=case["target_type"].lower(),
target_id=case["target_id"],
role_id=case["role_id"],
duration=duration,