You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
106 lines
3.0 KiB
Python
106 lines
3.0 KiB
Python
5 years ago
|
import json
|
||
|
|
import logging
|
||
|
|
import os.path
|
||
|
|
|
||
|
|
from gevent.queue import Queue
|
||
|
|
from geventwebsocket import WebSocketError
|
||
|
|
from werkzeug.exceptions import RequestTimeout
|
||
|
|
from werkzeug.wrappers import Response
|
||
|
|
from wsgigzip import gzip
|
||
|
|
|
||
|
|
|
||
|
|
# map between Python levels and the console method in Javascript
|
||
|
|
levels = {
|
||
|
|
logging.CRITICAL: 'error',
|
||
|
|
logging.ERROR: 'error',
|
||
|
|
logging.WARNING: 'warn',
|
||
|
|
logging.INFO: 'info',
|
||
|
|
logging.DEBUG: 'debug',
|
||
|
|
logging.NOTSET: 'log',
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
class DictHandler(logging.Handler):
|
||
|
|
def __init__(self, queue):
|
||
|
|
super().__init__()
|
||
|
|
self.queue = queue
|
||
|
|
|
||
|
|
def emit(self, record):
|
||
|
|
record.pathname = os.path.abspath(record.pathname)
|
||
|
|
message = {
|
||
|
|
'level': levels[record.levelno],
|
||
|
|
'content': record.msg,
|
||
|
|
}
|
||
|
|
try:
|
||
|
|
payload = json.dumps(message)
|
||
|
|
except TypeError:
|
||
|
|
message['content'] = repr(record.msg)
|
||
|
|
payload = json.dumps(message)
|
||
|
|
|
||
|
|
self.queue.put_nowait(payload)
|
||
|
|
|
||
|
|
|
||
|
|
JAVASCRIPT = """
|
||
|
|
console.log('Starting...');
|
||
|
|
|
||
|
|
const ws = new WebSocket("ws://{base}/__ws__");
|
||
|
|
ws.onmessage = function (event) {{
|
||
|
|
const msg = JSON.parse(event.data);
|
||
|
|
console[msg.level](msg.content);
|
||
|
|
}};
|
||
|
|
"""
|
||
|
|
|
||
|
|
|
||
|
|
class ConsoleLog:
|
||
|
|
def __init__(self, app, logger, js_path='/__console__.js'):
|
||
|
|
self.app = app
|
||
|
|
self.queue = Queue()
|
||
|
|
self.logger = logger
|
||
|
|
self.js_path = js_path
|
||
|
|
|
||
|
|
handler = DictHandler(self.queue)
|
||
|
|
self.logger.addHandler(handler)
|
||
|
|
|
||
|
|
def __call__(self, environ, start_response):
|
||
|
|
if 'wsgi.websocket' in environ:
|
||
|
|
ws = environ["wsgi.websocket"]
|
||
|
|
while not ws.closed:
|
||
|
|
message = self.queue.get()
|
||
|
|
try:
|
||
|
|
ws.send(message)
|
||
|
|
except WebSocketError:
|
||
|
|
break
|
||
|
|
raise RequestTimeout()
|
||
|
|
elif environ["PATH_INFO"] == self.js_path:
|
||
|
|
if environ.get('HTTP_HOST'):
|
||
|
|
base = environ['HTTP_HOST']
|
||
|
|
else:
|
||
|
|
host = environ['SERVER_NAME']
|
||
|
|
port = environ['SERVER_PORT']
|
||
|
|
base = ':'.join((host, port))
|
||
|
|
response = Response(JAVASCRIPT.format(base=base))
|
||
|
|
return response(environ, start_response)
|
||
|
|
|
||
|
|
# request non-compressed response
|
||
|
|
http_accept_encoding = environ.pop('HTTP_ACCEPT_ENCODING', '')
|
||
|
|
response = Response.from_app(self.app, environ)
|
||
|
|
|
||
|
|
# inject JS
|
||
|
|
if response.mimetype == 'text/html':
|
||
|
|
response = self.inject(response)
|
||
|
|
|
||
|
|
# compress response, if necessary
|
||
|
|
if http_accept_encoding:
|
||
|
|
environ['HTTP_ACCEPT_ENCODING'] = http_accept_encoding
|
||
|
|
response = gzip()(response)
|
||
|
|
|
||
|
|
return response(environ, start_response)
|
||
|
|
|
||
|
|
def inject(self, response):
|
||
|
|
code = '<script src="{}" async="async"></script>'.format(self.js_path)
|
||
|
|
data = response.get_data()
|
||
|
|
payload = data.decode(response.charset)
|
||
|
|
response.data = '{code}\n{payload}'.format(code=code, payload=payload)
|
||
|
|
|
||
|
|
return response
|