fix: Use ConfirmOrIgnore for confirmed status in account updates

pull/15084/head
wangruidong 2025-03-20 11:12:49 +08:00 committed by w940853815
parent 33cb793c19
commit 2455afc2d2
1 changed files with 2 additions and 1 deletions

View File

@ -13,6 +13,7 @@ from accounts.filters import GatheredAccountFilterSet, NodeFilterBackend
from accounts.models import GatherAccountsAutomation, AutomationExecution, Account from accounts.models import GatherAccountsAutomation, AutomationExecution, Account
from accounts.models import GatheredAccount from accounts.models import GatheredAccount
from assets.models import Asset from assets.models import Asset
from common.const import ConfirmOrIgnore
from common.utils.http import is_true from common.utils.http import is_true
from orgs.mixins.api import OrgBulkModelViewSet from orgs.mixins.api import OrgBulkModelViewSet
from .base import AutomationExecutionViewSet from .base import AutomationExecutionViewSet
@ -97,7 +98,7 @@ class GatheredAccountViewSet(OrgBulkModelViewSet):
ids = validated_data.get('ids', []) ids = validated_data.get('ids', [])
new_status = validated_data.get('status') new_status = validated_data.get('status')
updated_instances = GatheredAccount.objects.filter(id__in=ids).select_related('asset') updated_instances = GatheredAccount.objects.filter(id__in=ids).select_related('asset')
if new_status == "confirmed": if new_status == ConfirmOrIgnore.confirmed:
GatheredAccount.sync_accounts(updated_instances) GatheredAccount.sync_accounts(updated_instances)
updated_instances.update(present=True) updated_instances.update(present=True)
updated_instances.update(status=new_status) updated_instances.update(status=new_status)