From f62344509f55e7e6686b1f24d882112e238e23c7 Mon Sep 17 00:00:00 2001
From: caoqianming <caoqianming@foxmail.com>
Date: Fri, 30 Jun 2023 16:15:04 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20bi=20exec=20=E6=94=AF=E6=8C=81code?=
 =?UTF-8?q?=E6=89=A7=E8=A1=8C?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/bi/views.py | 35 +++++++++++++++++++++++++++++++++++
 1 file changed, 35 insertions(+)

diff --git a/apps/bi/views.py b/apps/bi/views.py
index 037c352e..8c561312 100644
--- a/apps/bi/views.py
+++ b/apps/bi/views.py
@@ -11,6 +11,7 @@ from django.core.cache import cache
 from apps.utils.sql import execute_raw_sql, format_sqldata
 from apps.bi.services import check_sql_safe, format_json_with_placeholders
 from rest_framework.exceptions import ParseError
+from rest_framework.generics import get_object_or_404
 # Create your views here.
 
 class DatasetViewSet(CustomModelViewSet):
@@ -19,10 +20,44 @@ class DatasetViewSet(CustomModelViewSet):
     create_serializer_class = DatasetCreateUpdateSerializer
     update_serializer_class = DatasetCreateUpdateSerializer
     search_fields = ['name', 'code']
+    
+    def get_object(self):
+        """
+        Returns the object the view is displaying.
 
+        You may want to override this if you need to provide non-standard
+        queryset lookups.  Eg if objects are referenced using multiple
+        keyword arguments in the url conf.
+        """
+        queryset = self.filter_queryset(self.get_queryset())
+
+        # Perform the lookup filtering.
+        lookup_url_kwarg = self.lookup_url_kwarg or self.lookup_field
+
+        assert lookup_url_kwarg in self.kwargs, (
+            'Expected view %s to be called with a URL keyword argument '
+            'named "%s". Fix your URL conf, or set the `.lookup_field` '
+            'attribute on the view correctly.' %
+            (self.__class__.__name__, lookup_url_kwarg)
+        )
+
+        filter_kwargs = {self.lookup_field: self.kwargs[lookup_url_kwarg]}
+        try:
+            obj = get_object_or_404(queryset, **filter_kwargs)
+        except:
+            filter_kwargs = {'code': self.kwargs[lookup_url_kwarg]}
+            obj = get_object_or_404(queryset, **filter_kwargs)
+
+        # May raise a permission denied
+        self.check_object_permissions(self.request, obj)
+
+        return obj
+    
     @action(methods=['post'], detail=True, perms_map={'post': 'dataset.exec'}, serializer_class=DataExecSerializer, cache_seconds=0)
     def exec(self, request, pk=None):
         """执行sql查询
+
+        执行sql查询支持code
         """
         dt = self.get_object()
         rdata = DatasetSerializer(instance=dt).data