diff --git a/aurora/importers/aurora.py b/aurora/importers/aurora.py index c184c1c..fd26518 100644 --- a/aurora/importers/aurora.py +++ b/aurora/importers/aurora.py @@ -114,6 +114,7 @@ class ImportAuroraView(ui.View): changes=changes, metadata=metadata, database=database, + return_obj=False ) await interaction.edit_original_response(content="Import complete.") diff --git a/aurora/models/moderation.py b/aurora/models/moderation.py index 1b43ee8..73f7bf0 100644 --- a/aurora/models/moderation.py +++ b/aurora/models/moderation.py @@ -298,7 +298,8 @@ class Moderation(AuroraGuildModel): expired: bool | None = None, changes: list | None = None, metadata: dict | None = None, - ) -> "Moderation": + return_obj: bool = True, + ) -> "Moderation" | None: from ..utilities.database import connect from ..utilities.json import dumps if not timestamp: @@ -390,4 +391,5 @@ class Moderation(AuroraGuildModel): case["metadata"], ) - return cls.find_by_id(bot=bot, moderation_id=moderation_id, guild_id=guild_id) + if return_obj: + return cls.find_by_id(bot=bot, moderation_id=moderation_id, guild_id=guild_id)