openmedialibrary/oml/websocket.py

89 lines
2.7 KiB
Python
Raw Normal View History

2014-05-04 17:26:43 +00:00
# -*- coding: utf-8 -*-
2014-09-02 22:32:44 +00:00
2014-05-04 17:26:43 +00:00
from tornado.websocket import WebSocketHandler
from tornado.ioloop import IOLoop
import json
from oxtornado import json_dumps
2014-05-04 17:26:43 +00:00
import state
2015-12-02 21:05:23 +00:00
import settings
2014-05-04 17:26:43 +00:00
2014-05-18 03:01:24 +00:00
import logging
2015-11-29 14:56:38 +00:00
logger = logging.getLogger(__name__)
2014-05-17 14:26:59 +00:00
2014-05-04 17:26:43 +00:00
class Handler(WebSocketHandler):
2019-01-17 10:30:22 +00:00
def initialize(self, public=False):
self._public = public
def check_origin(self, origin):
# allow access to websocket from site, installer and loader (local file)
2015-11-02 11:14:44 +00:00
return self.request.host in origin or \
origin in (
'http://127.0.0.1:9841',
'http://127.0.0.1:9842',
2015-11-03 17:32:53 +00:00
'file://',
'null'
)
2014-05-04 17:26:43 +00:00
def open(self):
if self.request.headers['origin'] not in ('null', 'file://', 'http://127.0.0.1:9842') \
and self.request.host not in self.request.headers['origin']:
2014-05-17 14:26:59 +00:00
logger.debug('reject cross site attempt to open websocket %s', self.request)
2014-05-14 09:57:11 +00:00
self.close()
if self not in state.websockets:
state.websockets.append(self)
if state.update:
2016-02-12 12:53:02 +00:00
trigger_event('updatestatus', state.update._status)
else:
from user.models import User
trigger_event('status', {
'id': settings.USER_ID,
'online': state.online
})
for u in User.query.filter(User.id!=settings.USER_ID).filter_by(peered=True).all():
u.trigger_status()
2014-05-04 17:26:43 +00:00
#websocket calls
def on_message(self, message):
2019-02-02 07:27:46 +00:00
try:
action, data = json.loads(message)
except json.decoder.JSONDecodeError:
logger.debug('invalid websocket message: %s', message)
return
if state.tasks:
state.tasks.queue(action, data)
2014-05-04 17:26:43 +00:00
def on_close(self):
if self in state.websockets:
state.websockets.remove(self)
def post(self, event, data):
message = json_dumps([event, data])
2015-11-17 18:23:21 +00:00
if self.ws_connection is None:
self.on_close()
else:
2019-02-02 07:27:46 +00:00
state.main.add_callback(lambda: self._write_message(message))
async def _write_message(self, message):
try:
task = self.write_message(message)
await task
except tornado.iostream.StreamClosedError as e:
self.on_close()
except tornado.websocket.WebSocketClosedError as e:
self.on_close()
2014-05-04 17:26:43 +00:00
def trigger_event(event, data):
2016-02-23 18:00:56 +00:00
#if len(state.websockets):
# logger.debug('trigger event %s %s %s', event, data, len(state.websockets))
2014-05-04 17:26:43 +00:00
for ws in state.websockets:
try:
ws.post(event, data)
2014-05-04 17:26:43 +00:00
except:
2016-01-24 09:13:03 +00:00
logger.debug('failed to send to ws %s %s %s', ws, event, data, exc_info=True)