mirror of https://github.com/jumpserver/jumpserver
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
111 lines
3.2 KiB
111 lines
3.2 KiB
# -*- coding: utf-8 -*- |
|
# |
|
import os |
|
import uuid |
|
|
|
from django.core.cache import cache |
|
from django.views.decorators.csrf import csrf_exempt |
|
|
|
from rest_framework.views import APIView |
|
from rest_framework.response import Response |
|
from rest_framework import generics, serializers |
|
from rest_framework.viewsets import GenericViewSet |
|
|
|
from common.permissions import IsValidUser |
|
from .http import HttpResponseTemporaryRedirect |
|
from .const import KEY_CACHE_RESOURCES_ID |
|
from .utils import get_logger |
|
from .mixins import CommonApiMixin |
|
|
|
__all__ = [ |
|
'LogTailApi', 'ResourcesIDCacheApi', 'CommonGenericViewSet' |
|
] |
|
|
|
logger = get_logger(__file__) |
|
|
|
|
|
class OutputSerializer(serializers.Serializer): |
|
output = serializers.CharField() |
|
is_end = serializers.BooleanField() |
|
mark = serializers.CharField() |
|
|
|
|
|
class LogTailApi(generics.RetrieveAPIView): |
|
permission_classes = (IsValidUser,) |
|
buff_size = 1024 * 10 |
|
serializer_class = OutputSerializer |
|
end = False |
|
mark = '' |
|
log_path = '' |
|
|
|
def is_file_finish_write(self): |
|
return True |
|
|
|
def get_log_path(self): |
|
raise NotImplementedError() |
|
|
|
def get_no_file_message(self, request): |
|
return 'Not found the log' |
|
|
|
def filter_line(self, line): |
|
""" |
|
过滤行，可能替换一些信息 |
|
:param line: |
|
:return: |
|
""" |
|
return line |
|
|
|
def read_from_file(self): |
|
with open(self.log_path, 'rt', encoding='utf8') as f: |
|
offset = cache.get(self.mark, 0) |
|
f.seek(offset) |
|
data = f.read(self.buff_size).replace('\n', '\r\n') |
|
|
|
new_mark = str(uuid.uuid4()) |
|
cache.set(new_mark, f.tell(), 5) |
|
|
|
if data == '' and self.is_file_finish_write(): |
|
self.end = True |
|
_data = '' |
|
for line in data.split('\r\n'): |
|
new_line = self.filter_line(line) |
|
if line == '': |
|
continue |
|
_data += new_line + '\r\n' |
|
return _data, self.end, new_mark |
|
|
|
def get(self, request, *args, **kwargs): |
|
self.mark = request.query_params.get("mark") or str(uuid.uuid4()) |
|
self.log_path = self.get_log_path() |
|
|
|
if not self.log_path or not os.path.isfile(self.log_path): |
|
msg = self.get_no_file_message(self.request) |
|
return Response({"data": msg}, status=200) |
|
|
|
data, end, new_mark = self.read_from_file() |
|
return Response({"data": data, 'end': end, 'mark': new_mark}) |
|
|
|
|
|
class ResourcesIDCacheApi(APIView): |
|
permission_classes = (IsValidUser,) |
|
|
|
def post(self, request, *args, **kwargs): |
|
spm = str(uuid.uuid4()) |
|
resources = request.data.get('resources') |
|
if resources is not None: |
|
cache_key = KEY_CACHE_RESOURCES_ID.format(spm) |
|
cache.set(cache_key, resources, 300) |
|
return Response({'spm': spm}) |
|
|
|
|
|
@csrf_exempt |
|
def redirect_plural_name_api(request, *args, **kwargs): |
|
resource = kwargs.get("resource", "") |
|
org_full_path = request.get_full_path() |
|
full_path = org_full_path.replace(resource, resource+"s", 1) |
|
logger.debug("Redirect {} => {}".format(org_full_path, full_path)) |
|
return HttpResponseTemporaryRedirect(full_path) |
|
|
|
|
|
class CommonGenericViewSet(CommonApiMixin, GenericViewSet): |
|
pass
|
|
|