auth.py 1.3 KB

12345678910111213141516171819202122232425262728293031323334353637383940
  1. from django.conf import settings
  2. from django.contrib import auth
  3. from django.shortcuts import redirect
  4. from django.utils.http import is_safe_url
  5. from django.utils.six.moves.urllib.parse import urlparse
  6. from django.views.decorators.cache import never_cache
  7. from django.views.decorators.csrf import csrf_protect
  8. from django.views.decorators.debug import sensitive_post_parameters
  9. from misago.core.embercli import is_ember_cli_request, get_embercli_host
  10. @sensitive_post_parameters()
  11. @never_cache
  12. @csrf_protect
  13. def login(request):
  14. if request.method == 'POST':
  15. redirect_to = request.POST.get('redirect_to')
  16. if redirect_to:
  17. is_redirect_safe = is_safe_url(
  18. url=redirect_to, host=request.get_host())
  19. if not is_redirect_safe and is_ember_cli_request(request):
  20. is_redirect_safe = is_safe_url(
  21. url=redirect_to, host=get_embercli_host())
  22. if is_redirect_safe:
  23. redirect_to_path = urlparse(redirect_to).path
  24. return redirect(redirect_to_path)
  25. return redirect(settings.LOGIN_REDIRECT_URL)
  26. @never_cache
  27. @csrf_protect
  28. def logout(request):
  29. if request.method == 'POST' and request.user.is_authenticated():
  30. auth.logout(request)
  31. return redirect(settings.LOGIN_REDIRECT_URL)