From b1f6798c65bc30e17fa45a28cab8fb25c8d82af4 Mon Sep 17 00:00:00 2001
From: caoqianming <caoqianming@foxmail.com>
Date: Fri, 25 Jul 2025 11:34:56 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20dataset=E6=B7=BB=E5=8A=A0enabled?=
 =?UTF-8?q?=E5=AD=97=E6=AE=B5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/bi/migrations/0006_dataset_enabled.py | 18 ++++++++++++++++++
 apps/bi/models.py                          |  1 +
 apps/bi/views.py                           |  2 ++
 3 files changed, 21 insertions(+)
 create mode 100644 apps/bi/migrations/0006_dataset_enabled.py

diff --git a/apps/bi/migrations/0006_dataset_enabled.py b/apps/bi/migrations/0006_dataset_enabled.py
new file mode 100644
index 00000000..7f0f4625
--- /dev/null
+++ b/apps/bi/migrations/0006_dataset_enabled.py
@@ -0,0 +1,18 @@
+# Generated by Django 3.2.12 on 2025-07-25 03:32
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('bi', '0005_datasetrecord'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='dataset',
+            name='enabled',
+            field=models.BooleanField(default=True, verbose_name='启用'),
+        ),
+    ]
diff --git a/apps/bi/models.py b/apps/bi/models.py
index b62354fb..f163e725 100644
--- a/apps/bi/models.py
+++ b/apps/bi/models.py
@@ -12,6 +12,7 @@ class Dataset(CommonBDModel):
     test_param = models.JSONField('测试查询参数', default=dict, blank=True)
     default_param = models.JSONField('默认查询参数', default=dict, blank=True)
     cache_seconds = models.PositiveIntegerField('缓存秒数', default=10, blank=True)
+    enabled = models.BooleanField('启用', default=True)
 
 
 # class Report(CommonBDModel):
diff --git a/apps/bi/views.py b/apps/bi/views.py
index 6321fe40..87e88192 100644
--- a/apps/bi/views.py
+++ b/apps/bi/views.py
@@ -64,6 +64,8 @@ class DatasetViewSet(CustomModelViewSet):
         执行sql查询支持code
         """
         dt: Dataset = self.get_object()
+        if not dt.enabled:
+            raise ParseError(f'{dt.name}-该查询未启用')
         rdata = DatasetSerializer(instance=dt).data
         xquery = request.data.get('query', {})
         is_test = request.data.get('is_test', False)