From b252343dc0c65a39a6e4eefd0824d1d5f57b208c Mon Sep 17 00:00:00 2001
From: Seaswimmer <seaswimmerthefsh@gmail.com>
Date: Tue, 4 Jun 2024 16:35:18 -0400
Subject: [PATCH] misc(aurora): allow `Moderation.log()` to skip returning the
 class object

---
 aurora/importers/aurora.py  | 1 +
 aurora/models/moderation.py | 6 ++++--
 2 files changed, 5 insertions(+), 2 deletions(-)

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)