Show More
wsgi_app_caller_client.py
90 lines
| 3.0 KiB
| text/x-python
|
PythonLexer
r1 | # -*- coding: utf-8 -*- | |||
r2487 | # Copyright (C) 2012-2018 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/ | ||||
""" | ||||
Utility to call a WSGI app wrapped in a WSGIAppCaller object. | ||||
""" | ||||
import logging | ||||
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. | ||||
""" | ||||
Martin Bornhold
|
r951 | def __init__(self, remote_wsgi, *args, **kwargs): | ||
r1 | """ | |||
: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) | ||||
r1409 | data, status, headers = self._remote_wsgi.handle( | |||
clean_environ, input_data, *self._args, **self._kwargs) | ||||
r1 | ||||
log.debug("Got result from proxy, returning to WSGI container") | ||||
start_response(status, headers) | ||||
return data | ||||