github
/
jumpserver
mirror of https://github.com/jumpserver/jumpserver
1
0
Fork 0
Code Issues Releases Wiki Activity

perf: 改密失败发给收件人 (#13009) 

Co-authored-by: feng <1304903146@qq.com>
pull/13011/head
fit2bot 2024-04-12 11:53:47 +08:00 committed by GitHub
parent 2ac87e4ad6
commit f3bd727c32
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 10 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
apps/accounts/automations/change_secret/manager.py
View File

@ -240,27 +240,28 @@ class ChangeSecretManager(AccountBasePlaybookManager):
r for r in recorders r for r in recorders
if r.status == ChangeSecretRecordStatusChoice.failed.value if r.status == ChangeSecretRecordStatusChoice.failed.value
] ]
super_users = User.get_super_admins()
if failed_recorders and super_users: recipients = self.execution.recipients
recipients = User.objects.filter(id__in=list(recipients.keys()))
if not recipients:
return
if failed_recorders:
name = self.execution.snapshot.get('name') name = self.execution.snapshot.get('name')
execution_id = str(self.execution.id) execution_id = str(self.execution.id)
_ids = [r.id for r in failed_recorders] _ids = [r.id for r in failed_recorders]
asset_account_errors = ChangeSecretRecord.objects.filter( asset_account_errors = ChangeSecretRecord.objects.filter(
id__in=_ids).values_list('asset__name', 'account__username', 'error') id__in=_ids).values_list('asset__name', 'account__username', 'error')
for user in super_users: for user in recipients:
ChangeSecretFailedMsg(name, execution_id, user, asset_account_errors).publish() ChangeSecretFailedMsg(name, execution_id, user, asset_account_errors).publish()
self.send_recorder_mail(recorders, summary) if not recorders:
def send_recorder_mail(self, recorders, summary):
recipients = self.execution.recipients
if not recorders or not recipients:
return return
recipients = User.objects.filter(id__in=list(recipients.keys())) self.send_recorder_mail(recipients, recorders, summary)
def send_recorder_mail(self, recipients, recorders, summary):
name = self.execution.snapshot['name'] name = self.execution.snapshot['name']
path = os.path.join(os.path.dirname(settings.BASE_DIR), 'tmp') path = os.path.join(os.path.dirname(settings.BASE_DIR), 'tmp')
filename = os.path.join(path, f'{name}-{local_now_filename()}-{time.time()}.xlsx') filename = os.path.join(path, f'{name}-{local_now_filename()}-{time.time()}.xlsx')