From 2dfc9d98240e45c2f6ccc9710d34f7ba69bbe661 Mon Sep 17 00:00:00 2001
From: SeaswimmerTheFsh <seaswimmerthefsh@gmail.com>
Date: Sat, 4 May 2024 15:37:07 -0400
Subject: [PATCH] fix(aurora): use AuroraBaseModels for the JSONEncoder class
 instead of just pydantic ones to prevent issues with other data types

---
 aurora/models.py         | 5 ++++-
 aurora/utilities/json.py | 4 ++--
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/aurora/models.py b/aurora/models.py
index 5ee33c5..62b5ea7 100644
--- a/aurora/models.py
+++ b/aurora/models.py
@@ -5,7 +5,10 @@ from typing import Dict, List, Optional
 from pydantic import BaseModel
 
 
-class Moderation(BaseModel):
+
+class AuroraBaseModel(BaseModel):
+    """Base class for all models in Aurora."""
+class Moderation(AuroraBaseModel):
     moderation_id: int
     guild_id: int
     timestamp: datetime
diff --git a/aurora/utilities/json.py b/aurora/utilities/json.py
index c3e79a3..e8678af 100644
--- a/aurora/utilities/json.py
+++ b/aurora/utilities/json.py
@@ -1,7 +1,7 @@
 import json
 from datetime import datetime, timedelta
 
-from pydantic import BaseModel
+from aurora.models import AuroraBaseModel
 
 
 class JSONEncoder(json.JSONEncoder):
@@ -10,7 +10,7 @@ class JSONEncoder(json.JSONEncoder):
             return int(o.timestamp())
         if isinstance(o, timedelta):
             return str(o)
-        if isinstance(o, BaseModel):
+        if isinstance(o, AuroraBaseModel):
             return o.model_dump()
         return super().default(o)