perf: Change secret update version

jumpserver · Dec 5, 2024 · ff0f9eb · ff0f9eb
1 parent d8dfaf0
commit ff0f9eb
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/apps/accounts/automations/change_secret/manager.py b/apps/accounts/automations/change_secret/manager.py
@@ -171,6 +171,8 @@ def on_host_success(self, host, result):
         if not account:
             print("Account not found, deleted ?")
             return
+
+        version_update_required = account.secret != recorder.new_secret
         account.secret = recorder.new_secret
         account.date_updated = timezone.now()
 
@@ -180,7 +182,10 @@ def on_host_success(self, host, result):
         while retry_count < max_retries:
             try:
                 recorder.save()
-                account.save(update_fields=['secret', 'date_updated'])
+                account_update_fields = ['secret', 'date_updated']
+                if version_update_required:
+                    account_update_fields.append('version')
+                account.save(update_fields=account_update_fields)
                 break
             except Exception as e:
                 retry_count += 1