Fixed stub problems.

Author: s3rius

python/natsrpy/_natsrpy_rs/__init__.pyi

@@ -2,10 +2,36 @@ from collections.abc import Awaitable, Callable
 from datetime import timedelta
 from typing import Any, final, overload
 
-from natsrpy._natsrpy_rs.js import JetStream
-from natsrpy._natsrpy_rs.message import Message
 from typing_extensions import Self
 
+from . import js
+
+@final
+class Message:
+    """
+    Simple NATS message.
+
+    Attributes:
+        subject: subject where message was published
+        reply: subject where reply should be sent, if any
+        payload: message payload
+        headers: dictionary of message headers,
+            every value can be a simple value or a list.
+        status: status is used for reply messages to indicate the status of the reply.
+            It is None for regular messages.
+        description: message description is used for reply messages to
+            provide additional information about the status.
+        length: a length of the message payload in bytes.
+    """
+
+    subject: str
+    reply: str | None
+    payload: bytes
+    headers: dict[str, Any]
+    status: int | None
+    description: str | None
+    length: int
+
 @final
 class IteratorSubscription:
     def __aiter__(self) -> IteratorSubscription: ...
@@ -69,6 +95,6 @@ class Nats:
         subject: str,
         callback: None = None,
     ) -> IteratorSubscription: ...
-    async def jetstream(self) -> JetStream: ...
+    async def jetstream(self) -> js.JetStream: ...
 
-__all__ = ["CallbackSubscription", "IteratorSubscription", "Message", "Nats"]
+__all__ = ["CallbackSubscription", "IteratorSubscription", "Message", "Nats", "js"]

python/natsrpy/_natsrpy_rs/js/__init__.pyi

@@ -1,15 +1,29 @@
 from datetime import datetime, timedelta
-from typing import Any, Literal, overload
+from typing import Any, Literal, final, overload
 
+from . import consumers, kv, managers, object_store, stream
 from .managers import KVManager, ObjectStoreManager, StreamsManager
 
+__all__ = [
+    "JetStream",
+    "JetStreamMessage",
+    "Publication",
+    "consumers",
+    "kv",
+    "managers",
+    "object_store",
+    "stream",
+]
+
+@final
 class Publication:
     stream: str
     sequence: int
     domain: str
     duplicate: bool
     value: str | None
 
+@final
 class JetStream:
     @overload
     async def publish(
@@ -48,6 +62,7 @@ class JetStream:
     @property
     def object_store(self) -> ObjectStoreManager: ...
 
+@final
 class JetStreamMessage:
     @property
     def subject(self) -> str: ...

python/natsrpy/_natsrpy_rs/js/consumers.pyi

@@ -1,8 +1,22 @@
 from datetime import timedelta
+from typing import final
 
 from natsrpy._natsrpy_rs.js import JetStreamMessage
 from typing_extensions import Self
 
+__all__ = [
+    "AckPolicy",
+    "DeliverPolicy",
+    "MessagesIterator",
+    "PriorityPolicy",
+    "PullConsumer",
+    "PullConsumerConfig",
+    "PushConsumer",
+    "PushConsumerConfig",
+    "ReplayPolicy",
+]
+
+@final
 class DeliverPolicy:
     ALL: DeliverPolicy
     LAST: DeliverPolicy
@@ -11,21 +25,25 @@ class DeliverPolicy:
     BY_START_TIME: DeliverPolicy
     LAST_PER_SUBJECT: DeliverPolicy
 
+@final
 class AckPolicy:
     EXPLICIT: AckPolicy
     NONE: AckPolicy
     ALL: AckPolicy
 
+@final
 class ReplayPolicy:
     INSTANT: ReplayPolicy
     ORIGINAL: ReplayPolicy
 
+@final
 class PriorityPolicy:
     NONE: PriorityPolicy
     OVERFLOW: PriorityPolicy
     PINNED_CLIENT: PriorityPolicy
     PRIORITIZED: PriorityPolicy
 
+@final
 class PullConsumerConfig:
     name: str | None
     durable_name: str | None
@@ -88,6 +106,7 @@ class PullConsumerConfig:
         pause_until: int | None = None,
     ) -> Self: ...
 
