Cbor integration (#172)

Author: maxwellflitton

pyproject.toml

@@ -35,7 +35,6 @@ dependencies = [
     "aiosignal==1.3.2",
     "async-timeout==5.0.1",
     "attrs==25.1.0",
-    "cbor2==5.6.5",
     "Cerberus==1.3.7",
     "certifi==2024.12.14",
     "charset-normalizer==3.4.1",

requirements.txt

@@ -3,7 +3,6 @@ aiohttp==3.11.11
 aiosignal==1.3.2
 async-timeout==5.0.1
 attrs==25.1.0
-cbor2==5.6.5
 Cerberus==1.3.7
 certifi==2024.12.14
 charset-normalizer==3.4.1

src/surrealdb/cbor2/__init__.py

@@ -0,0 +1,27 @@
+from typing import Any
+
+from surrealdb.cbor2._decoder import CBORDecoder as CBORDecoder
+from surrealdb.cbor2._decoder import load as load
+from surrealdb.cbor2._decoder import loads as loads
+from surrealdb.cbor2._encoder import CBOREncoder as CBOREncoder
+from surrealdb.cbor2._encoder import dump as dump
+from surrealdb.cbor2._encoder import dumps as dumps
+from surrealdb.cbor2._encoder import shareable_encoder as shareable_encoder
+from surrealdb.cbor2._types import CBORDecodeEOF as CBORDecodeEOF
+from surrealdb.cbor2._types import CBORDecodeError as CBORDecodeError
+from surrealdb.cbor2._types import CBORDecodeValueError as CBORDecodeValueError
+from surrealdb.cbor2._types import CBOREncodeError as CBOREncodeError
+from surrealdb.cbor2._types import CBOREncodeTypeError as CBOREncodeTypeError
+from surrealdb.cbor2._types import CBOREncodeValueError as CBOREncodeValueError
+from surrealdb.cbor2._types import CBORError as CBORError
+from surrealdb.cbor2._types import CBORSimpleValue as CBORSimpleValue
+from surrealdb.cbor2._types import CBORTag as CBORTag
+from surrealdb.cbor2._types import FrozenDict as FrozenDict
+from surrealdb.cbor2._types import undefined as undefined
+
+# Re-export imports so they look like they live directly in this package
+key: str
+value: Any
+for key, value in list(locals().items()):
+    if callable(value) and getattr(value, "__module__", "").startswith("cbor2."):
+        value.__module__ = __name__