mirror of
https://github.com/1Panel-dev/MaxKB.git
synced 2025-12-26 01:33:05 +00:00
37 lines
1.4 KiB
Python
37 lines
1.4 KiB
Python
# coding=utf-8
|
||
"""
|
||
@project: maxkb
|
||
@Author：虎虎
|
||
@file： authenticate.py
|
||
@date：2024/3/14 03:02
|
||
@desc: 用户认证
|
||
"""
|
||
from django.db.models import QuerySet
|
||
from common.auth.handle.auth_base_handle import AuthBaseHandle
|
||
from common.constants.authentication_type import AuthenticationType
|
||
from common.constants.cache_version import Cache_Version
|
||
from common.constants.permission_constants import Auth, RoleConstants
|
||
from common.exception.app_exception import AppAuthenticationFailed
|
||
from users.models import User
|
||
from django.core.cache import cache
|
||
from django.utils.translation import gettext_lazy as _
|
||
|
||
|
||
class UserToken(AuthBaseHandle):
|
||
def support(self, request, token: str, get_token_details):
|
||
auth_details = get_token_details()
|
||
if auth_details is None:
|
||
return False
|
||
return True
|
||
|
||
def handle(self, request, token: str, get_token_details):
|
||
cache_token = cache.get(token, version=Cache_Version.TOKEN)
|
||
if cache_token is None:
|
||
raise AppAuthenticationFailed(1002, _('Login expired'))
|
||
auth_details = get_token_details()
|
||
user = QuerySet(User).get(id=auth_details['id'])
|
||
role = RoleConstants[user.role]
|
||
return user, Auth([], [],
|
||
client_id=str(user.id),
|
||
client_type=AuthenticationType.SYSTEM_USER.value, current_role=role)
|