mastoposter-oss_images/mastoreposter/sources.py

20 lines
622 B
Python
Raw Normal View History

2022-08-24 08:09:41 +03:00
from json import loads
from typing import AsyncGenerator
from urllib.parse import urlencode
from mastoreposter.types import Status
async def websocket_source(url: str, **params) -> AsyncGenerator[Status, None]:
from websockets.client import connect
url = f"{url}?" + urlencode({"stream": "list", **params})
async with connect(url) as ws:
while (msg := await ws.recv()) != None:
event = loads(msg)
if "error" in event:
raise Exception(event["error"])
if event["event"] == "update":
yield Status.from_dict(loads(event["payload"]))