mirror of https://github.com/jumpserver/jumpserver
145 lines
4.4 KiB
Python
145 lines
4.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
from django.utils.translation import ugettext_lazy as _
|
|
from rest_framework import serializers
|
|
|
|
from common.drf.fields import LabeledChoiceField
|
|
from terminal.models import Session
|
|
from . import models
|
|
from .const import (
|
|
ActionChoices,
|
|
OperateChoices,
|
|
MFAChoices,
|
|
LoginStatusChoices,
|
|
LoginTypeChoices,
|
|
)
|
|
|
|
|
|
class FTPLogSerializer(serializers.ModelSerializer):
|
|
operate = LabeledChoiceField(choices=OperateChoices.choices, label=_("Operate"))
|
|
|
|
class Meta:
|
|
model = models.FTPLog
|
|
fields_mini = ["id"]
|
|
fields_small = fields_mini + [
|
|
"user",
|
|
"remote_addr",
|
|
"asset",
|
|
"system_user",
|
|
"org_id",
|
|
"operate",
|
|
"filename",
|
|
"is_success",
|
|
"date_start",
|
|
]
|
|
fields = fields_small
|
|
|
|
|
|
class UserLoginLogSerializer(serializers.ModelSerializer):
|
|
mfa = LabeledChoiceField(choices=MFAChoices.choices, label=_("MFA"))
|
|
type = LabeledChoiceField(choices=LoginTypeChoices.choices, label=_("Type"))
|
|
status = LabeledChoiceField(choices=LoginStatusChoices.choices, label=_("Status"))
|
|
|
|
class Meta:
|
|
model = models.UserLoginLog
|
|
fields_mini = ["id"]
|
|
fields_small = fields_mini + [
|
|
"username",
|
|
"type",
|
|
"ip",
|
|
"city",
|
|
"user_agent",
|
|
"mfa",
|
|
"reason",
|
|
"reason_display",
|
|
"backend",
|
|
"backend_display",
|
|
"status",
|
|
"datetime",
|
|
]
|
|
fields = fields_small
|
|
extra_kwargs = {
|
|
"user_agent": {"label": _("User agent")},
|
|
"reason_display": {"label": _("Reason display")},
|
|
"backend_display": {"label": _("Authentication backend")},
|
|
}
|
|
|
|
|
|
class OperateLogActionDetailSerializer(serializers.ModelSerializer):
|
|
class Meta:
|
|
model = models.OperateLog
|
|
fields = ('before', 'after')
|
|
|
|
|
|
class OperateLogSerializer(serializers.ModelSerializer):
|
|
action = LabeledChoiceField(choices=ActionChoices.choices, label=_("Action"))
|
|
|
|
class Meta:
|
|
model = models.OperateLog
|
|
fields_mini = ["id"]
|
|
fields_small = fields_mini + [
|
|
"user",
|
|
"action",
|
|
"resource_type",
|
|
"resource_type_display",
|
|
"resource",
|
|
"remote_addr",
|
|
"datetime",
|
|
"org_id",
|
|
]
|
|
fields = fields_small
|
|
extra_kwargs = {"resource_type_display": {"label": _("Resource Type")}}
|
|
|
|
|
|
class PasswordChangeLogSerializer(serializers.ModelSerializer):
|
|
class Meta:
|
|
model = models.PasswordChangeLog
|
|
fields = ("id", "user", "change_by", "remote_addr", "datetime")
|
|
|
|
|
|
class SessionAuditSerializer(serializers.ModelSerializer):
|
|
class Meta:
|
|
model = Session
|
|
fields = "__all__"
|
|
|
|
|
|
#
|
|
# class CommandExecutionSerializer(serializers.ModelSerializer):
|
|
# is_success = serializers.BooleanField(read_only=True, label=_('Is success'))
|
|
# hosts_display = serializers.ListSerializer(
|
|
# child=serializers.CharField(), source='hosts', read_only=True, label=_('Hosts display')
|
|
# )
|
|
#
|
|
# class Meta:
|
|
# model = CommandExecution
|
|
# fields_mini = ['id']
|
|
# fields_small = fields_mini + [
|
|
# 'command', 'is_finished', 'user',
|
|
# 'date_start', 'result', 'is_success', 'org_id'
|
|
# ]
|
|
# fields = fields_small + ['hosts', 'hosts_display', 'user_display']
|
|
# extra_kwargs = {
|
|
# 'result': {'label': _('Result')}, # model 上的方法，只能在这修改
|
|
# 'is_success': {'label': _('Is success')},
|
|
# 'hosts': {'label': _('Hosts')}, # 外键，会生成 sql。不在 model 上修改
|
|
# 'user': {'label': _('User')},
|
|
# 'user_display': {'label': _('User display')},
|
|
# }
|
|
#
|
|
# @classmethod
|
|
# def setup_eager_loading(cls, queryset):
|
|
# """ Perform necessary eager loading of data. """
|
|
# queryset = queryset.prefetch_related('user', 'hosts')
|
|
# return queryset
|
|
#
|
|
#
|
|
# class CommandExecutionHostsRelationSerializer(BulkSerializerMixin, serializers.ModelSerializer):
|
|
# asset_display = serializers.ReadOnlyField()
|
|
# commandexecution_display = serializers.ReadOnlyField()
|
|
#
|
|
# class Meta:
|
|
# model = CommandExecution.hosts.through
|
|
# fields = [
|
|
# 'id', 'asset', 'asset_display', 'commandexecution', 'commandexecution_display'
|
|
# ]
|