mirror of
https://github.com/jumpserver/jumpserver.git
synced 2026-04-07 21:03:15 +00:00
0771b804d1
* refactor: 重构危险命令告警类型: Warning * Update _msg_command_warning.html * Update _msg_command_warning.html * Update command.py * Update django.po * perf: 优化 command acl warning 的代码逻辑 * perf: 优化 command acl warning 的代码逻辑 * perf: 优化 CommandWarningMessage 逻辑 --------- Co-authored-by: fangfang.dong <fangfang.dong@fit2cloud.com> Co-authored-by: Bai <baijiangjie@gmail.com>
58 lines
1.9 KiB
Python
58 lines
1.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
import uuid
|
|
from datetime import datetime
|
|
|
|
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
|
|
from terminal.const import RiskLevelChoices
|
|
|
|
|
|
class AbstractSessionCommand(OrgModelMixin):
|
|
|
|
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=RiskLevelChoices.accept, choices=RiskLevelChoices.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(RiskLevelChoices.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
|