perf: 修改 csrf 登录时判断

This commit is contained in:
ibuler
2023-08-18 20:36:58 +08:00
parent 9dd2dc8907
commit c8a1f4b092
2 changed files with 7 additions and 4 deletions

View File

@@ -140,13 +140,16 @@ class UserLoginContextMixin:
if not self.request.GET.get('csrf_failure'):
return context
http_origin = self.request.META.get('HTTP_ORIGIN')
http_referer = self.request.META.get('HTTP_REFERER')
if not http_referer:
http_origin = http_origin or http_referer
if not http_origin:
return context
try:
referer = urlparse(http_referer)
context['error_origin'] = str(referer.netloc)
origin = urlparse(http_origin)
context['error_origin'] = str(origin.netloc)
except ValueError:
pass
return context