Skip to content

Commit

Permalink
suppress error by overriding class func (blakeblackshear#8431)
Browse files Browse the repository at this point in the history
  • Loading branch information
hawkeye217 committed Nov 2, 2023
1 parent aefecad commit fc36be4
Showing 1 changed file with 14 additions and 1 deletion.
15 changes: 14 additions & 1 deletion frigate/comms/ws.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
"""Websocket communicator."""

import errno
import json
import logging
import threading
Expand All @@ -12,14 +13,26 @@
WSGIServer,
)
from ws4py.server.wsgiutils import WebSocketWSGIApplication
from ws4py.websocket import WebSocket
from ws4py.websocket import WebSocket as WebSocket_

from frigate.comms.dispatcher import Communicator
from frigate.config import FrigateConfig

logger = logging.getLogger(__name__)


class WebSocket(WebSocket_):
def unhandled_error(self, error):
"""
Handles the unfriendly socket closures on the server side
without showing a confusing error message
"""
if hasattr(error, "errno") and error.errno == errno.ECONNRESET:
pass
else:
logging.getLogger("ws4py").exception("Failed to receive data")


class WebSocketClient(Communicator): # type: ignore[misc]
"""Frigate wrapper for ws client."""

Expand Down

0 comments on commit fc36be4

Please sign in to comment.