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.
133 lines
4.7 KiB
133 lines
4.7 KiB
from __future__ import unicode_literals
|
|
|
|
import uuid
|
|
|
|
from django.db import models
|
|
from django.utils.translation import ugettext_lazy as _
|
|
|
|
from users.models import User
|
|
from .backends.command.models import AbstractSessionCommand
|
|
|
|
|
|
class Terminal(models.Model):
|
|
id = models.UUIDField(default=uuid.uuid4, primary_key=True)
|
|
name = models.CharField(max_length=32, verbose_name=_('Name'))
|
|
remote_addr = models.CharField(max_length=128, verbose_name=_('Remote Address'))
|
|
ssh_port = models.IntegerField(verbose_name=_('SSH Port'), default=2222)
|
|
http_port = models.IntegerField(verbose_name=_('HTTP Port'), default=5000)
|
|
user = models.OneToOneField(User, related_name='terminal', verbose_name='Application User', null=True, on_delete=models.CASCADE)
|
|
is_accepted = models.BooleanField(default=False, verbose_name='Is Accepted')
|
|
is_deleted = models.BooleanField(default=False)
|
|
date_created = models.DateTimeField(auto_now_add=True)
|
|
comment = models.TextField(blank=True, verbose_name=_('Comment'))
|
|
|
|
@property
|
|
def is_active(self):
|
|
if self.user and self.user.is_active:
|
|
return True
|
|
return False
|
|
|
|
@is_active.setter
|
|
def is_active(self, active):
|
|
if self.user:
|
|
self.user.is_active = active
|
|
self.user.save()
|
|
|
|
def create_app_user(self):
|
|
random = uuid.uuid4().hex[:6]
|
|
user, access_key = User.create_app_user(name="{}-{}".format(self.name, random), comment=self.comment)
|
|
self.user = user
|
|
self.save()
|
|
return user, access_key
|
|
|
|
def delete(self, using=None, keep_parents=False):
|
|
if self.user:
|
|
self.user.delete()
|
|
self.user = None
|
|
self.is_deleted = True
|
|
self.save()
|
|
return
|
|
|
|
def __str__(self):
|
|
status = "Active"
|
|
if not self.is_accepted:
|
|
status = "NotAccept"
|
|
elif self.is_deleted:
|
|
status = "Deleted"
|
|
elif not self.is_active:
|
|
status = "Disable"
|
|
return '%s: %s' % (self.name, status)
|
|
|
|
class Meta:
|
|
ordering = ('is_accepted',)
|
|
db_table = "terminal"
|
|
|
|
|
|
class Status(models.Model):
|
|
id = models.UUIDField(default=uuid.uuid4, primary_key=True)
|
|
session_online = models.IntegerField(verbose_name=_("Session Online"), default=0)
|
|
cpu_used = models.FloatField(verbose_name=_("CPU Usage"))
|
|
memory_used = models.FloatField(verbose_name=_("Memory Used"))
|
|
connections = models.IntegerField(verbose_name=_("Connections"))
|
|
threads = models.IntegerField(verbose_name=_("Threads"))
|
|
boot_time = models.FloatField(verbose_name=_("Boot Time"))
|
|
terminal = models.ForeignKey(Terminal, null=True, on_delete=models.CASCADE)
|
|
date_created = models.DateTimeField(auto_now_add=True)
|
|
|
|
class Meta:
|
|
db_table = 'terminal_status'
|
|
get_latest_by = 'date_created'
|
|
|
|
def __str__(self):
|
|
return self.date_created.strftime("%Y-%m-%d %H:%M:%S")
|
|
|
|
|
|
class Session(models.Model):
|
|
LOGIN_FROM_CHOICES = (
|
|
('ST', 'SSH Terminal'),
|
|
('WT', 'Web Terminal'),
|
|
)
|
|
|
|
id = models.UUIDField(default=uuid.uuid4, primary_key=True)
|
|
user = models.CharField(max_length=128, verbose_name=_("User"))
|
|
asset = models.CharField(max_length=1024, verbose_name=_("Asset"))
|
|
system_user = models.CharField(max_length=128, verbose_name=_("System user"))
|
|
login_from = models.CharField(max_length=2, choices=LOGIN_FROM_CHOICES, default="ST")
|
|
is_finished = models.BooleanField(default=False)
|
|
has_replay = models.BooleanField(default=False, verbose_name=_("Replay"))
|
|
has_command = models.BooleanField(default=False, verbose_name=_("Command"))
|
|
terminal = models.ForeignKey(Terminal, null=True, on_delete=models.CASCADE)
|
|
date_start = models.DateTimeField(verbose_name=_("Date start"))
|
|
date_end = models.DateTimeField(verbose_name=_("Date end"), null=True)
|
|
|
|
class Meta:
|
|
db_table = "terminal_session"
|
|
ordering = ["-date_start"]
|
|
|
|
def __str__(self):
|
|
return "{0.id} of {0.user} to {0.asset}".format(self)
|
|
|
|
|
|
class Task(models.Model):
|
|
NAME_CHOICES = (
|
|
("kill_session", "Kill Session"),
|
|
)
|
|
|
|
id = models.UUIDField(default=uuid.uuid4, primary_key=True)
|
|
name = models.CharField(max_length=128, choices=NAME_CHOICES, verbose_name=_("Name"))
|
|
args = models.CharField(max_length=1024, verbose_name=_("Args"))
|
|
terminal = models.ForeignKey(Terminal, null=True, on_delete=models.CASCADE)
|
|
is_finished = models.BooleanField(default=False)
|
|
date_created = models.DateTimeField(auto_now_add=True)
|
|
date_finished = models.DateTimeField(null=True)
|
|
|
|
class Meta:
|
|
db_table = "terminal_task"
|
|
|
|
|
|
class Command(AbstractSessionCommand):
|
|
|
|
class Meta:
|
|
db_table = "terminal_command"
|
|
ordering = ('-timestamp',)
|