From a6d040cd34ce27c0911665340ff80d7730df5099 Mon Sep 17 00:00:00 2001 From: feng <1304903146@qq.com> Date: Fri, 6 Dec 2024 17:59:30 +0800 Subject: [PATCH] perf: Automation filter org --- apps/assets/models/automations/base.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/apps/assets/models/automations/base.py b/apps/assets/models/automations/base.py index da0f73a1e..cdc3babad 100644 --- a/apps/assets/models/automations/base.py +++ b/apps/assets/models/automations/base.py @@ -10,7 +10,11 @@ from assets.tasks import execute_asset_automation_task from common.const.choices import Trigger from common.db.fields import EncryptJsonDictTextField from ops.mixin import PeriodTaskModelMixin -from orgs.mixins.models import OrgModelMixin, JMSOrgBaseModel +from orgs.mixins.models import OrgModelMixin, JMSOrgBaseModel, OrgManager + + +class BaseAutomationManager(OrgManager): + pass class BaseAutomation(PeriodTaskModelMixin, JMSOrgBaseModel): @@ -21,6 +25,8 @@ class BaseAutomation(PeriodTaskModelMixin, JMSOrgBaseModel): is_active = models.BooleanField(default=True, verbose_name=_("Is active")) params = models.JSONField(default=dict, verbose_name=_("Parameters")) + objects = BaseAutomationManager.from_queryset(models.QuerySet)() + def __str__(self): return self.name + '@' + str(self.created_by)