From 4eb5d51840550137f724c58fbebed344a8b617cc Mon Sep 17 00:00:00 2001 From: ibuler Date: Wed, 14 Oct 2020 19:39:55 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20domain=E7=AB=AF=E5=8F=A3=E5=8F=AF?= =?UTF-8?q?=E8=83=BD=E9=9A=8F=E4=BE=BF=E5=A1=AB=E5=86=99=E7=9A=84=E9=97=AE?= =?UTF-8?q?=E9=A2=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- apps/assets/api/domain.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/apps/assets/api/domain.py b/apps/assets/api/domain.py index 0e7108fd4..7c32883d8 100644 --- a/apps/assets/api/domain.py +++ b/apps/assets/api/domain.py @@ -1,7 +1,9 @@ # ~*~ coding: utf-8 ~*~ -from rest_framework.views import APIView, Response from django.views.generic.detail import SingleObjectMixin +from django.utils.translation import ugettext as _ +from rest_framework.views import APIView, Response +from rest_framework.serializers import ValidationError from common.utils import get_logger from common.permissions import IsOrgAdmin, IsOrgAdminOrAppUser @@ -42,6 +44,10 @@ class GatewayTestConnectionApi(SingleObjectMixin, APIView): def post(self, request, *args, **kwargs): self.object = self.get_object(Gateway.objects.all()) local_port = self.request.data.get('port') or self.object.port + try: + local_port = int(local_port) + except ValueError: + raise ValidationError({'port': _('Number required')}) ok, e = self.object.test_connective(local_port=local_port) if ok: return Response("ok")