mirror of https://github.com/jumpserver/jumpserver
				
				
				
			fix: mongodb无指定数据库时,一些自动化任务会失败
							parent
							
								
									789695f90b
								
							
						
					
					
						commit
						c34b7e69eb
					
				| 
						 | 
					@ -1,3 +1,6 @@
 | 
				
			||||||
 | 
					from rest_framework.serializers import ValidationError
 | 
				
			||||||
 | 
					from django.utils.translation import ugettext_lazy as _
 | 
				
			||||||
 | 
					
 | 
				
			||||||
from assets.models import Database
 | 
					from assets.models import Database
 | 
				
			||||||
from .common import AssetSerializer
 | 
					from .common import AssetSerializer
 | 
				
			||||||
from ..gateway import GatewayWithAccountSecretSerializer
 | 
					from ..gateway import GatewayWithAccountSecretSerializer
 | 
				
			||||||
| 
						 | 
					@ -14,6 +17,13 @@ class DatabaseSerializer(AssetSerializer):
 | 
				
			||||||
        ]
 | 
					        ]
 | 
				
			||||||
        fields = AssetSerializer.Meta.fields + extra_fields
 | 
					        fields = AssetSerializer.Meta.fields + extra_fields
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def validate(self, attrs):
 | 
				
			||||||
 | 
					        platform = attrs.get('platform')
 | 
				
			||||||
 | 
					        if platform and getattr(platform, 'name') == 'MongoDB' \
 | 
				
			||||||
 | 
					                and not attrs.get('db_name'):
 | 
				
			||||||
 | 
					            raise ValidationError({'db_name': _('This field is required.')})
 | 
				
			||||||
 | 
					        return attrs
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
class DatabaseWithGatewaySerializer(DatabaseSerializer):
 | 
					class DatabaseWithGatewaySerializer(DatabaseSerializer):
 | 
				
			||||||
    gateway = GatewayWithAccountSecretSerializer()
 | 
					    gateway = GatewayWithAccountSecretSerializer()
 | 
				
			||||||
| 
						 | 
					
 | 
				
			||||||
		Loading…
	
		Reference in New Issue