mirror of https://github.com/jumpserver/jumpserver
76 lines
2.9 KiB
Python
76 lines
2.9 KiB
Python
from django.utils.translation import ugettext_lazy as _
|
|
from rest_framework import serializers
|
|
|
|
from acls.serializers.rules import ip_group_child_validator, ip_group_help_text
|
|
from common.serializers import BulkModelSerializer
|
|
from ..models import Endpoint, EndpointRule
|
|
from ..utils import db_port_manager
|
|
|
|
__all__ = ['EndpointSerializer', 'EndpointRuleSerializer']
|
|
|
|
|
|
class EndpointSerializer(BulkModelSerializer):
|
|
# 解决 luna 处理繁琐的问题, 返回 magnus 监听的当前 db 的 port
|
|
oracle_port = serializers.SerializerMethodField(label=_('Oracle port'))
|
|
oracle_port_range = serializers.CharField(
|
|
max_length=128, default=db_port_manager.oracle_port_range, read_only=True,
|
|
label=_('Oracle port range'),
|
|
help_text=_(
|
|
'Oracle proxy server listen port is dynamic, Each additional Oracle database instance adds a port listener'
|
|
)
|
|
)
|
|
|
|
class Meta:
|
|
model = Endpoint
|
|
fields_mini = ['id', 'name']
|
|
fields_small = [
|
|
'host', 'https_port', 'http_port', 'ssh_port', 'rdp_port',
|
|
'mysql_port', 'mariadb_port', 'postgresql_port', 'redis_port',
|
|
'oracle_port_range', 'oracle_port',
|
|
]
|
|
fields = fields_mini + fields_small + [
|
|
'comment', 'date_created', 'date_updated', 'created_by'
|
|
]
|
|
extra_kwargs = {
|
|
'host': {'help_text': 'Visit IP/host, if empty, use the current request instead'},
|
|
}
|
|
|
|
def get_oracle_port(self, obj: Endpoint):
|
|
view = self.context.get('view')
|
|
if not view or view.action not in ['smart']:
|
|
return 0
|
|
return obj.get_port(view.target_instance, view.target_protocol)
|
|
|
|
def get_extra_kwargs(self):
|
|
extra_kwargs = super().get_extra_kwargs()
|
|
model_fields = self.Meta.model._meta.fields
|
|
for field in model_fields:
|
|
if field.name.endswith('_port'):
|
|
kwargs = extra_kwargs.get(field.name, {})
|
|
kwargs = {'default': field.default, **kwargs}
|
|
extra_kwargs[field.name] = kwargs
|
|
return extra_kwargs
|
|
|
|
|
|
class EndpointRuleSerializer(BulkModelSerializer):
|
|
_ip_group_help_text = '{} <br> {}'.format(
|
|
ip_group_help_text,
|
|
_('If asset IP addresses under different endpoints conflict, use asset labels')
|
|
)
|
|
ip_group = serializers.ListField(
|
|
default=['*'], label=_('IP'), help_text=_ip_group_help_text,
|
|
child=serializers.CharField(max_length=1024, validators=[ip_group_child_validator])
|
|
)
|
|
endpoint_display = serializers.ReadOnlyField(source='endpoint.name', label=_('Endpoint'))
|
|
|
|
class Meta:
|
|
model = EndpointRule
|
|
fields_mini = ['id', 'name']
|
|
fields_small = fields_mini + ['ip_group', 'priority']
|
|
fields_fk = ['endpoint', 'endpoint_display']
|
|
fields = fields_mini + fields_small + fields_fk + [
|
|
'comment', 'date_created', 'date_updated', 'created_by'
|
|
]
|
|
extra_kwargs = {
|
|
}
|