github/jumpserver
1
0
Fork 0
mirror of https://github.com/jumpserver/jumpserver.git synced 2026-04-02 18:32:34 +00:00
Code Issues Projects Releases Wiki Activity

perf: 修改获取 ip

Browse Source
This commit is contained in:
wangruidong
2023-10-26 15:09:36 +08:00
committed by 老广
parent 29ab6bbee7
commit cc656a8a97
2 changed files with 30 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
apps/authentication/mixins.py
View File

@@ -18,7 +18,7 @@ from django.contrib.auth import (
from django.core.exceptions import ImproperlyConfigured
from django.shortcuts import reverse, redirect, get_object_or_404
from common.utils import get_request_ip, get_logger, bulk_get, FlashMessageUtil
from common.utils import get_request_ip_or_data, get_request_ip, get_logger, bulk_get, FlashMessageUtil
from acls.models import LoginACL
from users.models import User
from users.utils import LoginBlockUtil, MFABlockUtils, LoginIpBlockUtil
@@ -92,13 +92,12 @@ auth.authenticate = authenticate
class CommonMixin:
request: Request
_ip = ''
def get_request_ip(self):
ip = ''
if hasattr(self.request, 'data'):
ip = self.request.data.get('remote_addr', '')
ip = ip or get_request_ip(self.request)
return ip
if not self._ip:
self._ip = get_request_ip_or_data(self.request)
def raise_credential_error(self, error):
raise self.partial_credential_error(error=error)