Skip to content

Commit 3ab08e0

Browse files
committed
deprecate message_type in favor of message_types
1 parent 8aa9e6f commit 3ab08e0

File tree

1 file changed

+19
-2
lines changed

1 file changed

+19
-2
lines changed

src/aleph/sdk/client.py

Lines changed: 19 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
import queue
66
import threading
77
import time
8+
import warnings
89
from datetime import datetime
910
from pathlib import Path
1011
from typing import (
@@ -630,6 +631,7 @@ async def get_messages(
630631
pagination: int = 200,
631632
page: int = 1,
632633
message_type: Optional[MessageType] = None,
634+
message_types: Optional[List[MessageType]] = None,
633635
content_types: Optional[Iterable[str]] = None,
634636
content_keys: Optional[Iterable[str]] = None,
635637
refs: Optional[Iterable[str]] = None,
@@ -648,7 +650,8 @@ async def get_messages(
648650
649651
:param pagination: Number of items to fetch (Default: 200)
650652
:param page: Page to fetch, begins at 1 (Default: 1)
651-
:param message_type: Filter by message type, can be "AGGREGATE", "POST", "PROGRAM", "VM", "STORE" or "FORGET"
653+
:param message_type: [DEPRECATED] Filter by message type, can be "AGGREGATE", "POST", "PROGRAM", "VM", "STORE" or "FORGET"
654+
:param message_types: Filter by message types, can be any combination of "AGGREGATE", "POST", "PROGRAM", "VM", "STORE" or "FORGET"
652655
:param content_types: Filter by content type
653656
:param content_keys: Filter by content key
654657
:param refs: If set, only fetch posts that reference these hashes (in the "refs" field)
@@ -674,7 +677,13 @@ async def get_messages(
674677
params: Dict[str, Any] = dict(pagination=pagination, page=page)
675678

676679
if message_type is not None:
680+
warnings.warn(
681+
"The message_type parameter is deprecated, please use message_types instead.",
682+
DeprecationWarning,
683+
)
677684
params["msgType"] = message_type.value
685+
if message_types is not None:
686+
params["msgTypes"] = ",".join([t.value for t in message_types])
678687
if content_types is not None:
679688
params["contentTypes"] = ",".join(content_types)
680689
if content_keys is not None:
@@ -772,6 +781,7 @@ async def get_message(
772781
async def watch_messages(
773782
self,
774783
message_type: Optional[MessageType] = None,
784+
message_types: Optional[Iterable[MessageType]] = None,
775785
content_types: Optional[Iterable[str]] = None,
776786
refs: Optional[Iterable[str]] = None,
777787
addresses: Optional[Iterable[str]] = None,
@@ -785,7 +795,8 @@ async def watch_messages(
785795
"""
786796
Iterate over current and future matching messages asynchronously.
787797
788-
:param message_type: Type of message to watch
798+
:param message_type: [DEPRECATED] Type of message to watch
799+
:param message_types: Types of messages to watch
789800
:param content_types: Content types to watch
790801
:param refs: References to watch
791802
:param addresses: Addresses to watch
@@ -799,7 +810,13 @@ async def watch_messages(
799810
params: Dict[str, Any] = dict()
800811

801812
if message_type is not None:
813+
warnings.warn(
814+
"The message_type parameter is deprecated, please use message_types instead.",
815+
DeprecationWarning,
816+
)
802817
params["msgType"] = message_type.value
818+
if message_types is not None:
819+
params["msgTypes"] = ",".join([t.value for t in message_types])
803820
if content_types is not None:
804821
params["contentTypes"] = ",".join(content_types)
805822
if refs is not None:

0 commit comments

Comments
 (0)