fix: 修改session command 翻译

pull/4107/head
ibuler 2020-06-16 13:39:49 +08:00
parent b81e424e80
commit da4ad11a69
2 changed files with 17 additions and 8 deletions

View File

@ -26,7 +26,8 @@ class CommandQueryMixin:
command_store = get_command_storage() command_store = get_command_storage()
permission_classes = [IsOrgAdminOrAppUser | IsOrgAuditor] permission_classes = [IsOrgAdminOrAppUser | IsOrgAuditor]
filter_fields = [ filter_fields = [
"asset", "system_user", "user", "session", "asset", "system_user", "user", "session", "risk_level",
"input"
] ]
default_days_ago = 5 default_days_ago = 5

View File

@ -1,18 +1,26 @@
# ~*~ coding: utf-8 ~*~ # ~*~ coding: utf-8 ~*~
from django.utils.translation import ugettext_lazy as _
from rest_framework import serializers from rest_framework import serializers
from .models import AbstractSessionCommand
class SessionCommandSerializer(serializers.Serializer): class SessionCommandSerializer(serializers.Serializer):
"""使用这个类作为基础Command Log Serializer类, 用来序列化""" """使用这个类作为基础Command Log Serializer类, 用来序列化"""
id = serializers.UUIDField(read_only=True) id = serializers.UUIDField(read_only=True)
user = serializers.CharField(max_length=64) user = serializers.CharField(max_length=64, label=_("User"))
asset = serializers.CharField(max_length=128) asset = serializers.CharField(max_length=128, label=_("Asset"))
system_user = serializers.CharField(max_length=64) system_user = serializers.CharField(max_length=64, label=_("System user"))
input = serializers.CharField(max_length=128) input = serializers.CharField(max_length=128, label=_("Command"))
output = serializers.CharField(max_length=1024, allow_blank=True) output = serializers.CharField(max_length=1024, allow_blank=True, label=_("Output"))
session = serializers.CharField(max_length=36) session = serializers.CharField(max_length=36, label=_("Session"))
risk_level = serializers.IntegerField(required=False) risk_level = serializers.ChoiceField(required=False, label=_("Risk level"), choices=AbstractSessionCommand.RISK_LEVEL_CHOICES)
risk_level_display = serializers.SerializerMethodField()
org_id = serializers.CharField(max_length=36, required=False, default='', allow_null=True, allow_blank=True) org_id = serializers.CharField(max_length=36, required=False, default='', allow_null=True, allow_blank=True)
timestamp = serializers.IntegerField() timestamp = serializers.IntegerField()
@staticmethod
def get_risk_level_display(obj):
risk_mapper = dict(AbstractSessionCommand.RISK_LEVEL_CHOICES)
return risk_mapper.get(obj.risk_level)