76 lines
2.4 KiB
Python
76 lines
2.4 KiB
Python
from functools import wraps
|
|
|
|
from django.conf import settings
|
|
from django.http import HttpResponseForbidden
|
|
from django.shortcuts import get_object_or_404
|
|
from django.urls import reverse_lazy
|
|
|
|
from .models import Map
|
|
from .views import simple_json_response
|
|
|
|
LOGIN_URL = getattr(settings, "LOGIN_URL", "login")
|
|
LOGIN_URL = reverse_lazy(LOGIN_URL) if not LOGIN_URL.startswith("/") else LOGIN_URL
|
|
|
|
|
|
def login_required_if_not_anonymous_allowed(view_func):
|
|
@wraps(view_func)
|
|
def wrapper(request, *args, **kwargs):
|
|
if (
|
|
not getattr(settings, "UMAP_ALLOW_ANONYMOUS", False)
|
|
and not request.user.is_authenticated
|
|
):
|
|
return simple_json_response(login_required=str(LOGIN_URL))
|
|
return view_func(request, *args, **kwargs)
|
|
|
|
return wrapper
|
|
|
|
|
|
def can_edit_map(view_func):
|
|
"""
|
|
Used for URLs dealing with editing the map.
|
|
"""
|
|
|
|
@wraps(view_func)
|
|
def wrapper(request, *args, **kwargs):
|
|
map_inst = get_object_or_404(Map, pk=kwargs["map_id"])
|
|
user = request.user
|
|
kwargs["map_inst"] = map_inst # Avoid rerequesting the map in the view
|
|
if map_inst.edit_status >= map_inst.EDITORS:
|
|
can_edit = map_inst.can_edit(user=user, request=request)
|
|
if not can_edit:
|
|
if map_inst.owner and not user.is_authenticated:
|
|
return simple_json_response(login_required=str(LOGIN_URL))
|
|
return HttpResponseForbidden()
|
|
return view_func(request, *args, **kwargs)
|
|
|
|
return wrapper
|
|
|
|
|
|
def can_view_map(view_func):
|
|
"""
|
|
Used for URLs dealing with viewing the map.
|
|
"""
|
|
|
|
@wraps(view_func)
|
|
def wrapper(request, *args, **kwargs):
|
|
map_inst = get_object_or_404(Map, pk=kwargs["map_id"])
|
|
kwargs["map_inst"] = map_inst # Avoid rerequesting the map in the view
|
|
if not map_inst.can_view(request):
|
|
return HttpResponseForbidden()
|
|
return view_func(request, *args, **kwargs)
|
|
|
|
return wrapper
|
|
|
|
|
|
def jsonize_view(view_func):
|
|
@wraps(view_func)
|
|
def wrapper(request, *args, **kwargs):
|
|
response = view_func(request, *args, **kwargs)
|
|
response_kwargs = {}
|
|
if hasattr(response, "rendered_content"):
|
|
response_kwargs["html"] = response.rendered_content
|
|
if response.has_header("location"):
|
|
response_kwargs["redirect"] = response["location"]
|
|
return simple_json_response(**response_kwargs)
|
|
|
|
return wrapper
|