Coverage for faststream / _internal / producer.py: 69%

1from abc import abstractmethod (empty)

2from typing import TYPE_CHECKING, Any, Protocol (empty)

3 

4from faststream._internal.types import PublishCommandType_contra (empty)

5from faststream.exceptions import IncorrectState (empty)

6 

7if TYPE_CHECKING: (empty)

8 from faststream._internal.types import AsyncCallable 

9 from faststream.response import PublishCommand 

10 

11 

12class ProducerProto(Protocol[PublishCommandType_contra]): (empty)

13 _parser: "AsyncCallable" (empty)

14 _decoder: "AsyncCallable" (empty)

15 

16 @abstractmethod (empty)

17 async def publish(self, cmd: "PublishCommandType_contra") -> Any: (empty)

18 """Publishes a message asynchronously.""" 

19 ... 

20 

21 @abstractmethod (empty)

22 async def request(self, cmd: "PublishCommandType_contra") -> Any: (empty)

23 """Publishes a message synchronously.""" 

24 ... 

25 

26 @abstractmethod (empty)

27 async def publish_batch(self, cmd: "PublishCommandType_contra") -> Any: (empty)

28 """Publishes a messages batch asynchronously.""" 

29 ... 

30 

31 

32class ProducerFactory(Protocol): (empty)

33 def __call__( (empty)

34 self, 

35 parser: "AsyncCallable", (empty)

36 decoder: "AsyncCallable", (empty)

37 ) -> ProducerProto: ... (empty)

38 

39 

40class ProducerUnset(ProducerProto): (empty)

41 msg = "Producer is unset yet. You should set producer in broker initial method." (empty)

42 

43 def __bool__(self) -> bool: (empty)

44 return False 

45 

46 @property (empty)

47 def _parser(self) -> "AsyncCallable": (empty)

48 raise IncorrectState(self.msg) 

49 

50 @_parser.setter (empty)

51 def _parser(self, value: "AsyncCallable", /) -> "AsyncCallable": (empty)

52 raise IncorrectState(self.msg) 

53 

54 @property (empty)

55 def _decoder(self) -> "AsyncCallable": (empty)

56 raise IncorrectState(self.msg) 

57 

58 @_decoder.setter (empty)

59 def _decoder(self, value: "AsyncCallable", /) -> "AsyncCallable": (empty)

60 raise IncorrectState(self.msg) 

61 

62 async def publish(self, cmd: "PublishCommand") -> Any | None: (empty)

63 raise IncorrectState(self.msg) 

64 

65 async def request(self, cmd: "PublishCommand") -> Any: (empty)

66 raise IncorrectState(self.msg) 

67 

68 async def publish_batch(self, cmd: "PublishCommand") -> None: (empty)

69 raise IncorrectState(self.msg)