Merge pull request #1666 from st1020/feat/add-request-type

feat: add request type

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:
+        ...

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
 

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

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")

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")

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,