Commit 4371d5c8 authored by Bryton Lacquement's avatar Bryton Lacquement 🚪

Localizer: update the 'Global Request' monkey-patch to support WSGIPublisher

parent 59212692
......@@ -28,7 +28,7 @@ from .itools.i18n import AcceptLanguageType
# Import from Zope
import Globals
from ZPublisher import Publish
from ZPublisher import Publish, WSGIPublisher
from ZPublisher.HTTPRequest import HTTPRequest
......@@ -65,23 +65,19 @@ def get_request():
"""Get a request object"""
return _requests.get(get_ident(), None)
def new_publish(request, module_name, after_list, debug=0,
zope_publish=Publish.publish):
def get_new_publish(zope_publish):
def publish(request, *args, **kwargs):
# Get the process id
ident = get_ident()
# Add the request object to the global dictionnary
_requests_lock.acquire()
try:
with _requests_lock:
_requests[ident] = request
finally:
_requests_lock.release()
# Call the old publish
try:
# Publish
x = zope_publish(request, module_name, after_list, debug)
return zope_publish(request, *args, **kwargs)
finally:
# Remove the request object.
# When conflicts occur the "publish" method is called again,
......@@ -90,20 +86,23 @@ def new_publish(request, module_name, after_list, debug=0,
# request does not exist anymore. For this reason we check first
# wether the request is there or not.
if ident in _requests:
_requests_lock.acquire()
try:
with _requests_lock:
del _requests[ident]
finally:
_requests_lock.release()
return x
return publish
if patch is False:
logger.info('Install "Globals.get_request".')
# Apply the patch
Publish.publish = new_publish
Publish.publish = get_new_publish(Publish.publish)
WSGIPublisher.publish = get_new_publish(WSGIPublisher.publish)
# Update WSGIPublisher.publish_module.__defaults__, otherwise it will use
# the unpatched WSGIPublisher.publish.
WSGIPublisher.publish_module.__defaults__ = (
WSGIPublisher.publish,
) + WSGIPublisher.publish_module.__defaults__[1:]
# First import (it's not a refresh operation).
# We need to apply the patches.
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment