Exposed exceptions to Python API. (#22)

Author: s3rius

python/natsrpy/__init__.py

@@ -1,4 +1,11 @@
-from . import js
+from . import exceptions, js
 from ._natsrpy_rs import CallbackSubscription, IteratorSubscription, Message, Nats
 
-__all__ = ["CallbackSubscription", "IteratorSubscription", "Message", "Nats", "js"]
+__all__ = [
+    "CallbackSubscription",
+    "IteratorSubscription",
+    "Message",
+    "Nats",
+    "exceptions",
+    "js",
+]

python/natsrpy/_natsrpy_rs/__init__.pyi

@@ -4,7 +4,7 @@ from typing import Any, final, overload
 
 from typing_extensions import Self
 
-from . import js
+from . import exceptions, js
 
 @final
 class Message:
@@ -107,4 +107,11 @@ class Nats:
         backpressure_on_inflight: bool | None = None,
     ) -> js.JetStream: ...
 
-__all__ = ["CallbackSubscription", "IteratorSubscription", "Message", "Nats", "js"]
+__all__ = [
+    "CallbackSubscription",
+    "IteratorSubscription",
+    "Message",
+    "Nats",
+    "exceptions",
+    "js",
+]

python/natsrpy/_natsrpy_rs/exceptions.pyi

@@ -0,0 +1,9 @@
+class NatsrpyBaseError(Exception): ...
+class NatsrpySessionError(NatsrpyBaseError): ...
+class NatsrpyPublishError(NatsrpyBaseError): ...
+
+__all__ = [
+    "NatsrpyBaseError",
+    "NatsrpyPublishError",
+    "NatsrpySessionError",
+]

python/natsrpy/exceptions.py

@@ -0,0 +1,11 @@
+from ._natsrpy_rs.exceptions import (
+    NatsrpyBaseError,
+    NatsrpyPublishError,
+    NatsrpySessionError,
+)
+
+__all__ = [
+    "NatsrpyBaseError",
+    "NatsrpyPublishError",
+    "NatsrpySessionError",
+]

python/natsrpy/js.py

@@ -10,7 +10,13 @@
     ReplayPolicy,
 )
 from ._natsrpy_rs.js.kv import KeyValue, KVConfig
-from ._natsrpy_rs.js.object_store import ObjectStore, ObjectStoreConfig
+from ._natsrpy_rs.js.object_store import (
+    ObjectInfo,
+    ObjectInfoIterator,
+    ObjectLink,
+    ObjectStore,
+    ObjectStoreConfig,
+)
 from ._natsrpy_rs.js.stream import (
     Compression,
     ConsumerLimits,
@@ -38,6 +44,9 @@
     "JetStream",
     "KVConfig",
     "KeyValue",
+    "ObjectInfo",
+    "ObjectInfoIterator",
+    "ObjectLink",
     "ObjectStore",
     "ObjectStoreConfig",
     "PersistenceMode",

python/tests/conftest.py

@@ -27,9 +27,10 @@ async def nats(nats_url: str) -> AsyncGenerator[Nats, None]:
     nats = Nats(addrs=[nats_url])
     await nats.startup()
 
-    yield nats
-
-    await nats.shutdown()
+    try:
+        yield nats
+    finally:
+        await nats.shutdown()
 
 
 @pytest.fixture(scope="session")

python/tests/test_object_store.py

@@ -5,13 +5,11 @@
 from pathlib import Path
 
 import pytest
-from natsrpy._natsrpy_rs.js.object_store import (
+from natsrpy.js import (
+    JetStream,
     ObjectInfo,
     ObjectInfoIterator,
     ObjectLink,
-)
-from natsrpy.js import (
-    JetStream,
     ObjectStore,
     ObjectStoreConfig,
     StorageType,

src/js/kv.rs

@@ -181,18 +181,15 @@ impl KeyValue {
     ) -> NatsrpyResult<Bound<'py, PyAny>> {
         let store = self.store.clone();
         let data = bytes::Bytes::copy_from_slice(value.as_bytes());
-        natsrpy_future(py, async move {
-            let status = store.read().await.put(key, data).await?;
-            Ok(status)
-        })
+        natsrpy_future(
+            py,
+            async move { Ok(store.read().await.put(key, data).await?) },
+        )
     }
 
     pub fn delete<'py>(&self, py: Python<'py>, key: String) -> NatsrpyResult<Bound<'py, PyAny>> {
         let store = self.store.clone();
-        natsrpy_future(py, async move {
-            let kv = store.read().await;
-            Ok(kv.delete(key).await?)
-        })
+        natsrpy_future(py, async move { Ok(store.read().await.delete(key).await?) })
     }
 }
 

src/lib.rs

@@ -40,6 +40,9 @@ pub mod _natsrpy_rs {
     #[pymodule_export]
     use super::subscriptions::{callback::CallbackSubscription, iterator::IteratorSubscription};
 
+    #[pymodule_export]
+    use super::exceptions::py_err::pymod as exceptions;
+
     #[pymodule_export]
     use super::js::pymod as js;
 

src/utils/mod.rs

@@ -3,4 +3,5 @@ pub mod headers;
 pub mod py;
 pub mod py_types;
 pub mod streamer;
+
 pub use futures::natsrpy_future;