Improve pika (#13739)

Author: donbarbos

stubs/pika/pika/__init__.pyi

@@ -1,3 +1,5 @@
+from typing import Final
+
 from pika import adapters as adapters
 from pika.adapters import (
     BaseConnection as BaseConnection,
@@ -9,3 +11,5 @@ from pika.connection import ConnectionParameters as ConnectionParameters, SSLOpt
 from pika.credentials import PlainCredentials as PlainCredentials
 from pika.delivery_mode import DeliveryMode as DeliveryMode
 from pika.spec import BasicProperties as BasicProperties
+
+__version__: Final[str]

stubs/pika/pika/adapters/asyncio_connection.pyi

@@ -1,12 +1,11 @@
-from _typeshed import Incomplete
 from asyncio import AbstractEventLoop
 from collections.abc import Callable
 from logging import Logger
 from typing_extensions import Self
 
 from ..connection import Parameters
 from .base_connection import BaseConnection
-from .utils import io_services_utils, nbio_interface
+from .utils import connection_workflow, io_services_utils, nbio_interface
 
 LOGGER: Logger
 
@@ -22,7 +21,11 @@ class AsyncioConnection(BaseConnection):
     ) -> None: ...
     @classmethod
     def create_connection(
-        cls, connection_configs, on_done, custom_ioloop: AbstractEventLoop | None = None, workflow: Incomplete | None = None
+        cls,
+        connection_configs,
+        on_done,
+        custom_ioloop: AbstractEventLoop | None = None,
+        workflow: connection_workflow.AbstractAMQPConnectionWorkflow | None = None,
     ): ...
 
 class _AsyncioIOServicesAdapter(
@@ -31,7 +34,7 @@ class _AsyncioIOServicesAdapter(
     nbio_interface.AbstractIOServices,
     nbio_interface.AbstractFileDescriptorServices,
 ):
-    def __init__(self, loop: Incomplete | None = None) -> None: ...
+    def __init__(self, loop: AbstractEventLoop | None = None) -> None: ...
     def get_native_ioloop(self): ...
     def close(self) -> None: ...
     def run(self) -> None: ...

stubs/pika/pika/adapters/base_connection.pyi

@@ -1,12 +1,13 @@
 import abc
 from _typeshed import Incomplete
 from collections.abc import Callable
+from logging import Logger
 from typing_extensions import Self
 
 from ..adapters.utils import nbio_interface
 from ..connection import Connection
 
-LOGGER: Incomplete
+LOGGER: Logger
 
 class BaseConnection(Connection, metaclass=abc.ABCMeta):
     def __init__(

stubs/pika/pika/adapters/blocking_connection.pyi

@@ -1,5 +1,6 @@
 from _typeshed import Incomplete, Unused
 from collections.abc import Generator, Sequence
+from logging import Logger
 from types import TracebackType
 from typing import NamedTuple
 from typing_extensions import Self
@@ -9,7 +10,7 @@ from ..data import _ArgumentMapping
 from ..exchange_type import ExchangeType
 from ..spec import BasicProperties
 
-LOGGER: Incomplete
+LOGGER: Logger
 
 class _CallbackResult:
     def __init__(self, value_class: Incomplete | None = None) -> None: ...

stubs/pika/pika/adapters/gevent_connection.pyi

@@ -1,10 +1,11 @@
 from _typeshed import Incomplete
+from logging import Logger
 
 from pika.adapters.base_connection import BaseConnection
 from pika.adapters.utils.nbio_interface import AbstractIOReference
 from pika.adapters.utils.selector_ioloop_adapter import AbstractSelectorIOLoop, SelectorIOServicesAdapter
 
-LOGGER: Incomplete
+LOGGER: Logger
 
 class GeventConnection(BaseConnection):
     def __init__(

stubs/pika/pika/adapters/select_connection.pyi

@@ -1,11 +1,12 @@
 import abc
 from _typeshed import Incomplete
+from logging import Logger
 
 import pika.compat
 from pika.adapters.base_connection import BaseConnection
 from pika.adapters.utils.selector_ioloop_adapter import AbstractSelectorIOLoop
 
-LOGGER: Incomplete
+LOGGER: Logger
 SELECT_TYPE: Incomplete
 
 class SelectConnection(BaseConnection):

stubs/pika/pika/adapters/tornado_connection.pyi

@@ -1,8 +1,9 @@
 from _typeshed import Incomplete
+from logging import Logger
 
 from pika.adapters import base_connection
 
-LOGGER: Incomplete
+LOGGER: Logger
 
 class TornadoConnection(base_connection.BaseConnection):
     def __init__(

stubs/pika/pika/adapters/twisted_connection.pyi

@@ -2,6 +2,7 @@
 # We don't want to force it as a dependency but that means we also can't test it with type-checkers given the current setup.
 
 from _typeshed import Incomplete
+from logging import Logger
 from typing import Generic, NamedTuple, TypeVar
 
 import pika.connection
@@ -17,7 +18,7 @@ from twisted.python.failure import Failure  # type: ignore[import-not-found]  #
 
 _T = TypeVar("_T")
 
-LOGGER: Incomplete
+LOGGER: Logger
 
 class ClosableDeferredQueue(DeferredQueue[_T], Generic[_T]):  # pyright: ignore[reportUntypedBaseClass]  # noqa: Y060
     closed: Failure | BaseException | None

stubs/pika/pika/adapters/utils/selector_ioloop_adapter.pyi

@@ -1,9 +1,10 @@
 import abc
 from _typeshed import Incomplete
+from logging import Logger
 
 from pika.adapters.utils import io_services_utils, nbio_interface
 
-LOGGER: Incomplete
+LOGGER: Logger
 
 class AbstractSelectorIOLoop(metaclass=abc.ABCMeta):
     @property

stubs/pika/pika/callback.pyi

@@ -1,8 +1,13 @@
 from _typeshed import Incomplete
+from collections.abc import Callable
+from logging import Logger
+from typing import Literal
 
-LOGGER: Incomplete
+from pika import amqp_object, frame
 
-def name_or_value(value): ...
+LOGGER: Logger
+
+def name_or_value(value: amqp_object.AMQPObject | frame.Frame | int | str) -> str: ...
 def sanitize_prefix(function): ...
 def check_for_prefix_and_key(function): ...
 
@@ -16,16 +21,22 @@ class CallbackManager:
     def __init__(self) -> None: ...
     def add(
         self,
-        prefix,
-        key,
-        callback,
+        prefix: str | int,
+        key: str | object,
+        callback: Callable[[Incomplete], Incomplete],
         one_shot: bool = True,
-        only_caller: Incomplete | None = None,
+        only_caller: object | None = None,
         arguments: Incomplete | None = None,
-    ): ...
+    ) -> tuple[str | int, str | object]: ...
     def clear(self) -> None: ...
-    def cleanup(self, prefix): ...
-    def pending(self, prefix, key): ...
-    def process(self, prefix, key, caller, *args, **keywords): ...
-    def remove(self, prefix, key, callback_value: Incomplete | None = None, arguments: Incomplete | None = None): ...
-    def remove_all(self, prefix, key) -> None: ...
+    def cleanup(self, prefix: str | int) -> bool: ...
+    def pending(self, prefix: str | int, key: str | object) -> int | None: ...
+    def process(self, prefix: str | int, key: str | object, caller, *args, **keywords) -> bool: ...
+    def remove(
+        self,
+        prefix: str | int,
+        key: str | object,
+        callback_value: Callable[[Incomplete], Incomplete] | None = None,
+        arguments: Incomplete | None = None,
+    ) -> Literal[True]: ...
+    def remove_all(self, prefix: str | int, key: str | object) -> None: ...