52 lines
1.9 KiB
Python
52 lines
1.9 KiB
Python
|
# coding=utf-8
|
|||
|
|
"""
|
|||
|
|
@project: MaxKB
|
|||
|
|
@Author：虎虎
|
|||
|
|
@file： login.py
|
|||
|
|
@date：2025/4/14 11:08
|
|||
|
|
@desc:
|
|||
|
|
"""
|
|||
|
|
from django.core import signing
|
|||
|
|
from django.core.cache import cache
|
|||
|
|
from django.db.models import QuerySet
|
|||
|
|
from django.utils.translation import gettext_lazy as _
|
|||
|
|
from rest_framework import serializers
|
|||
|
|
|
|||
|
|
from common.constants.authentication_type import AuthenticationType
|
|||
|
|
from common.constants.cache_version import Cache_Version
|
|||
|
|
from common.exception.app_exception import AppApiException
|
|||
|
|
from common.utils.common import password_encrypt
|
|||
|
|
from users.models import User
|
|||
|
|
|
|||
|
|
|
|||
|
|
class LoginRequest(serializers.Serializer):
|
|||
|
|
username = serializers.CharField(required=True, max_length=64, help_text=_("Username"), label=_("Username"))
|
|||
|
|
password = serializers.CharField(required=True, max_length=128, label=_("Password"))
|
|||
|
|
|
|||
|
|
|
|||
|
|
class LoginResponse(serializers.Serializer):
|
|||
|
|
"""
|
|||
|
|
登录响应对象
|
|||
|
|
"""
|
|||
|
|
token = serializers.CharField(required=True, label=_("token"))
|
|||
|
|
|
|||
|
|
|
|||
|
|
class LoginSerializer(serializers.Serializer):
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
def login(instance):
|
|||
|
|
LoginRequest(data=instance).is_valid(raise_exception=True)
|
|||
|
|
username = instance.get('username')
|
|||
|
|
password = instance.get('password')
|
|||
|
|
user = QuerySet(User).filter(username=username, password=password_encrypt(password)).first()
|
|||
|
|
if user is None:
|
|||
|
|
raise AppApiException(500, _('The username or password is incorrect'))
|
|||
|
|
if not user.is_active:
|
|||
|
|
raise AppApiException(1005, _("The user has been disabled, please contact the administrator!"))
|
|||
|
|
token = signing.dumps({'username': user.username,
|
|||
|
|
'id': str(user.id),
|
|||
|
|
'email': user.email,
|
|||
|
|
'type': AuthenticationType.SYSTEM_USER.value})
|
|||
|
|
cache.set(token, user, version=Cache_Version.TOKEN)
|
|||
|
|
return {'token': token}
|