mirror of https://github.com/jumpserver/jumpserver
76 lines
2.6 KiB
Python
76 lines
2.6 KiB
Python
from django.db import models
|
|
from django.db.models import Q
|
|
from django.utils.translation import gettext_lazy as _
|
|
|
|
from accounts.const import AutomationTypes, Source
|
|
from accounts.models import Account
|
|
from orgs.mixins.models import JMSOrgBaseModel
|
|
from .base import AccountBaseAutomation
|
|
|
|
__all__ = ['GatherAccountsAutomation', 'GatheredAccount']
|
|
|
|
|
|
class GatheredAccount(JMSOrgBaseModel):
|
|
present = models.BooleanField(default=True, verbose_name=_("Present"))
|
|
date_last_login = models.DateTimeField(null=True, verbose_name=_("Date login"))
|
|
asset = models.ForeignKey('assets.Asset', on_delete=models.CASCADE, verbose_name=_("Asset"))
|
|
username = models.CharField(max_length=32, blank=True, db_index=True, verbose_name=_('Username'))
|
|
address_last_login = models.CharField(max_length=39, default='', verbose_name=_("Address login"))
|
|
|
|
@property
|
|
def address(self):
|
|
return self.asset.address
|
|
|
|
@staticmethod
|
|
def sync_accounts(gathered_accounts):
|
|
account_objs = []
|
|
for gathered_account in gathered_accounts:
|
|
asset_id = gathered_account.asset_id
|
|
username = gathered_account.username
|
|
accounts = Account.objects.filter(
|
|
Q(asset_id=asset_id, username=username) |
|
|
Q(asset_id=asset_id, name=username)
|
|
)
|
|
if accounts.exists():
|
|
continue
|
|
account = Account(
|
|
asset_id=asset_id, username=username,
|
|
name=username, source=Source.COLLECTED
|
|
)
|
|
account_objs.append(account)
|
|
Account.objects.bulk_create(account_objs)
|
|
|
|
class Meta:
|
|
verbose_name = _("Gather asset accounts")
|
|
unique_together = [
|
|
('username', 'asset'),
|
|
]
|
|
ordering = ['asset']
|
|
|
|
def __str__(self):
|
|
return '{}: {}'.format(self.asset, self.username)
|
|
|
|
|
|
class GatherAccountsAutomation(AccountBaseAutomation):
|
|
is_sync_account = models.BooleanField(
|
|
default=False, blank=True, verbose_name=_("Is sync account")
|
|
)
|
|
recipients = models.ManyToManyField('users.User', verbose_name=_("Recipient"), blank=True)
|
|
|
|
def to_attr_json(self):
|
|
attr_json = super().to_attr_json()
|
|
attr_json.update({
|
|
'is_sync_account': self.is_sync_account,
|
|
'recipients': [
|
|
str(recipient.id) for recipient in self.recipients.all()
|
|
]
|
|
})
|
|
return attr_json
|
|
|
|
def save(self, *args, **kwargs):
|
|
self.type = AutomationTypes.gather_accounts
|
|
super().save(*args, **kwargs)
|
|
|
|
class Meta:
|
|
verbose_name = _('Gather account automation')
|