From ac8998b9eef623b10fd7139b465ab47ac48041c6 Mon Sep 17 00:00:00 2001
From: feng <1304903146@qq.com>
Date: Wed, 6 Aug 2025 17:01:05 +0800
Subject: [PATCH] perf: Account risk delete normal account

---
 apps/accounts/automations/check_account/manager.py | 5 +++++
 apps/accounts/models/automations/check_account.py  | 6 ++++--
 2 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/apps/accounts/automations/check_account/manager.py b/apps/accounts/automations/check_account/manager.py
index 8ea9f4582..0a2120f33 100644
--- a/apps/accounts/automations/check_account/manager.py
+++ b/apps/accounts/automations/check_account/manager.py
@@ -240,6 +240,11 @@ class CheckAccountManager(BaseManager):
 
                 print("Check: {} => {}".format(account, msg))
                 if not error:
+                    AccountRisk.objects.filter(
+                        asset=account.asset,
+                        username=account.username,
+                        risk=handler.risk
+                    ).delete()
                     continue
                 self.add_risk(handler.risk, account)
             self.commit_risks(_assets)
diff --git a/apps/accounts/models/automations/check_account.py b/apps/accounts/models/automations/check_account.py
index 03636568a..aa6e4b2cd 100644
--- a/apps/accounts/models/automations/check_account.py
+++ b/apps/accounts/models/automations/check_account.py
@@ -68,8 +68,10 @@ class AccountRisk(JMSOrgBaseModel):
         related_name='risks', null=True
     )
     risk = models.CharField(max_length=128, verbose_name=_('Risk'), choices=RiskChoice.choices)
-    status = models.CharField(max_length=32, choices=ConfirmOrIgnore.choices, default=ConfirmOrIgnore.pending,
-                              blank=True, verbose_name=_('Status'))
+    status = models.CharField(
+        max_length=32, choices=ConfirmOrIgnore.choices, default=ConfirmOrIgnore.pending,
+        blank=True, verbose_name=_('Status')
+    )
     details = models.JSONField(default=list, verbose_name=_('Detail'))
 
     class Meta: