Show More
request_wrapper.py
65 lines
| 2.3 KiB
| text/x-python
|
PythonLexer
r1 | # -*- coding: utf-8 -*- | |||
r4306 | # Copyright (C) 2016-2020 RhodeCode GmbH | |||
r1 | # | |||
# This program is free software: you can redistribute it and/or modify | ||||
# it under the terms of the GNU Affero General Public License, version 3 | ||||
# (only), as published by the Free Software Foundation. | ||||
# | ||||
# This program is distributed in the hope that it will be useful, | ||||
# but WITHOUT ANY WARRANTY; without even the implied warranty of | ||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | ||||
# GNU General Public License for more details. | ||||
# | ||||
# You should have received a copy of the GNU Affero General Public License | ||||
# along with this program. If not, see <http://www.gnu.org/licenses/>. | ||||
# | ||||
# This program is dual-licensed. If you wish to learn more about the | ||||
# RhodeCode Enterprise Edition, including its added features, Support services, | ||||
# and proprietary license terms, please see https://rhodecode.com/licenses/ | ||||
import time | ||||
import logging | ||||
r3926 | import rhodecode | |||
r4347 | from rhodecode.lib.auth import AuthUser | |||
r2351 | from rhodecode.lib.base import get_ip_addr, get_access_path, get_user_agent | |||
r4347 | from rhodecode.lib.utils2 import safe_str, get_current_rhodecode_user | |||
r1 | ||||
log = logging.getLogger(__name__) | ||||
class RequestWrapperTween(object): | ||||
def __init__(self, handler, registry): | ||||
self.handler = handler | ||||
self.registry = registry | ||||
# one-time configuration code goes here | ||||
def __call__(self, request): | ||||
start = time.time() | ||||
r4157 | log.debug('Starting request time measurement') | |||
r1 | try: | |||
response = self.handler(request) | ||||
finally: | ||||
end = time.time() | ||||
r2351 | total = end - start | |||
r3874 | count = request.request_count() | |||
r3926 | _ver_ = rhodecode.__version__ | |||
r4347 | default_user_info = AuthUser.repr_user(ip=get_ip_addr(request.environ)) | |||
user_info = get_current_rhodecode_user(request) or default_user_info | ||||
r3061 | log.info( | |||
r4347 | 'Req[%4s] %s %s Request to %s time: %.4fs [%s], RhodeCode %s', | |||
count, user_info, request.environ.get('REQUEST_METHOD'), | ||||
r2351 | safe_str(get_access_path(request.environ)), total, | |||
r3926 | get_user_agent(request. environ), _ver_ | |||
r1 | ) | |||
return response | ||||
def includeme(config): | ||||
config.add_tween( | ||||
'rhodecode.lib.middleware.request_wrapper.RequestWrapperTween', | ||||
r3430 | ) | |||