##// END OF EJS Templates
fix(core): don't rely on pkgutil to extract the version info of project
fix(core): don't rely on pkgutil to extract the version info of project

File last commit:

r1152:a0c49580 default
r1188:dc28b397 default
Show More
wsgi_app_caller.py
116 lines | 3.7 KiB | text/x-python | PythonLexer
initial commit
r0 # RhodeCode VCSServer provides access to different vcs backends via network.
source-code: updated copyrights to 2023
r1126 # Copyright (C) 2014-2023 RhodeCode GmbH
initial commit
r0 #
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 3 of the License, or
# (at your option) any later version.
#
# 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 General Public License
# along with this program; if not, write to the Free Software Foundation,
# Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
"""Extract the responses of a WSGI app."""
__all__ = ('WSGIAppCaller',)
import io
import logging
import os
packages: move the str utils to it's own module
r1060 from vcsserver.str_utils import ascii_bytes
initial commit
r0
log = logging.getLogger(__name__)
DEV_NULL = open(os.devnull)
python3: code change for py3 support...
r1048 def _complete_environ(environ, input_data: bytes):
initial commit
r0 """Update the missing wsgi.* variables of a WSGI environment.
:param environ: WSGI environment to update
:type environ: dict
:param input_data: data to be read by the app
python3: code change for py3 support...
r1048 :type input_data: bytes
initial commit
r0 """
environ.update({
'wsgi.version': (1, 0),
'wsgi.url_scheme': 'http',
'wsgi.multithread': True,
'wsgi.multiprocess': True,
'wsgi.run_once': False,
'wsgi.input': io.BytesIO(input_data),
'wsgi.errors': DEV_NULL,
})
# pylint: disable=too-few-public-methods
lint: auto-fixes
r1152 class _StartResponse:
initial commit
r0 """Save the arguments of a start_response call."""
__slots__ = ['status', 'headers', 'content']
def __init__(self):
self.status = None
self.headers = None
self.content = []
def __call__(self, status, headers, exc_info=None):
# TODO(skreft): do something meaningful with the exc_info
exc_info = None # avoid dangling circular reference
self.status = status
self.headers = headers
return self.write
def write(self, content):
"""Write method returning when calling this object.
All the data written is then available in content.
"""
self.content.append(content)
lint: auto-fixes
r1152 class WSGIAppCaller:
initial commit
r0 """Calls a WSGI app."""
def __init__(self, app):
"""
:param app: WSGI app to call
"""
self.app = app
def handle(self, environ, input_data):
"""Process a request with the WSGI app.
The returned data of the app is fully consumed into a list.
:param environ: WSGI environment to update
:type environ: dict
:param input_data: data to be read by the app
python3: code change for py3 support...
r1048 :type input_data: str/bytes
initial commit
r0
:returns: a tuple with the contents, status and headers
:rtype: (list<str>, str, list<(str, str)>)
"""
python3: code change for py3 support...
r1048 _complete_environ(environ, ascii_bytes(input_data, allow_bytes=True))
initial commit
r0 start_response = _StartResponse()
log.debug("Calling wrapped WSGI application")
responses = self.app(environ, start_response)
responses_list = list(responses)
existing_responses = start_response.content
if existing_responses:
python3: code change for py3 support...
r1048 log.debug("Adding returned response to response written via write()")
initial commit
r0 existing_responses.extend(responses_list)
responses_list = existing_responses
if hasattr(responses, 'close'):
log.debug("Closing iterator from WSGI application")
responses.close()
log.debug("Handling of WSGI request done, returning response")
return responses_list, start_response.status, start_response.headers