mirror of
https://github.com/morpheus65535/bazarr.git
synced 2024-11-15 12:14:51 +08:00
36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
import json
|
|
import sseclient
|
|
from ._transport import Transport
|
|
|
|
|
|
class ServerSentEventsTransport(Transport):
|
|
def __init__(self, session, connection):
|
|
Transport.__init__(self, session, connection)
|
|
self.__response = None
|
|
|
|
def _get_name(self):
|
|
return 'serverSentEvents'
|
|
|
|
def start(self):
|
|
connect_url = self._get_url('connect')
|
|
self.__response = iter(sseclient.SSEClient(connect_url, session=self._session))
|
|
self._session.get(self._get_url('start'))
|
|
|
|
def _receive():
|
|
try:
|
|
notification = next(self.__response)
|
|
except StopIteration:
|
|
return
|
|
else:
|
|
if notification.data != 'initialized':
|
|
self._handle_notification(notification.data)
|
|
|
|
return _receive
|
|
|
|
def send(self, data):
|
|
response = self._session.post(self._get_url('send'), data={'data': json.dumps(data)})
|
|
parsed = json.loads(response.content)
|
|
self._connection.received.fire(**parsed)
|
|
|
|
def close(self):
|
|
self._session.get(self._get_url('abort'))
|