bk-resource 0.4.12__py3-none-any.whl → 0.4.13b0__py3-none-any.whl

bk_resource/utils/request.py

@@ -15,8 +15,11 @@ specific language governing permissions and limitations under the License.
 We undertake not to change the open source license (MIT license) applicable
 to the current version of the project delivered to anyone in the future.
 """
+from typing import Optional, Type
 
+from django.http import HttpRequest
 from django.test import RequestFactory
+from django.urls import Resolver404, resolve
 
 from bk_resource.utils.local import local
 
@@ -44,3 +47,32 @@ def set_local_username(username):
 
 def get_mock_request(**kwargs):
     return RequestFactory().request(**kwargs)
+
+
+def get_resource_by_request(request: HttpRequest = None) -> Optional[Type["bk_resource.Resource"]]:
+    """根据 request 对象获取具体的 Resource 类"""
+
+    from blueapps.utils.request_provider import get_local_request
+
+    request = request or get_local_request()
+    if not request:
+        return None
+
+    try:
+        match = resolve(request.path_info)
+    except Resolver404:
+        try:
+            match = resolve(f"{request.path_info}/")
+        except Resolver404:
+            return None
+
+    try:
+        from bk_resource.viewsets import RESOURCE_MAPPING
+
+        view_set_path = f"{match.func.cls.__module__}.{match.func.cls.__name__}"
+        resource_clz = RESOURCE_MAPPING.get(
+            (request.method, f"{view_set_path}.{match.func.actions.get(request.method.lower())}")
+        )
+        return resource_clz
+    except Exception:  # pylint: disable=broad-except
+        return None

bk_resource/viewsets.py

@@ -35,6 +35,9 @@ from bk_resource.base import Resource
 from bk_resource.settings import bk_resource_settings
 
 
+RESOURCE_MAPPING = {}
+
+
 class ResourceRoute(object):
     """
     Resource的视图配置，应用于viewsets
@@ -124,6 +127,7 @@ class ResourceViewSet(viewsets.GenericViewSet):
 
     @classmethod
     def generate_endpoint(cls):
+
         for resource_route in cls.resource_routes:
             # 生成方法模版
             function = cls._generate_function_template(resource_route)
@@ -202,6 +206,9 @@ class ResourceViewSet(viewsets.GenericViewSet):
                     setattr(cls, cls.EMPTY_ENDPOINT_METHODS[resource_route.method], function)
                 else:
                     raise AssertionError(gettext("不支持的请求方法: %s，请确认resource_routes配置是否正确!") % resource_route.method)
+
+                resource_action = cls.EMPTY_ENDPOINT_METHODS[resource_route.method]
+
             else:
                 function = method_decorator(cache_control(max_age=0, private=True))(function)
                 function.__name__ = resource_route.endpoint
@@ -217,6 +224,12 @@ class ResourceViewSet(viewsets.GenericViewSet):
                 function = decorator_function(function)
                 setattr(cls, resource_route.endpoint, function)
 
+                resource_action = resource_route.endpoint
+
+            RESOURCE_MAPPING[
+                (resource_route.method, f"{cls.__module__}.{cls.__name__}.{resource_action}")
+            ] = resource_route.resource_class
+
     @classmethod
     def _generate_function_template(cls, resource_route: ResourceRoute):
         """

{bk_resource-0.4.12.dist-info → bk_resource-0.4.13b0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: bk_resource
-Version: 0.4.12
+Version: 0.4.13b0
 Summary: Bk Resource
 Home-page: https://bk.tencent.com
 Author: blueking
@@ -19,7 +19,7 @@ Requires-Dist: celery >=4.4.0
 ![logo.png](assests/logo.png)
 
 [![License](https://img.shields.io/badge/license-MIT-brightgreen.svg?style=flat)](https://github.com/TencentBlueKing/bk-resource/blob/main/LICENSE.txt)
-[![Release Version](https://img.shields.io/badge/release-0.4.12-brightgreen.svg)](https://github.com/TencentBlueKing/bk-resource/releases)
+[![Release Version](https://img.shields.io/badge/release-0.4.13b0-brightgreen.svg)](https://github.com/TencentBlueKing/bk-resource/releases)
 [![PRs Welcome](https://img.shields.io/badge/PRs-welcome-brightgreen.svg)](https://github.com/TencentBlueKing/bk-resource/pulls)
 [![codecov](https://codecov.io/gh/TencentBlueKing/bk-resource/branch/main/graph/badge.svg)](https://codecov.io/gh/TencentBlueKing/bk-resource)
 [![Unittest Py3](https://github.com/TencentBlueKing/bk-resource/actions/workflows/unittest.yml/badge.svg)](https://github.com/TencentBlueKing/bk-resource/actions/workflows/unittest.yml)

{bk_resource-0.4.12.dist-info → bk_resource-0.4.13b0.dist-info}/RECORD

@@ -7,7 +7,7 @@ bk_resource/serializers.py,sha256=Az_2Qsp_5gUrhU_0iZS45komm5upU5u1UpNE-ArSX4s,26
 bk_resource/settings.py,sha256=QCGKeetjdDIWHHu1_93MKPeJ_VpGlQXvW7gpln1pGlU,3827
 bk_resource/tasks.py,sha256=OeBy41GfSDNL_0adGeKBG6LlVRqNUfP7KzXe7pxpr3I,4090
 bk_resource/tools.py,sha256=fPmiDNxh2FWo5Yo7CXtvYyHxyG3_JrGs5wYZMaVFAhE,8166
-bk_resource/viewsets.py,sha256=QhCIHiJ227N8njqr6bSJZxP5eBh3z5Q4u1dGzBjHHXs,11614
+bk_resource/viewsets.py,sha256=KWx2OaPGTZVFCTJTN4qCOM5SGz6UZ0hVOdWC_-6Vsbg,11955
 bk_resource/conf/__init__.py,sha256=39m-gTk6pr0j_6xwdfTFEFuj_ER_MDoEll2zkEdCPBo,891
 bk_resource/conf/app_template/__init__.py-tpl,sha256=iwhKnzeBJLKxpRVjvzwiRE63_zNpIBfaKLITauVph-0,24
 bk_resource/conf/app_template/admin.py-tpl,sha256=n0NQrWDbx74lHja8g2s5vTKjR954SCtizSehZCQLBK0,60
@@ -43,13 +43,13 @@ bk_resource/utils/generators.py,sha256=qhs0msHJdmVm8z6phfkC-GjibhxJckq6xgM4IM8w5
 bk_resource/utils/inspectors.py,sha256=xxuB2iHInCGjl9A4QRz9gCw-iD5GFkXlhp_nCsnnJPw,1176
 bk_resource/utils/local.py,sha256=qPxDc5lYA_7ffmpUfGqYVsUG-WxBQ0jGCvc6Ti0WoCw,3337
 bk_resource/utils/logger.py,sha256=VPp5nGsKawMtH5pKctLmdDpZpfYtjCr3DUYHhEsuXBU,950
-bk_resource/utils/request.py,sha256=5Lfv_LCvE0I3yp3-jxGve8PU8TF4rR0vGTNBYGxzvIA,1593
+bk_resource/utils/request.py,sha256=Ipah61TzZF1ArjW5nj6ThaG1hsl2Swj7kwvbJBGtsF4,2623
 bk_resource/utils/request_log.py,sha256=VMSFu0c8e53e7TUFJtdKXpkM0_pl-mk1UaFFdrGYzPo,3317
 bk_resource/utils/text.py,sha256=exu7gEFANcU6apQOut_1VSzoddOvcKR3CXEF1Z19M4Y,3313
 bk_resource/utils/thread_backend.py,sha256=AdMWzXreFARinPdDaMiRJwwxSWL1EB4MUSz1Q7kYz90,5017
 bk_resource/utils/time_tools.py,sha256=y-eP-6Gu5ur3ET0bOp2ML8Z0HtrguvTPlbsSA1dNhWU,10967
-bk_resource-0.4.12.dist-info/LICENSE.txt,sha256=dnlkOMrlbRgRXlBt-a_QF6cs1MFG6WYmcotYbI6_gpo,859
-bk_resource-0.4.12.dist-info/METADATA,sha256=wKkceKDIj65xMUI_yY8HkXdgFydRkG1mG35rsqNYS04,5501
-bk_resource-0.4.12.dist-info/WHEEL,sha256=y4mX-SOX4fYIkonsAGA5N0Oy-8_gI4FXw5HNI1xqvWg,91
-bk_resource-0.4.12.dist-info/top_level.txt,sha256=UJxO1tbrqlmnvSTYCLoqg39_gvBesvjp3CcGEPjzXY4,12
-bk_resource-0.4.12.dist-info/RECORD,,
+bk_resource-0.4.13b0.dist-info/LICENSE.txt,sha256=dnlkOMrlbRgRXlBt-a_QF6cs1MFG6WYmcotYbI6_gpo,859
+bk_resource-0.4.13b0.dist-info/METADATA,sha256=mKLna1KHSTKdn_-3LmosYThky7alFkVurgGvF7QEgZY,5505
+bk_resource-0.4.13b0.dist-info/WHEEL,sha256=y4mX-SOX4fYIkonsAGA5N0Oy-8_gI4FXw5HNI1xqvWg,91
+bk_resource-0.4.13b0.dist-info/top_level.txt,sha256=UJxO1tbrqlmnvSTYCLoqg39_gvBesvjp3CcGEPjzXY4,12
+bk_resource-0.4.13b0.dist-info/RECORD,,