jumpserver/apps/ops/models/adhoc.py

# ~*~ coding: utf-8 ~*~
import uuid

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

from common.utils import get_logger

__all__ = ["AdHoc"]

from common.db.models import JMSBaseModel
from ops.const import AdHocModules, Scope

logger = get_logger(__file__)


class AdHoc(JMSBaseModel):
    id = models.UUIDField(default=uuid.uuid4, primary_key=True)
    name = models.CharField(max_length=128, verbose_name=_('Name'))
    pattern = models.CharField(max_length=1024, verbose_name=_("Pattern"), default='all')
    module = models.CharField(max_length=128, choices=AdHocModules.choices, default=AdHocModules.shell,
                              verbose_name=_('Module'))
    args = models.CharField(max_length=8192, default='', verbose_name=_('Args'))
    creator = models.ForeignKey('users.User', verbose_name=_("Creator"), on_delete=models.SET_NULL, null=True)
    comment = models.CharField(max_length=1024, default='', verbose_name=_('Comment'), null=True, blank=True)
    scope = models.CharField(max_length=64, default=Scope.public, verbose_name=_('Scope'))

    @property
    def row_count(self):
        if len(self.args) == 0:
            return 0
        count = str(self.args).count('\n')
        return count + 1

    @property
    def size(self):
        return len(self.args)

    def __str__(self):
        return "{}: {}".format(self.module, self.args)

    class Meta:
        unique_together = [('name', 'creator')]
        verbose_name = _("Adhoc")