Skip to content

Commit acddf1c

Browse files
committed
Refactor message handling to use StatusMessage in queue operations
1 parent 77fa17d commit acddf1c

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

azure-kusto-ingest/azure/kusto/ingest/_status_q.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -37,15 +37,15 @@ def is_empty(self) -> bool:
3737
"""Checks if Status queue has any messages"""
3838
return len(self.peek(1, raw=True)) == 0
3939

40-
def _deserialize_message(self, m: QueueMessage) -> StatusMessage:
40+
def _deserialize_message(self, m: QueueMessage) -> "StatusMessage":
4141
"""Deserialize a message and return at as `message_cls`
4242
:param m: original message m.
4343
"""
4444
return self.message_cls(m.content)
4545

4646
# TODO: current implementation takes a union top n / len(queues), which is not ideal,
4747
# because the user is not supposed to know that there can be multiple underlying queues
48-
def peek(self, n=1, raw=False) -> List[StatusMessage]:
48+
def peek(self, n=1, raw=False) -> List["StatusMessage"]:
4949
"""Peek status queue
5050
:param int n: number of messages to return as part of peek.
5151
:param bool raw: should message content be returned as is (no parsing).
@@ -90,7 +90,7 @@ def _peek_specific_q(_q: QueueClient, _n: int) -> bool:
9090

9191
# TODO: current implementation takes a union top n / len(queues), which is not ideal,
9292
# because the user is not supposed to know that there can be multiple underlying queues
93-
def pop(self, n: int = 1, raw: bool = False, delete: bool = True) -> List[StatusMessage]:
93+
def pop(self, n: int = 1, raw: bool = False, delete: bool = True) -> List["StatusMessage"]:
9494
"""Pop status queue
9595
:param int n: number of messages to return as part of peek.
9696
:param bool raw: should message content be returned as is (no parsing).

0 commit comments

Comments
 (0)