refactor: move serialization helpers into one place (#40)

advanced_alchemy/_serialization.py

@@ -0,0 +1,24 @@
+from typing import Any
+
+__all__ = ("encode_json", "decode_json")
+
+try:
+    from msgspec.json import Decoder, Encoder
+
+    encoder, decoder = Encoder(), Decoder()
+    decode_json = decoder.decode
+
+    def encode_json(data: Any) -> str:
+        return encoder.encode(data).decode("utf-8")
+
+except ImportError:
+    try:
+        from orjson import dumps as _encode_json
+        from orjson import loads as decode_json  # type: ignore[no-redef]
+
+        def encode_json(data: Any) -> str:
+            return _encode_json(data).decode("utf-8")  # type: ignore[no-any-return]
+
+    except ImportError:
+        from json import dumps as encode_json  # type: ignore[assignment]
+        from json import loads as decode_json  # type: ignore[assignment]

advanced_alchemy/config/engine.py

@@ -3,6 +3,7 @@
 from dataclasses import dataclass
 from typing import TYPE_CHECKING, Callable, Literal
 
+from advanced_alchemy._serialization import decode_json, encode_json
 from advanced_alchemy.config.types import Empty
 
 if TYPE_CHECKING:
@@ -14,26 +15,6 @@
 
     from advanced_alchemy.config.types import EmptyType
 
-try:
-    from msgspec.json import Decoder, Encoder
-
-    encoder, decoder = Encoder(), Decoder()
-    decode_json = decoder.decode
-
-    def encode_json(data: Any) -> str:
-        return encoder.encode(data).decode("utf-8")
-
-except ImportError:
-    try:
-        from orjson import dumps as _encode_json
-        from orjson import loads as decode_json  # type: ignore[no-redef]
-
-        def encode_json(data: Any) -> str:
-            return _encode_json(data).decode("utf-8")  # type: ignore[no-any-return]
-
-    except ImportError:
-        from json import dumps as encode_json  # type: ignore[assignment]
-        from json import loads as decode_json  # type: ignore[assignment]
 
 __all__ = ("EngineConfig",)
 

advanced_alchemy/types.py

@@ -5,8 +5,6 @@
 from base64 import b64decode
 from typing import TYPE_CHECKING, Any, cast
 
-from msgspec.json import decode as decode_json
-from msgspec.json import encode as encode_json
 from sqlalchemy import DateTime, text, util
 from sqlalchemy.dialects.oracle import BLOB as ORA_BLOB
 from sqlalchemy.dialects.oracle import RAW as ORA_RAW
@@ -15,6 +13,8 @@
 from sqlalchemy.types import BINARY, CHAR, BigInteger, Integer, SchemaType, TypeDecorator, TypeEngine
 from sqlalchemy.types import JSON as _JSON
 
+from advanced_alchemy._serialization import decode_json, encode_json
+
 if TYPE_CHECKING:
     from sqlalchemy.engine import Dialect