mirror of https://github.com/jumpserver/jumpserver
61 lines
2.2 KiB
Python
61 lines
2.2 KiB
Python
from datetime import datetime
|
|
|
|
from celery import shared_task
|
|
from django.shortcuts import get_object_or_404
|
|
from django.utils.translation import gettext_lazy as _
|
|
|
|
from orgs.utils import tmp_to_root_org, tmp_to_org
|
|
|
|
|
|
@shared_task(
|
|
verbose_name=_('Template sync info to related accounts'),
|
|
activity_callback=lambda self, template_id, *args, **kwargs: (template_id, None)
|
|
)
|
|
def template_sync_related_accounts(template_id, user_id=None):
|
|
from accounts.models import Account, AccountTemplate
|
|
with tmp_to_root_org():
|
|
template = get_object_or_404(AccountTemplate, id=template_id)
|
|
org_id = template.org_id
|
|
|
|
with tmp_to_org(org_id):
|
|
accounts = Account.objects.filter(source_id=template_id)
|
|
if not accounts:
|
|
print('\033[35m>>> 没有需要同步的账号, 结束任务')
|
|
print('\033[0m')
|
|
return
|
|
|
|
failed, succeeded = 0, 0
|
|
succeeded_account_ids = []
|
|
name = template.name
|
|
username = template.username
|
|
secret_type = template.secret_type
|
|
print(f'\033[32m>>> 开始同步模版名称、用户名、密钥类型到相关联的账号 ({datetime.now().strftime("%Y-%m-%d %H:%M:%S")})')
|
|
with tmp_to_org(org_id):
|
|
for account in accounts:
|
|
account.name = name
|
|
account.username = username
|
|
account.secret_type = secret_type
|
|
try:
|
|
account.save(update_fields=['name', 'username', 'secret_type'])
|
|
succeeded += 1
|
|
succeeded_account_ids.append(account.id)
|
|
except Exception as e:
|
|
account.source_id = None
|
|
account.save(update_fields=['source_id'])
|
|
print(f'\033[31m- 同步失败: [{account}] 原因: [{e}]')
|
|
failed += 1
|
|
accounts = Account.objects.filter(id__in=succeeded_account_ids)
|
|
if accounts:
|
|
print(f'\033[33m>>> 批量更新账号密文 ({datetime.now().strftime("%Y-%m-%d %H:%M:%S")})')
|
|
template.bulk_sync_account_secret(accounts, user_id)
|
|
|
|
total = succeeded + failed
|
|
print(
|
|
f'\033[33m>>> 同步完成:, '
|
|
f'共计: {total}, '
|
|
f'成功: {succeeded}, '
|
|
f'失败: {failed}, '
|
|
f'({datetime.now().strftime("%Y-%m-%d %H:%M:%S")}) '
|
|
)
|
|
print('\033[0m')
|