73 lines
2.2 KiB
Python
73 lines
2.2 KiB
Python
# Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors
|
|
# MIT License. See license.txt
|
|
|
|
from __future__ import unicode_literals
|
|
from cProfile import Profile
|
|
from pstats import Stats
|
|
import json
|
|
import uuid
|
|
|
|
import frappe
|
|
import os
|
|
from werkzeug.exceptions import NotFound
|
|
from werkzeug.wsgi import SharedDataMiddleware
|
|
from frappe.utils import get_site_name, get_site_path, get_site_base_path, get_path, cstr
|
|
|
|
class StaticDataMiddleware(SharedDataMiddleware):
|
|
def __call__(self, environ, start_response):
|
|
self.environ = environ
|
|
return super(StaticDataMiddleware, self).__call__(environ, start_response)
|
|
|
|
def get_directory_loader(self, directory):
|
|
def loader(path):
|
|
site = get_site_name(frappe.app._site or self.environ.get('HTTP_HOST'))
|
|
path = os.path.join(directory, site, 'public', 'files', cstr(path))
|
|
if os.path.isfile(path):
|
|
return os.path.basename(path), self._opener(path)
|
|
else:
|
|
raise NotFound
|
|
# return None, None
|
|
|
|
return loader
|
|
|
|
|
|
class RecorderMiddleware(object):
|
|
def __init__(self, app):
|
|
self._app = app
|
|
|
|
def __call__(self, environ, start_response):
|
|
response_body = []
|
|
|
|
def persist(uuid, stats):
|
|
# Keys of stats.stats are tuples, JSON doesn't like that, str(key) works
|
|
# First four entries in values are interesting numbers
|
|
# Last value is a dictionary of mapping of callers to tuple of number same four numbers
|
|
# We don't actually need this redundancy
|
|
_stats = {
|
|
str(key): value[:4] + tuple(value[4]) for key, value in stats.stats.items()
|
|
}
|
|
frappe.cache().set("recorder-stats-{}".format(uuid), json.dumps(_stats))
|
|
|
|
def catching_start_response(status, headers, exc_info=None):
|
|
start_response(status, headers, exc_info)
|
|
return response_body.append
|
|
|
|
def runapp():
|
|
appiter = self._app(environ, catching_start_response)
|
|
response_body.extend(appiter)
|
|
if hasattr(appiter, 'close'):
|
|
appiter.close()
|
|
|
|
# Every request is assigned a uuid here,
|
|
# uuid is available to everyone as frappe.request.environ["uuid"]
|
|
# SQL calls and profile details can't be recorded and accessed without this
|
|
environ["uuid"] = str(uuid.uuid1())
|
|
|
|
profile = Profile()
|
|
profile.runcall(runapp)
|
|
body = [b''.join(response_body)]
|
|
stats = Stats(profile)
|
|
|
|
persist(environ["uuid"], stats)
|
|
|
|
return body
|