diff --git a/apps/chat/serializers/chat_authentication.py b/apps/chat/serializers/chat_authentication.py index 9777607f6e6..1a37e98cd23 100644 --- a/apps/chat/serializers/chat_authentication.py +++ b/apps/chat/serializers/chat_authentication.py @@ -7,13 +7,13 @@ @desc: """ import uuid_utils.compat as uuid -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 application.models import ApplicationAccessToken, ChatUserType, Application, ApplicationVersion +from common import signing from common.auth.common import ChatUserToken, ChatAuthentication, FileToken from common.constants.authentication_type import AuthenticationType from common.constants.cache_version import Cache_Version diff --git a/apps/common/auth/authenticate.py b/apps/common/auth/authenticate.py index 71ee51bf1f6..db09e009c00 100644 --- a/apps/common/auth/authenticate.py +++ b/apps/common/auth/authenticate.py @@ -10,11 +10,11 @@ from django.conf import settings from django.core import cache -from django.core import signing from django.utils.translation import gettext_lazy as _ from drf_spectacular.extensions import OpenApiAuthenticationExtension from rest_framework.authentication import TokenAuthentication +from common import signing from common.exception.app_exception import AppAuthenticationFailed, AppEmbedIdentityFailed, AppChatNumOutOfBoundsFailed, \ AppApiException from common.utils.logger import maxkb_logger diff --git a/apps/common/auth/common.py b/apps/common/auth/common.py index c64ceeb48a1..7e1b05a63f5 100644 --- a/apps/common/auth/common.py +++ b/apps/common/auth/common.py @@ -10,8 +10,9 @@ import json import threading -from django.core import signing, cache +from django.core import cache +from common import signing from common.constants.cache_version import Cache_Version from common.utils.rsa_util import encrypt, decrypt diff --git a/apps/common/signing.py b/apps/common/signing.py new file mode 100644 index 00000000000..c49ec051027 --- /dev/null +++ b/apps/common/signing.py @@ -0,0 +1,20 @@ +# coding=utf-8 +""" + @project: MaxKB + @Author: 虎虎虎 + @file: rsa_util.py + @date: 2026/09/21 13:57 + @desc: +""" +from common.utils.rsa_util import get_key_pair +from django.core import signing + + +def loads(s: str): + value = get_key_pair().get('value') + return signing.loads(s, key=value) + + +def dumps(obj): + value = get_key_pair().get('value') + return signing.dumps(obj, key=value) diff --git a/apps/users/serializers/login.py b/apps/users/serializers/login.py index 12210f8bad9..6c82e35427a 100644 --- a/apps/users/serializers/login.py +++ b/apps/users/serializers/login.py @@ -11,12 +11,12 @@ import json from captcha.image import ImageCaptcha -from django.core import signing from django.core.cache import cache from django.utils.translation import gettext_lazy as _ from rest_framework import serializers from application.models import ApplicationAccessToken +from common import signing from common.auth.common import FileToken from common.constants.authentication_type import AuthenticationType from common.constants.cache_version import Cache_Version