jumpserver/apps/assets/models/account.py

from django.db import models
from django.utils.translation import gettext_lazy as _
from simple_history.models import HistoricalRecords

from common.db.models import JMSBaseModel
from .protocol import ProtocolMixin
from .base import BaseUser, AbsConnectivity


__all__ = ['Account']


class Account(BaseUser, AbsConnectivity, ProtocolMixin):
    class Type(models.TextChoices):
        common = 'common', _('Common user')
        admin = 'admin', _('Admin user')

    protocol = models.CharField(
        max_length=16, choices=ProtocolMixin.Protocol.choices,
        default='ssh', verbose_name=_('Protocol')
    )
    type = models.CharField(max_length=16, choices=Type.choices, default=Type.common, verbose_name=_("Type"))
    asset = models.ForeignKey('assets.Asset', on_delete=models.CASCADE, verbose_name=_('Asset'))
    version = models.IntegerField(default=0, verbose_name=_('Version'))
    history = HistoricalRecords()

    class Meta:
        verbose_name = _('Account')
        unique_together = [('username', 'asset')]
        permissions = [
            ('view_accountsecret', _('Can view asset account secret')),
            ('change_accountsecret', _('Can change asset account secret')),
            ('view_historyaccount', _('Can view asset history account')),
            ('view_historyaccountsecret', _('Can view asset history account secret')),
        ]

    def __str__(self):
        return '{}://{}@{}'.format(self.protocol, self.username, self.asset.hostname)


class AccountTemplate(JMSBaseModel):
    pass