diff --git a/dongtai_common/utils/request_type.py b/dongtai_common/utils/request_type.py new file mode 100644 index 000000000..a90a19535 --- /dev/null +++ b/dongtai_common/utils/request_type.py @@ -0,0 +1,15 @@ +from rest_framework.request import Request as DRFRequest + +from dongtai_common.models.user import User + + +class Request(DRFRequest): + """used in type check only.""" + + @property + def user(self) -> User: + ... + + @user.setter + def user(self, value) -> None: + ... diff --git a/dongtai_protocol/views/agent_register.py b/dongtai_protocol/views/agent_register.py index 6a2a77912..836c9a6f9 100644 --- a/dongtai_protocol/views/agent_register.py +++ b/dongtai_protocol/views/agent_register.py @@ -8,7 +8,6 @@ from django.db import transaction from django.utils.translation import gettext_lazy as _ from drf_spectacular.utils import extend_schema -from rest_framework.request import Request from dongtai_common.endpoint import OpenApiEndPoint, R from dongtai_common.models.agent import IastAgent @@ -18,6 +17,7 @@ ) from dongtai_common.models.project_version import IastProjectVersion from dongtai_common.models.server import IastServer +from dongtai_common.utils.request_type import Request from dongtai_protocol.api_schema import DongTaiParameter from dongtai_protocol.decrypter import parse_data diff --git a/dongtai_protocol/views/engine_download.py b/dongtai_protocol/views/engine_download.py index 16d80bbaf..617103f3c 100644 --- a/dongtai_protocol/views/engine_download.py +++ b/dongtai_protocol/views/engine_download.py @@ -7,9 +7,9 @@ from django.http import FileResponse from drf_spectacular.utils import extend_schema from rest_framework import status -from rest_framework.request import Request from dongtai_common.endpoint import OpenApiEndPoint, R +from dongtai_common.utils.request_type import Request from dongtai_conf.settings import BUCKET_NAME_BASE_URL, VERSION from dongtai_protocol.api_schema import DongTaiParameter from dongtai_protocol.utils import OssDownloader diff --git a/dongtai_protocol/views/properties.py b/dongtai_protocol/views/properties.py index db14650f4..dcc21d924 100644 --- a/dongtai_protocol/views/properties.py +++ b/dongtai_protocol/views/properties.py @@ -4,11 +4,11 @@ from django.utils.translation import gettext_lazy as _ from drf_spectacular.utils import extend_schema -from rest_framework.request import Request from dongtai_common.endpoint import OpenApiEndPoint, R from dongtai_common.models.agent import IastAgent from dongtai_common.models.agent_properties import IastAgentProperties +from dongtai_common.utils.request_type import Request from dongtai_protocol.serializers.agent_properties import AgentPropertiesSerialize logger = logging.getLogger("django") diff --git a/dongtai_protocol/views/startuptime.py b/dongtai_protocol/views/startuptime.py index 8e1d37071..d84c67dd5 100644 --- a/dongtai_protocol/views/startuptime.py +++ b/dongtai_protocol/views/startuptime.py @@ -10,10 +10,10 @@ import logging from drf_spectacular.utils import extend_schema -from rest_framework.request import Request from dongtai_common.endpoint import OpenApiEndPoint, R from dongtai_common.models.agent import IastAgent +from dongtai_common.utils.request_type import Request from dongtai_protocol.decrypter import parse_data logger = logging.getLogger("django") diff --git a/dongtai_web/systemmonitor/project_warning.py b/dongtai_web/systemmonitor/project_warning.py index a0ecb88d5..e5b823042 100644 --- a/dongtai_web/systemmonitor/project_warning.py +++ b/dongtai_web/systemmonitor/project_warning.py @@ -4,10 +4,10 @@ from django.http import JsonResponse from django.utils.translation import gettext_lazy as _ from rest_framework import serializers -from rest_framework.request import Request from dongtai_common.endpoint import R, UserEndPoint from dongtai_common.models.profile import IastProfile +from dongtai_common.utils.request_type import Request from dongtai_engine.plugins.project_status import ( PROJECT_WARNING_TIME_KEY, get_project_warning_time,