mirror of https://github.com/jumpserver/jumpserver
34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
from rest_framework.serializers import ValidationError
|
|
from django.utils.translation import ugettext_lazy as _
|
|
|
|
from assets.models import Database
|
|
from .common import AssetSerializer
|
|
from ..gateway import GatewayWithAccountSecretSerializer
|
|
|
|
__all__ = ['DatabaseSerializer', 'DatabaseWithGatewaySerializer']
|
|
|
|
|
|
class DatabaseSerializer(AssetSerializer):
|
|
class Meta(AssetSerializer.Meta):
|
|
model = Database
|
|
extra_fields = [
|
|
'db_name', 'use_ssl', 'ca_cert', 'client_cert',
|
|
'client_key', 'allow_invalid_cert'
|
|
]
|
|
fields = AssetSerializer.Meta.fields + extra_fields
|
|
|
|
def validate(self, attrs):
|
|
platform = attrs.get('platform')
|
|
db_type_required = ('mongodb', 'postgresql')
|
|
if platform and getattr(platform, 'type') in db_type_required \
|
|
and not attrs.get('db_name'):
|
|
raise ValidationError({'db_name': _('This field is required.')})
|
|
return attrs
|
|
|
|
|
|
class DatabaseWithGatewaySerializer(DatabaseSerializer):
|
|
gateway = GatewayWithAccountSecretSerializer()
|
|
|
|
class Meta(DatabaseSerializer.Meta):
|
|
fields = DatabaseSerializer.Meta.fields + ['gateway']
|