mirror of https://github.com/jumpserver/jumpserver
Finish terminal app
parent
303659cb0e
commit
7513474366
|
@ -2,7 +2,10 @@
|
|||
#
|
||||
|
||||
from rest_framework.generics import ListCreateAPIView
|
||||
from rest_framework.views import APIView, Response
|
||||
from rest_framework.permissions import AllowAny
|
||||
|
||||
from common.utils import unsign, get_object_or_none
|
||||
from .models import Terminal
|
||||
from .serializers import TerminalSerializer
|
||||
|
||||
|
@ -10,5 +13,43 @@ from .serializers import TerminalSerializer
|
|||
class TerminalApi(ListCreateAPIView):
|
||||
queryset = Terminal.objects.all()
|
||||
serializer_class = TerminalSerializer
|
||||
permission_classes = (AllowAny,)
|
||||
|
||||
def post(self, request, *args, **kwargs):
|
||||
name = unsign(request.data.get('name', ''))
|
||||
if name:
|
||||
terminal = get_object_or_none(Terminal, name=name)
|
||||
if terminal:
|
||||
if terminal.is_accepted and terminal.is_active:
|
||||
return Response(data={'data': {'name': name, 'ip': terminal.ip},
|
||||
'msg': 'Success'},
|
||||
status=200)
|
||||
else:
|
||||
return Response(data={'data': {'name': name, 'ip': terminal.ip},
|
||||
'msg': 'Need admin accept or active it'},
|
||||
status=204)
|
||||
|
||||
else:
|
||||
ip = request.META.get('X-Real-IP') or request.META.get('REMOTE_ADDR')
|
||||
terminal = Terminal.objects.create(name=name, ip=ip)
|
||||
return Response(data={'data': {'name': name, 'ip': terminal.ip},
|
||||
'msg': 'Need admin accept or active it'},
|
||||
status=204)
|
||||
else:
|
||||
return Response(data={'error': 'Secrete key invalid'}, status=401)
|
||||
|
||||
|
||||
# class TerminalRegister(APIView):
|
||||
# def post(self, request, format='json'):
|
||||
# return Response(data={'hello': request.META.get('REMOTE_ADDR')})
|
||||
# name = unsign(request.data.get('name', ''))
|
||||
# if name:
|
||||
# terminal = get_object_or_none(Terminal, name=name)
|
||||
# if terminal:
|
||||
# return Response(data={'name': name, 'ip': terminal.ip}, status=200)
|
||||
# else:
|
||||
# ip = request.Meta.get('X-Real-IP')
|
||||
# Terminal.objects.create(name=name, ip=request.META.host)
|
||||
|
||||
|
||||
|
||||
|
|
|
@ -11,7 +11,7 @@ class Terminal(models.Model):
|
|||
('S', 'SSH Terminal'),
|
||||
('WT', 'Web Terminal')
|
||||
)
|
||||
name = models.CharField(max_length=30, verbose_name=_('Name'))
|
||||
name = models.CharField(max_length=30, unique=True, verbose_name=_('Name'))
|
||||
ip = models.GenericIPAddressField(verbose_name=_('From ip'))
|
||||
is_active = models.BooleanField(default=False, verbose_name=_('Is active'))
|
||||
is_bound_ip = models.BooleanField(default=False, verbose_name=_('Is bound ip'))
|
||||
|
|
|
@ -9,7 +9,7 @@ from .models import Terminal
|
|||
class TerminalSerializer(serializers.ModelSerializer):
|
||||
class Meta:
|
||||
model = Terminal
|
||||
fields = ['name', 'ip', 'type', 'url', 'comment', 'is_active',
|
||||
fields = ['name', 'ip', 'type', 'url', 'comment', 'is_active', 'is_accepted',
|
||||
'get_type_display']
|
||||
|
||||
|
||||
|
|
Loading…
Reference in New Issue