+@final
 class PushConsumerConfig:
     deliver_subject: str
     name: str | None
@@ -148,6 +167,7 @@ class PushConsumerConfig:
         pause_until: int | None = None,
     ) -> Self: ...
 
+@final
 class MessagesIterator:
     def __aiter__(self) -> MessagesIterator: ...
     async def __anext__(self) -> JetStreamMessage: ...
@@ -156,9 +176,11 @@ class MessagesIterator:
         timeout: float | timedelta | None = None,
     ) -> JetStreamMessage: ...
 
+@final
 class PushConsumer:
     async def messages(self) -> MessagesIterator: ...
 
+@final
 class PullConsumer:
     async def fetch(
         self,

python/natsrpy/_natsrpy_rs/js/kv.pyi

@@ -1,6 +1,14 @@
+from typing import final
+
 from natsrpy._natsrpy_rs.js.stream import Placement, Republish, Source, StorageType
 from typing_extensions import Self
 
+__all__ = [
+    "KVConfig",
+    "KeyValue",
+]
+
+@final
 class KVConfig:
     """
     KV bucket config.
@@ -43,6 +51,7 @@ class KVConfig:
         limit_markers: float | None = None,
     ) -> Self: ...
 
+@final
 class KeyValue:
     @property
     def stream_name(self) -> str: ...

python/natsrpy/_natsrpy_rs/js/managers.pyi

@@ -1,5 +1,5 @@
 from datetime import timedelta
-from typing import overload
+from typing import final, overload
 
 from .consumers import (
     PullConsumer,
@@ -11,20 +11,30 @@ from .kv import KeyValue, KVConfig
 from .object_store import ObjectStore, ObjectStoreConfig
 from .stream import Stream, StreamConfig
 
+__all__ = [
+    "ConsumersManager",
+    "KVManager",
+    "ObjectStoreManager",
+    "StreamsManager",
+]
+
+@final
 class StreamsManager:
     async def create(self, config: StreamConfig) -> Stream: ...
     async def create_or_update(self, config: StreamConfig) -> Stream: ...
     async def get(self, name: str) -> Stream: ...
     async def delete(self, name: str) -> bool: ...
     async def update(self, config: StreamConfig) -> Stream: ...
 
+@final
 class KVManager:
     async def create(self, config: KVConfig) -> KeyValue: ...
     async def create_or_update(self, config: KVConfig) -> KeyValue: ...
     async def get(self, bucket: str) -> KeyValue: ...
     async def delete(self, bucket: str) -> bool: ...
     async def update(self, config: KVConfig) -> KeyValue: ...
 
+@final
 class ConsumersManager:
     @overload
     async def create(self, config: PullConsumerConfig) -> PullConsumer: ...
@@ -40,6 +50,7 @@ class ConsumersManager:
     async def pause(self, name: str, delay: float | timedelta) -> bool: ...
     async def resume(self, name: str) -> bool: ...
 
+@final
 class ObjectStoreManager:
     async def create(self, config: ObjectStoreConfig) -> ObjectStore: ...
     async def get(self, bucket: str) -> ObjectStore: ...

python/natsrpy/_natsrpy_rs/js/object_store.pyi

@@ -1,9 +1,16 @@
 from datetime import timedelta
+from typing import final
 
 from typing_extensions import Self, Writer
 
 from .stream import Placement, StorageType
 
+__all__ = [
+    "ObjectStore",
+    "ObjectStoreConfig",
+]
+
+@final
 class ObjectStoreConfig:
     bucket: str
     description: str | None
@@ -26,18 +33,19 @@ class ObjectStoreConfig:
         placement: Placement | None = None,
     ) -> Self: ...
 
+@final
 class ObjectStore:
     async def get(
         self,
         name: str,
         writer: Writer[bytes],
-        chunk_size: int | None = 24576,  # 24MB
+        chunk_size: int | None = 24576,
     ) -> None: ...
     async def put(
         self,
         name: str,
         value: bytes | str,
-        chunk_size: int = 24576,  # 24MB
+        chunk_size: int = 24576,
         description: str | None = None,
         headers: dict[str, str | list[str]] | None = None,
         metadata: dict[str, str] | None = None,