Skip to content

Commit 0634a4f

Browse files
authored
fix: Set the corresponding task to be available for application tool publishing (#4711)
1 parent 0707809 commit 0634a4f

File tree

2 files changed

+7
-1
lines changed

2 files changed

+7
-1
lines changed

apps/application/serializers/application.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -54,6 +54,7 @@
5454
from system_manage.serializers.user_resource_permission import UserResourcePermissionSerializer
5555
from tools.models import Tool, ToolScope
5656
from tools.serializers.tool import ToolExportModelSerializer
57+
from trigger.models import TriggerTask
5758
from users.models import User
5859
from users.serializers.user import is_workspace_manage
5960

@@ -893,6 +894,8 @@ def publish(self, instance, with_valid=True):
893894
else:
894895
access_token = application_access_token.access_token
895896
del_application_access_token(access_token)
897+
QuerySet(TriggerTask).filter(source_type="APPLICATION", source_id=self.data.get("application_id")).update(
898+
is_active=True)
896899
return self.one(with_valid=False)
897900

898901
@staticmethod

apps/tools/serializers/tool.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,7 @@
3838
from system_manage.models import AuthTargetType, WorkspaceUserResourcePermission
3939
from system_manage.serializers.user_resource_permission import UserResourcePermissionSerializer
4040
from tools.models import Tool, ToolScope, ToolFolder, ToolType
41+
from trigger.models import TriggerTask
4142
from users.serializers.user import is_workspace_manage
4243

4344
tool_executor = ToolExecutor()
@@ -518,7 +519,9 @@ def edit(self, instance, with_valid=True):
518519

519520
edit_dict['update_time'] = timezone.now()
520521
QuerySet(Tool).filter(id=self.data.get('id')).update(**edit_dict)
521-
522+
if 'is_active' in instance:
523+
QuerySet(TriggerTask).filter(source_type="TOOL", source_id=self.data.get('id')).update(
524+
is_active=instance.get('is_active'))
522525
return self.one()
523526

524527
def delete(self):

0 commit comments

Comments
 (0)