|
|
|
# -*- coding: utf-8 -*-
|
|
|
|
#
|
|
|
|
import ipaddress
|
|
|
|
from urllib.parse import urljoin, urlparse
|
|
|
|
|
|
|
|
from django.conf import settings
|
|
|
|
from django.utils.translation import gettext_lazy as _
|
|
|
|
|
|
|
|
from audits.const import DEFAULT_CITY
|
|
|
|
from users.models import User
|
|
|
|
from audits.models import UserLoginLog
|
|
|
|
from common.utils import get_logger, get_object_or_none
|
|
|
|
from common.utils import validate_ip, get_ip_city, get_request_ip
|
|
|
|
from .notifications import DifferentCityLoginMessage
|
|
|
|
|
|
|
|
logger = get_logger(__file__)
|
|
|
|
|
|
|
|
|
|
|
|
def check_different_city_login_if_need(user, request):
|
|
|
|
if not settings.SECURITY_CHECK_DIFFERENT_CITY_LOGIN:
|
|
|
|
return
|
|
|
|
|
|
|
|
ip = get_request_ip(request) or '0.0.0.0'
|
|
|
|
city_white = [_('LAN'), 'LAN']
|
|
|
|
is_private = ipaddress.ip_address(ip).is_private
|
|
|
|
if is_private:
|
|
|
|
return
|
|
|
|
last_user_login = UserLoginLog.objects.exclude(
|
|
|
|
city__in=city_white
|
|
|
|
).filter(username=user.username, status=True).first()
|
|
|
|
if not last_user_login:
|
|
|
|
return
|
|
|
|
|
|
|
|
city = get_ip_city(ip)
|
|
|
|
last_city = get_ip_city(last_user_login.ip)
|
|
|
|
if city == last_city:
|
|
|
|
return
|
|
|
|
|
|
|
|
DifferentCityLoginMessage(user, ip, city).publish_async()
|
|
|
|
|
|
|
|
|
|
|
|
def build_absolute_uri(request, path=None):
|
|
|
|
""" Build absolute redirect """
|
|
|
|
if path is None:
|
|
|
|
path = '/'
|
|
|
|
site_url = urlparse(settings.SITE_URL)
|
|
|
|
scheme = site_url.scheme or request.scheme
|
|
|
|
host = request.get_host()
|
|
|
|
url = f'{scheme}://{host}'
|
|
|
|
redirect_uri = urljoin(url, path)
|
|
|
|
return redirect_uri
|
|
|
|
|
|
|
|
|
|
|
|
def build_absolute_uri_for_oidc(request, path=None):
|
|
|
|
""" Build absolute redirect uri for OIDC """
|
|
|
|
if path is None:
|
|
|
|
path = '/'
|
|
|
|
if settings.BASE_SITE_URL:
|
|
|
|
# OIDC 专用配置项
|
|
|
|
redirect_uri = urljoin(settings.BASE_SITE_URL, path)
|
|
|
|
return redirect_uri
|
|
|
|
return build_absolute_uri(request, path=path)
|
|
|
|
|
|
|
|
|
|
|
|
def check_user_property_is_correct(username, **properties):
|
|
|
|
user = get_object_or_none(User, username=username)
|
|
|
|
for attr, value in properties.items():
|
|
|
|
if getattr(user, attr, None) != value:
|
|
|
|
user = None
|
|
|
|
break
|
|
|
|
return user
|