hkc
/
mastoposter
2
0
Fork 2
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Set websocket open timeout to 1 minute and log connection attempts #3

Merged
hkc merged 3 commits from vftdan/mastoposter:log-websocket-connection into master 2024-06-12 23:32:35 +03:00
Conversation 3 Commits 3 Files Changed 3 +8 -2
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 3e82b5c979 - Show all commits

10
mastoposter/sources.py
View File

@ -28,10 +28,16 @@ async def websocket_source(
from websockets.client import connect from websockets.client import connect
from websockets.exceptions import WebSocketException from websockets.exceptions import WebSocketException
url = f"{url}?" + urlencode({"stream": "list", **params}) param_dict = {"stream": "list", **params}
public_param_dict = param_dict.copy()
public_param_dict["access_token"] = 'SCRUBBED'
public_url = f"{url}?" + urlencode(public_param_dict)
url = f"{url}?" + urlencode(param_dict)
while True: while True:
try: try:
async with connect(url) as ws: logger.info("attempting to connect to %s", public_url)
async with connect(url, open_timeout=60) as ws:
hkc commented 2024-06-08 18:08:49 +03:00
Outdated
Review
Copy Link

maybe respect timeout option from main config here?

maybe respect timeout option from main config here?
logger.info("Connected to WebSocket")
while (msg := await ws.recv()) is not None: while (msg := await ws.recv()) is not None:
event = loads(msg) event = loads(msg)
logger.debug("data: %r", event) logger.debug("data: %r", event)