from django.conf import settings from django.db import models from django.utils.translation import gettext_lazy as _ from accounts.const import AutomationTypes from accounts.models import Account from .base import AccountBaseAutomation from .change_secret import ChangeSecretMixin __all__ = ['PushAccountAutomation'] class PushAccountAutomation(ChangeSecretMixin, AccountBaseAutomation): triggers = models.JSONField(max_length=16, default=list, verbose_name=_('Triggers')) username = models.CharField(max_length=128, verbose_name=_('Username')) action = models.CharField(max_length=16, verbose_name=_('Action')) def create_nonlocal_accounts(self, usernames, asset): secret_type = self.secret_type account_usernames = asset.accounts \ .filter(secret_type=self.secret_type) \ .values_list('username', flat=True) create_usernames = set(usernames) - set(account_usernames) create_account_objs = [ Account( name=f'{username}-{secret_type}', username=username, secret_type=secret_type, asset=asset, ) for username in create_usernames ] Account.objects.bulk_create(create_account_objs) @property def dynamic_username(self): return self.username == '@USER' @dynamic_username.setter def dynamic_username(self, value): if value: self.username = '@USER' def save(self, *args, **kwargs): self.type = AutomationTypes.push_account if not settings.XPACK_LICENSE_IS_VALID: self.is_periodic = False super().save(*args, **kwargs) def to_attr_json(self): attr_json = super().to_attr_json() attr_json.update({ 'username': self.username, 'params': self.params, }) return attr_json class Meta: verbose_name = _("Push asset account")