From 7bac1b42e46247e8be1ed8c36035224af2ae7672 Mon Sep 17 00:00:00 2001
From: feng <1304903146@qq.com>
Date: Fri, 14 Mar 2025 17:40:07 +0800
Subject: [PATCH] perf: risk add account

---
 apps/accounts/risk_handlers.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/apps/accounts/risk_handlers.py b/apps/accounts/risk_handlers.py
index 7184ef26d..995e1a479 100644
--- a/apps/accounts/risk_handlers.py
+++ b/apps/accounts/risk_handlers.py
@@ -97,11 +97,15 @@ class RiskHandler:
             "secret_type": SecretType.PASSWORD,
             "source": "collected",
         }
-        self.asset.accounts.get_or_create(defaults=data, username=self.username)
+        account, __ = self.asset.accounts.get_or_create(defaults=data, username=self.username)
         GatheredAccount.objects.filter(asset=self.asset, username=self.username).update(
             present=True, status=ConfirmOrIgnore.confirmed
         )
         self.risk = RiskChoice.new_found
+        
+        risk = self.get_risk()
+        risk.account = account
+        risk.save()
 
     def handle_disable_remote(self):
         pass