Dev (#397)

src/red_utils/ext/sqlalchemy_utils/custom_types/type_classes.py

@@ -28,7 +28,12 @@ class SomeModel(Base):
 from sqlalchemy.dialects.postgresql import UUID
 from sqlalchemy.engine.interfaces import Dialect
 import sqlalchemy.orm as so
-from sqlalchemy.types import CHAR, TypeDecorator
+from sqlalchemy.types import CHAR
+from sqlalchemy import TypeDecorator, types
+
+import json
+
+
 
 class CompatibleUUID(TypeDecorator):
     """Define a custom UUID, overriding SQLAlchemy's UUId type.
@@ -81,3 +86,26 @@ def process_result_value(self, value: Any | None, dialect: Dialect) -> Any | Non
             if not isinstance(value, uuid.UUID):
                 value = uuid.UUID(value)
             return value
+
+
+
+class CustomJSON(TypeDecorator):
+    """Class to handle storing JSON in a database."""
+
+    @property
+    def python_type(self):
+        return object
+
+    impl = types.String
+
+    def process_bind_param(self, value, dialect):
+        return json.dumps(value)
+
+    def process_literal_param(self, value, dialect):
+        return value
+
+    def process_result_value(self, value, dialect):
+        try:
+            return json.loads(value)
+        except (ValueError, TypeError):
+            return None