Show More
request_wrapper.py
60 lines
| 1.9 KiB
| text/x-python
|
PythonLexer
r1 | # -*- coding: utf-8 -*- | |||
r3363 | # Copyright (C) 2016-2019 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 | ||||
r2351 | from rhodecode.lib.base import get_ip_addr, get_access_path, get_user_agent | |||
r1 | from rhodecode.lib.utils2 import safe_str | |||
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() | ||||
try: | ||||
response = self.handler(request) | ||||
finally: | ||||
end = time.time() | ||||
r2351 | total = end - start | |||
r3061 | log.info( | |||
r3430 | 'IP: %s %s Request to %s time: %.3fs [%s]', | |||
get_ip_addr(request.environ), request.environ.get('REQUEST_METHOD'), | ||||
r2351 | safe_str(get_access_path(request.environ)), total, | |||
r3061 | get_user_agent(request. environ) | |||
r1 | ) | |||
return response | ||||
def includeme(config): | ||||
config.add_tween( | ||||
'rhodecode.lib.middleware.request_wrapper.RequestWrapperTween', | ||||
r3430 | ) | |||