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)