133 lines
3.3 KiB
Python
133 lines
3.3 KiB
Python
"""
|
|
request工具类
|
|
"""
|
|
import json
|
|
from apps.system.models import LoginLog
|
|
from django.urls.resolvers import ResolverMatch
|
|
from user_agents import parse
|
|
|
|
def get_request_ip(request):
|
|
"""
|
|
获取请求IP
|
|
:param request:
|
|
:return:
|
|
"""
|
|
x_forwarded_for = request.META.get('HTTP_X_FORWARDED_FOR', '')
|
|
if x_forwarded_for:
|
|
ip = x_forwarded_for.split(',')[-1].strip()
|
|
return ip
|
|
ip = request.META.get('REMOTE_ADDR', '') or getattr(request, 'request_ip', None)
|
|
return ip or 'unknown'
|
|
|
|
def get_request_data(request):
|
|
"""
|
|
获取请求参数
|
|
:param request:
|
|
:return:
|
|
"""
|
|
request_data = getattr(request, 'request_data', None)
|
|
if request_data:
|
|
return request_data
|
|
data: dict = {**request.GET.dict(), **request.POST.dict()}
|
|
if not data:
|
|
try:
|
|
body = request.body
|
|
if body:
|
|
data = json.loads(body)
|
|
except Exception as e:
|
|
pass
|
|
if not isinstance(data, dict):
|
|
data = {'data': data}
|
|
return data
|
|
|
|
def get_request_path(request, *args, **kwargs):
|
|
"""
|
|
获取请求路径
|
|
:param request:
|
|
:param args:
|
|
:param kwargs:
|
|
:return:
|
|
"""
|
|
request_path = getattr(request, 'request_path', None)
|
|
if request_path:
|
|
return request_path
|
|
values = []
|
|
for arg in args:
|
|
if len(arg) == 0:
|
|
continue
|
|
if isinstance(arg, str):
|
|
values.append(arg)
|
|
elif isinstance(arg, (tuple, set, list)):
|
|
values.extend(arg)
|
|
elif isinstance(arg, dict):
|
|
values.extend(arg.values())
|
|
if len(values) == 0:
|
|
return request.path
|
|
path: str = request.path
|
|
for value in values:
|
|
path = path.replace('/' + value, '/' + '{id}')
|
|
return path
|
|
|
|
def get_request_canonical_path(request, ):
|
|
"""
|
|
获取请求路径
|
|
:param request:
|
|
:param args:
|
|
:param kwargs:
|
|
:return:
|
|
"""
|
|
request_path = getattr(request, 'request_canonical_path', None)
|
|
if request_path:
|
|
return request_path
|
|
path: str = request.path
|
|
resolver_match: ResolverMatch = request.resolver_match
|
|
for value in resolver_match.args:
|
|
path = path.replace(f"/{value}", "/{id}")
|
|
for key, value in resolver_match.kwargs.items():
|
|
if key == 'pk':
|
|
path = path.replace(f"/{value}", f"/{{id}}")
|
|
continue
|
|
path = path.replace(f"/{value}", f"/{{{key}}}")
|
|
|
|
return path
|
|
|
|
def get_browser(request, ):
|
|
"""
|
|
获取浏览器名
|
|
:param request:
|
|
:param args:
|
|
:param kwargs:
|
|
:return:
|
|
"""
|
|
ua_string = request.META['HTTP_USER_AGENT']
|
|
user_agent = parse(ua_string)
|
|
return user_agent.get_browser()
|
|
|
|
def get_os(request, ):
|
|
"""
|
|
获取操作系统
|
|
:param request:
|
|
:param args:
|
|
:param kwargs:
|
|
:return:
|
|
"""
|
|
ua_string = request.META['HTTP_USER_AGENT']
|
|
user_agent = parse(ua_string)
|
|
return user_agent.get_os()
|
|
|
|
def save_login_log(request, user):
|
|
"""
|
|
保存登录日志
|
|
:return:
|
|
"""
|
|
ip = get_request_ip(request=request)
|
|
analysis_data = {
|
|
'username': user.username,
|
|
'ip': ip,
|
|
'agent': str(parse(request.META['HTTP_USER_AGENT'])),
|
|
'browser': get_browser(request),
|
|
'os': get_os(request),
|
|
'creator_id': user.id
|
|
}
|
|
LoginLog.objects.create(**analysis_data)
|