##// END OF EJS Templates
pytest: Update docstrings to reflect changes.
pytest: Update docstrings to reflect changes.

File last commit:

r1:854a839a default
r216:3ec05fb7 default
Show More
wsgi_app_caller_client.py
98 lines | 3.3 KiB | text/x-python | PythonLexer
/ rhodecode / lib / middleware / utils / wsgi_app_caller_client.py
# -*- coding: utf-8 -*-
# Copyright (C) 2012-2016 RhodeCode GmbH
#
# 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/
"""
Utility to call a WSGI app wrapped in a WSGIAppCaller object.
"""
import logging
from Pyro4.errors import ConnectionClosedError
log = logging.getLogger(__name__)
def _get_clean_environ(environ):
"""Return a copy of the WSGI environment without wsgi.* keys.
It also omits any non-string values.
:param environ: WSGI environment to clean
:type environ: dict
:returns: WSGI environment to pass to WSGIAppCaller.handle.
:rtype: dict
"""
clean_environ = dict(
(k, v) for k, v in environ.iteritems()
if type(v) == str and type(k) == str and not k.startswith('wsgi.')
)
return clean_environ
# pylint: disable=too-few-public-methods
class RemoteAppCaller(object):
"""Create and calls a remote WSGI app using the given factory.
It first cleans the environment, so as to reduce the data transferred.
"""
def __init__(self, remote_wsgi, *args, **kwargs):
"""
:param remote_wsgi: The remote wsgi object that creates a
WSGIAppCaller. This object
has to have a handle method, with the signature:
handle(environ, start_response, *args, **kwargs)
:param args: args to be passed to the app creation
:param kwargs: kwargs to be passed to the app creation
"""
self._remote_wsgi = remote_wsgi
self._args = args
self._kwargs = kwargs
def __call__(self, environ, start_response):
"""
:param environ: WSGI environment with which the app will be run
:type environ: dict
:param start_response: callable of WSGI protocol
:type start_response: callable
:returns: an iterable with the data returned by the app
:rtype: iterable<str>
"""
log.debug("Forwarding WSGI request via proxy %s", self._remote_wsgi)
input_data = environ['wsgi.input'].read()
clean_environ = _get_clean_environ(environ)
try:
data, status, headers = self._remote_wsgi.handle(
clean_environ, input_data, *self._args, **self._kwargs)
except ConnectionClosedError:
log.debug('Remote Pyro Server ConnectionClosedError')
self._remote_wsgi._pyroReconnect(tries=15)
data, status, headers = self._remote_wsgi.handle(
clean_environ, input_data, *self._args, **self._kwargs)
log.debug("Got result from proxy, returning to WSGI container")
start_response(status, headers)
return data