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

bk_resource/contrib/model.py

@@ -26,13 +26,10 @@ from rest_framework.generics import GenericAPIView
 from rest_framework.viewsets import ModelViewSet
 
 from bk_resource import Resource
+from bk_resource.base import Empty
 from bk_resource.utils.request import get_mock_request
 
 
-class Empty:
-    ...
-
-
 class ViewMixin(GenericAPIView):
     filter_fields = []
     search_fields = []

bk_resource/utils/cache.py

@@ -25,7 +25,7 @@ from django.core.cache import cache, caches
 from django.utils.encoding import force_bytes
 from django.utils.translation import gettext
 
-from bk_resource.contrib.model import Empty
+from bk_resource.base import Empty
 from bk_resource.settings import bk_resource_settings
 from bk_resource.utils.common_utils import count_md5
 from bk_resource.utils.local import local

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.12b0.dist-info → bk_resource-0.4.13b0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: bk_resource
-Version: 0.4.12b0
+Version: 0.4.13b0
 Summary: Bk Resource
 Home-page: https://bk.tencent.com
 Author: blueking
@@ -14,12 +14,12 @@ Requires-Dist: drf-yasg >=1.20.0
 Requires-Dist: pyinstrument >=3.4.2
 Requires-Dist: arrow >=1.2.0
 Requires-Dist: django-rest-framework-condition >=0.1.1
-Requires-Dist: celery >=5
+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.12b0-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.12b0.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
@@ -22,7 +22,7 @@ bk_resource/contrib/__init__.py,sha256=bP18WkBTSVynqrmGO6UDjhTdOCO0w18myXynyHMg6
 bk_resource/contrib/api.py,sha256=DFcjJsuwU9pKwvQE-nZhN2wApqkaC0feBIh4Es5mUxc,8658
 bk_resource/contrib/bk_api.py,sha256=HJW5GWM0tT-oR6AnYHjcU9JaymyKPrrx_0aCnZD1BaM,6304
 bk_resource/contrib/cache.py,sha256=Gp91ajqDUelXKDu2WTC_QXoTJiO--iWEMsNjl_hxAlg,2969
-bk_resource/contrib/model.py,sha256=ashkD_NzIR4WSEEQHPWPCp66wvKWwiaL1AHfQgWfyAY,5941
+bk_resource/contrib/model.py,sha256=-zQOjnXllDYDMo6nxcvD8WrfzQuThfzcqf97gnRhwaE,5953
 bk_resource/locale/en/LC_MESSAGES/django.mo,sha256=2Pf2WZXzYQeGcnnNWo78xZ58vuOr7Tj5sc9v675SwSk,2943
 bk_resource/locale/en/LC_MESSAGES/django.po,sha256=VIssUCnLPT1BUKjTB93yR7uyc0IvHdgGS93wpV18cCg,4322
 bk_resource/locale/zh_CN/LC_MESSAGES/django.mo,sha256=3GCXKryGcXqFnoIDIDaz97SBpl5C99nbSzjSkM6bBkg,3022
@@ -36,20 +36,20 @@ bk_resource/management/commands/__init__.py,sha256=zTvK51uqIT4Sy_bCfeXQbIyI-Sskb
 bk_resource/management/commands/generate_resource_stub_file.py,sha256=M_S1eLxx-1eeF3VwbzJJl0DE2ABB3Z-i75Kb23nyi38,1110
 bk_resource/management/commands/start_resource.py,sha256=BHohJIXIsT2Ur0GbK1FctrEdY2RbJtIGs2myyRsOb8M,1243
 bk_resource/utils/__init__.py,sha256=zTvK51uqIT4Sy_bCfeXQbIyI-SskbMWjUfVNrQzE-0k,891
-bk_resource/utils/cache.py,sha256=g_d8iD9AWX_kOe_zVSaUVSJG9hgQclp_-gjtqDc5hME,10609
+bk_resource/utils/cache.py,sha256=QNaXdEaAJDPhMvJmNdMB3vVA34bs4cPdkQJ5SZMA0Xk,10600
 bk_resource/utils/common_utils.py,sha256=uJ2O_64At_LhPexC1ZifMFNauG9oh73v9EdnUB3prUo,13205
 bk_resource/utils/factory.py,sha256=FtGnGqhi_ubUOxveEkfgrlpYFE2-m-oPFmSV9347UD8,5104
 bk_resource/utils/generators.py,sha256=qhs0msHJdmVm8z6phfkC-GjibhxJckq6xgM4IM8w5bU,2080
 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.12b0.dist-info/LICENSE.txt,sha256=dnlkOMrlbRgRXlBt-a_QF6cs1MFG6WYmcotYbI6_gpo,859
-bk_resource-0.4.12b0.dist-info/METADATA,sha256=w11I_AfW-LozH5LyRbyaVoDZnLNQS3pmVSB_ndtX4Fo,5501
-bk_resource-0.4.12b0.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-bk_resource-0.4.12b0.dist-info/top_level.txt,sha256=UJxO1tbrqlmnvSTYCLoqg39_gvBesvjp3CcGEPjzXY4,12
-bk_resource-0.4.12b0.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,,

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

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.43.0)
+Generator: setuptools (70.2.0)
 Root-Is-Purelib: true
 Tag: py3-none-any