WebSocket Connection

import zcatalyst_sdk
from zcatalyst_sdk.datastreams.websocket_client import DataStreamsWebSocket
from zcatalyst_sdk.types.datastreams import DataStreamsConfig
# Initialize SDK
app_sdk = zcatalyst_sdk.initialize_app()
data_streams = app_sdk.datastreams()
# Get token pair (required to authenticate WebSocket)
token = data_streams.get_token_pair("CHANNEL_ID", user_id="USER_ID")
# Create WebSocket config
config = DataStreamsConfig(
    url=token["url"],
    zuid=token["wss-id"],
    key=token["key"],
    enable_logging=True,
)
# Create WebSocket client
websocket_client = DataStreamsWebSocket(config=config)
def on_open(event):
    # Subscribe after connection is established
    websocket_client.subscribe("0")
def on_message(event):
    # event.data: inline payload
    # event.url + event.method: bulk payload retrieval info
    print("streaming_id:", event.streaming_id)
    print("data:", event.data)
    print("url:", event.url)
    print("method:", event.method)
    # Acknowledge to receive the next message
    websocket_client.send_ack()
def on_error(event):
    # CustomEvent with code/message/data
    print("error:", event.code, event.message, event.data)
def on_close(event):
    # Close details (if any)
    print("closed:", event)
websocket_client.set_on_open(on_open)
websocket_client.set_on_message(on_message)
websocket_client.set_on_error(on_error)
websocket_client.set_on_close(on_close)
# Connect (runs WebSocket in a background thread)
websocket_client.connect()