mirror of https://github.com/jumpserver/jumpserver
58 lines
2.0 KiB
Python
58 lines
2.0 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
from datetime import datetime
|
|
import uuid
|
|
from django.db import models
|
|
from django.utils.translation import ugettext_lazy as _
|
|
|
|
from common.utils.common import lazyproperty
|
|
from orgs.mixins.models import OrgModelMixin
|
|
|
|
|
|
class AbstractSessionCommand(OrgModelMixin):
|
|
RISK_LEVEL_ORDINARY = 0
|
|
RISK_LEVEL_DANGEROUS = 5
|
|
RISK_LEVEL_CHOICES = (
|
|
(RISK_LEVEL_ORDINARY, _('Ordinary')),
|
|
(RISK_LEVEL_DANGEROUS, _('Dangerous')),
|
|
)
|
|
id = models.UUIDField(default=uuid.uuid4, primary_key=True)
|
|
user = models.CharField(max_length=64, db_index=True, verbose_name=_("User"))
|
|
asset = models.CharField(max_length=128, db_index=True, verbose_name=_("Asset"))
|
|
account = models.CharField(max_length=64, db_index=True, verbose_name=_("Account"))
|
|
input = models.CharField(max_length=128, db_index=True, verbose_name=_("Input"))
|
|
output = models.CharField(max_length=1024, blank=True, verbose_name=_("Output"))
|
|
session = models.CharField(max_length=36, db_index=True, verbose_name=_("Session"))
|
|
risk_level = models.SmallIntegerField(default=RISK_LEVEL_ORDINARY, choices=RISK_LEVEL_CHOICES, db_index=True, verbose_name=_("Risk level"))
|
|
timestamp = models.IntegerField(db_index=True)
|
|
|
|
class Meta:
|
|
abstract = True
|
|
|
|
@lazyproperty
|
|
def timestamp_display(self):
|
|
return datetime.fromtimestamp(self.timestamp)
|
|
|
|
@lazyproperty
|
|
def remote_addr(self):
|
|
from terminal.models import Session
|
|
session = Session.objects.filter(id=self.session).first()
|
|
if session:
|
|
return session.remote_addr
|
|
else:
|
|
return ''
|
|
|
|
@classmethod
|
|
def get_risk_level_str(cls, risk_level):
|
|
risk_mapper = dict(cls.RISK_LEVEL_CHOICES)
|
|
return risk_mapper.get(risk_level)
|
|
|
|
def to_dict(self):
|
|
d = {}
|
|
for field in self._meta.fields:
|
|
d[field.name] = getattr(self, field.name)
|
|
return d
|
|
|
|
def __str__(self):
|
|
return self.input
|