mirror of
https://github.com/jumpserver/jumpserver.git
synced 2026-01-18 00:18:49 +00:00
f3dc9b886b
* [Update] 拆分filter org * [Update] 修改session支持protocol搜索 * [Bugfix] 修复判断问题 * [Update] 支持收藏资产 * [update] 修改org resource queryset * [Update] 修改form serializer 对应的多对多字段 * [Bugfix] 修复其他组织取消收藏的bug * [Update] 去掉debug信息 * [Update] 修改remote app get queryset * [Update] 修改remote app get queryset * [Update] 修改没有授权时显示情况 * [Bugfix] 修复组织管理员查看用户权限失败问题 * [Update] 优化forms assets queryset设置
48 lines
1.5 KiB
Python
48 lines
1.5 KiB
Python
# ~*~ coding: utf-8 ~*~
|
|
|
|
from rest_framework.views import APIView, Response
|
|
from django.views.generic.detail import SingleObjectMixin
|
|
|
|
from common.utils import get_logger
|
|
from common.permissions import IsOrgAdmin, IsOrgAdminOrAppUser
|
|
from orgs.mixins.api import OrgBulkModelViewSet
|
|
from ..models import Domain, Gateway
|
|
from .. import serializers
|
|
|
|
|
|
logger = get_logger(__file__)
|
|
__all__ = ['DomainViewSet', 'GatewayViewSet', "GatewayTestConnectionApi"]
|
|
|
|
|
|
class DomainViewSet(OrgBulkModelViewSet):
|
|
model = Domain
|
|
permission_classes = (IsOrgAdminOrAppUser,)
|
|
serializer_class = serializers.DomainSerializer
|
|
|
|
def get_serializer_class(self):
|
|
if self.request.query_params.get('gateway'):
|
|
return serializers.DomainWithGatewaySerializer
|
|
return super().get_serializer_class()
|
|
|
|
|
|
class GatewayViewSet(OrgBulkModelViewSet):
|
|
model = Gateway
|
|
filter_fields = ("domain__name", "name", "username", "ip", "domain")
|
|
search_fields = filter_fields
|
|
permission_classes = (IsOrgAdmin,)
|
|
serializer_class = serializers.GatewaySerializer
|
|
|
|
|
|
class GatewayTestConnectionApi(SingleObjectMixin, APIView):
|
|
permission_classes = (IsOrgAdmin,)
|
|
object = None
|
|
|
|
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
|
|
ok, e = self.object.test_connective(local_port=local_port)
|
|
if ok:
|
|
return Response("ok")
|
|
else:
|
|
return Response({"failed": e}, status=404)
|