Tiny code to decode Discord WS Zlib ETF messages (doesn't always work)

decode.py

import zlib
import base64
import earl

ZLIB_SUFFIX = b'\x00\x00\xff\xff'
buffer = bytearray()

def uncompress(str):
    str = base64.decodebytes(str)
    buffer.extend(str)

    if len(str) >= 4 or str[-4:] == ZLIB_SUFFIX:
        str = zlib.decompressobj().decompress(buffer)
        return str

def decode(str):
    str = base64.decodebytes(str)
    str = earl.unpack(str)
    return str

def main (obj):
    if (obj["type"] == "received"):
        return decode(uncompress(obj["text"]))
    elif (obj["type"] == "sent"):
        return decode(obj["text"])

msg = {
    "text": ""
    #,"type": "sent"
    #,"type": "received"
    #uncomment the type, sent by the client or received by the client
}

msg["text"] = msg["text"].encode("ascii")

print(main(msg))