1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
|
import functools
from asgiref.sync import iscoroutinefunction
from django.http import Http404
from django.utils.translation import get_language, override as language_override
from debug_toolbar import settings as dt_settings
def require_show_toolbar(view):
"""
Async compatible decorator to restrict access to a view
based on the Debug Toolbar's visibility settings.
"""
from debug_toolbar.middleware import get_show_toolbar
if iscoroutinefunction(view):
@functools.wraps(view)
async def inner(request, *args, **kwargs):
show_toolbar = get_show_toolbar(async_mode=True)
if not await show_toolbar(request):
raise Http404
return await view(request, *args, **kwargs)
else:
@functools.wraps(view)
def inner(request, *args, **kwargs):
show_toolbar = get_show_toolbar(async_mode=False)
if not show_toolbar(request):
raise Http404
return view(request, *args, **kwargs)
return inner
def render_with_toolbar_language(view):
"""Force any rendering within the view to use the toolbar's language."""
@functools.wraps(view)
def inner(request, *args, **kwargs):
lang = dt_settings.get_config()["TOOLBAR_LANGUAGE"] or get_language()
with language_override(lang):
return view(request, *args, **kwargs)
return inner
|
