feat: 优化 DB Listen Port 映射规则逻辑

pull/8892/head
Jiangjie.Bai 2022-09-22 14:51:19 +08:00
parent c9afd94714
commit b8ec60dea1
1 changed files with 8 additions and 5 deletions

View File

@ -1,7 +1,7 @@
# coding: utf-8 # coding: utf-8
# #
from orgs.mixins.api import OrgBulkModelViewSet from orgs.mixins.api import OrgBulkModelViewSet
from rest_framework import generics, status from rest_framework import status
from rest_framework.decorators import action from rest_framework.decorators import action
from rest_framework.response import Response from rest_framework.response import Response
from rest_framework.viewsets import GenericViewSet from rest_framework.viewsets import GenericViewSet
@ -60,8 +60,11 @@ class DBListenPortViewSet(GenericViewSet):
def db_info(self, request, *args, **kwargs): def db_info(self, request, *args, **kwargs):
port = request.data.get("port") port = request.data.get("port")
db, msg = db_port_manager.get_db_by_port(port) db, msg = db_port_manager.get_db_by_port(port)
if db is None: if db:
serializer = serializers.AppSerializer(instance=db)
data = serializer.data
_status = status.HTTP_200_OK
else:
data = {'error': msg} data = {'error': msg}
return Response(data=data, status=status.HTTP_404_NOT_FOUND) _status = status.HTTP_404_NOT_FOUND
serializer = serializers.AppSerializer(instance=db) return Response(data=data, status=_status)
return Response(data=serializer.data, status=status.HTTP_201_CREATED)