# -*- coding: utf-8 -*-
"""
@File    : debugtalk.py
@Time    : 2023/2/22 15:07
@Author  : geekbing
@LastEditTime : -
@LastEditors : -
@Description : 驱动代码视图
"""

from django.core.exceptions import ObjectDoesNotExist
from django.utils.decorators import method_decorator
from django.utils import timezone
from rest_framework.response import Response
from rest_framework.viewsets import GenericViewSet

from lunarlink import models
from lunarlink import serializers
from lunarlink.utils import response
from lunarlink.utils.decorator import request_log
from lunarlink.utils.runner import DebugCode


class DebugTalkView(GenericViewSet):
    """
    DebugTalk update
    """

    serializer_class = serializers.DebugTalkSerializer
    queryset = models.Debugtalk.objects

    @method_decorator(request_log(level="INFO"))
    def debugtalk(self, request, pk):
        """
        获取debugtalk code
        """
        try:
            queryset = models.Debugtalk.objects.get(project__id=pk)
        except ObjectDoesNotExist:
            return Response(response.DEBUGTALK_NOT_EXISTS)

        serializer = self.get_serializer(queryset, many=False)

        return Response(serializer.data)

    @method_decorator(request_log(level="INFO"))
    def update(self, request):
        """
        编辑debugtalk.py代码并保存
        {
            id: int # debugtalk id
            code: str
        }
        """

        try:
            debugtalk_id = request.data["id"]
            debugtalk_code = request.data["code"]
        except KeyError:
            return Response(response.KEY_MISS)

        try:
            models.Debugtalk.objects.get(id=debugtalk_id)
            models.Debugtalk.objects.filter(id=debugtalk_id).update(
                code=debugtalk_code,
                updater=request.user.id,
                update_time=timezone.now(),
            )
        except ObjectDoesNotExist:
            return Response(response.DEBUGTALK_NOT_EXISTS)

        return Response(response.DEBUGTALK_UPDATE_SUCCESS)

    @method_decorator(request_log(level="INFO"))
    def run(self, request):
        """在线运行"""
        try:
            code = request.data["code"]
        except KeyError:
            return Response(response.KEY_MISS)

        debug = DebugCode(code)
        debug.run()
        resp = {"msg": debug.resp, "success": True, "code": "0001"}
        return Response(resp)