mirror of https://github.com/jumpserver/jumpserver
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
44 lines
1.5 KiB
44 lines
1.5 KiB
# ~*~ 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")
|
|
|