Lims-back2/TestLaboratory/apps/software/views/versionview.py

import datetime

from django.db.models import Q
from rest_framework.response import Response
from rest_framework.views import APIView
from django.http import HttpResponse

from TestLaboratory.settings import SOFTWARE_ROOT, CASE_FILE_EXPORT
from apps.file.models import FileManager
from apps.log.models import get_log, gen_log
from apps.software.models import Software, Version
from apps.task.models import TestCase
from apps.user.middleware.rolecontrol import RoleControl
from util_xlsx import write_to_xlsx
from utils.util_add_id import gen_next_id

import os

from utils.util_file.util_fileio import write_file


class VersionView(APIView):
    authentication_classes = []

    # create version
    @staticmethod
    @RoleControl
    def post(request, software_id, *args, **kwargs):
        version_num = request.POST.get('version')
        file = request.FILES.get('file')

        # step1:判断版本所属软件是否存在
        software = Software.objects.filter(id=software_id, delete=False)
        if not software:
            return HttpResponse(status=404, content='软件已删除或不存在')
        software = software[0]

        # step2:判断版本是否存在
        software_version = software.version_set.filter(number=version_num, delete=False)
        if software_version:
            return HttpResponse(status=404, content='软件版本已存在，请修改版本号后重新上传')

        # step3:判断文件是否上传
        if not file:
            return HttpResponse(status=404, content='软件上传失败')

        # step4：创建版本id
        id_version = None
        if not Version.objects.order_by('create_time').last():
            id_version = "TestLaboratory_V1_Version_1"
        else:
            id_version = gen_next_id(Version.objects.order_by('create_time').last().id)

        # step5：设置创建时间和修改时间
        create_time = update_time = datetime.datetime.now()

        # step6:上传文件：
        file_dir = SOFTWARE_ROOT + software.id + "/" + id_version + "/" + str(create_time)
        if not os.path.exists(file_dir):
            os.makedirs(file_dir)
        file_path = file_dir + "/" + file.name
        with open(file_path, 'wb') as f:
            f.write(file.read())
        f.close()

        # step7：创建文件对象
        id_file = None
        if not FileManager.objects.last():
            id_file = "TestLaboratory_V1_File_1"
        else:
            id_file = gen_next_id(FileManager.objects.order_by('create_time').last().id)

        try:
            # step5：创建版本
            version = Version.objects.create(id=id_version, number=version_num, software=software,
                                             version_file=id_file,
                                             create_time=create_time,
                                             update_time=update_time)
        except:
            return HttpResponse(content='软件版本创建失败', status=500)
        try:
            file_new = FileManager.objects.create(id=id_file, category="software",
                                                  path=software.id + "/" + version.id + "/" +
                                                       str(create_time) + "/" + file.name,
                                                  create_time=create_time, update_time=update_time)
        except:
            return HttpResponse(content='文件数据创建失败', status=500)
        executor, action, method = get_log(request)
        gen_log(action, "目标版本", software.name+"-"+version.number, method, executor)
        return Response({
            'software_name': software.name,
            'software_id': software.id,
            'id': version.id,
            'number': version.number,
            'file_id': file_new.id,
            'file_name': file_new.path.split('/')[-1],
            'create_time': version.create_time,
            'update_time': version.update_time,
        })

    # edit version
    # 版本号不能修改，本质上只能修改上传的文件
    @staticmethod
    @RoleControl
    def put(request, software_id, *args, **kwargs):
        file = request.FILES.get('file')
        version_number = request.POST.get('version_number')

        # 判断软件是否存在
        software = Software.objects.filter(id=software_id, delete=False)
        if not software:
            return HttpResponse(status=404, content='软件已删除或不存在')
        software = software[0]

        # 判断版本是否存在
        software_version = software.version_set.filter(number=version_number, delete=False)
        if not software_version:
            return HttpResponse(status=404, content='软件版本不存在或已删除，请重新选择需要编辑的软件版本')
        software_version = software_version[0]

        # 删除原版本文件
        version_fid = software_version.version_file
        version_file = FileManager.objects.filter(id=version_fid)
        if not version_file:
            return HttpResponse(status=500, content='软件不存在，无法编辑')
        version_file = version_file[0]
        version_file.delete = True
        version_file.save()

        # step5：设置创建时间和修改时间
        create_time = update_time = datetime.datetime.now()

        # step6:上传文件：
        write_file(file, SOFTWARE_ROOT + software.id + "/" + software_version.id + "/" + str(create_time))

        # step7：创建文件对象
        id_file = None
        if not FileManager.objects.last():
            id_file = "TestLaboratory_V1_File_1"
        else:
            id_file = gen_next_id(FileManager.objects.order_by('create_time').last().id)
        software_version.version_file = id_file
        software_version.save()

        try:
            file_new = FileManager.objects.create(id=id_file, category="software",
                                                  path=software.id + "/" + software_version.id + "/" +
                                                       str(create_time) + "/" + file.name,
                                                  create_time=create_time,
                                                  update_time=update_time)
        except:
            return HttpResponse(content='软件版本编辑失败，请重试', status=500)
        executor, action, method = get_log(request)
        gen_log(action, "目标版本", software.name + "-" + software_version.number, method, executor)
        return HttpResponse(status=200, content='修改已保存')

    # delete version
    @staticmethod
    @RoleControl
    def delete(request, software_id, *args, **kwargs):
        version_number = request.GET.get('version_number')
        software = Software.objects.filter(id=software_id, delete=False)
        if not software:
            return HttpResponse(content='软件已删除或不存在', status=500)
        software = software[0]

        versions = len(list(software.version_set.filter(delete=False)))
        if versions == 0:
            return HttpResponse(content='被测软件无版本信息，请检查错误', status=500)
        elif versions == 1:
            version = software.version_set.filter(number=version_number, delete=False)[0]
            version_fid = version.version_file
            version_file = FileManager.objects.filter(id=version_fid)[0]
            software.delete = True
            version.delete = True
            version_file.delete = True

            # 删除测试计划
            plan_set = software.testplan_set.filter(delete=False)
            if plan_set:
                for plan in plan_set:
                    task_set = plan.testtask_set.filter(delete=False)
                    if task_set:
                        for task in task_set:
                            task.delete = True
                            task.save()
                    plan.delete = True
                    plan.save()

            software.save()
            version.save()
            version_file.save()
            return HttpResponse(status=200, content='软件及版本信息删除完成')
        else:
            version = software.version_set.filter(number=version_number, delete=False)[0]
            version_fid = version.version_file
            version_file = FileManager.objects.filter(id=version_fid)[0]
            version.delete = True
            version_file.delete = True

            # 删除测试计划
            plan_set = software.testplan_set.filter(delete=False)
            if plan_set:
                for plan in plan_set:
                    task_set = plan.testtask_set.filter(delete=False)
                    if task_set:
                        for task in task_set:
                            task.delete = True
                            task.save()
                    plan.delete = True
                    plan.save()

            version.save()
            version_file.save()
            executor, action, method = get_log(request)
            gen_log(action, "目标版本", software.name + "-" + version.number, method, executor)
            return HttpResponse(status=200, content='删除完成')

    @staticmethod
    # @RoleControl
    def get(request, software_id, *args, **kwargs):
        export_type = request.GET.get('type')
        version_number = request.GET.get('version_number')
        software = Software.objects.filter(id=software_id, delete=False)
        if not software:
            return HttpResponse(content='软件已删除或不存在', status=500)
        software = software[0]

        version = software.version_set.filter(number=version_number, delete=False)
        if not version:
            return HttpResponse(status=404, content='软件版本已删除或不存在')
        version = version[0]

        cases = None
        file_name = None
        if export_type == '0':
            cases = TestCase.objects.filter(software=software, version=version_number, delete=False)
            file_name = '全部测试用例.xls'
        elif export_type == '1':
            cases = TestCase.objects.filter(~Q(state=0), software=software, version=version_number, delete=False)
            file_name = '已执行测试用例.xls'
        else:
            cases = TestCase.objects.filter(software=software, version=version_number, state=2, delete=False)
            file_name = '不合格测试用例.xls'

        if not cases:
            return HttpResponse(status=404, content='无符合要求的测试用例')

        update_time = datetime.datetime.now()
        file_dir = CASE_FILE_EXPORT + software_id + "/" + str(update_time)
        if not os.path.exists(file_dir):
            os.makedirs(file_dir)
        file_path = file_dir + "/" + file_name

        cases_info = [['被测软件ID：', software_id],
                      ['被测软件名：', software.name],
                      ['被测软件版本', version_number],
                      ['编号', '名称', '前置条件', '优先级', '测试环境', '测试类型', '测试步骤', '预期结果', '实际结果', '状态',
                       '编写人', '评审员', '执行人', '备注']]

        for case in cases:
            state = '未执行'
            if case.state == 1:
                state = '执行成功'
            elif case.state == 2:
                state = '执行失败'
            cases_info.append([case.id_in_task, case.name, case.requisite, case.priority, case.environment,
                               case.type, case.process, case.expected_result, case.actual_result, state,
                               case.writer, case.assessor, case.executor, case.remark])

        sio = write_to_xlsx(file_path, cases_info)
        sio.seek(0)
        response = HttpResponse(sio.getvalue(), content_type='multipart/form-data')
        response['Content-Disposition'] = 'attachment; filename="{0}"'.format(file_name).encode('utf-8')
        response.write(sio.getvalue())
        # executor, action, method = get_log(request)
        # gen_log(action, "目标版本", software.name + "-" + version.number, method, executor)
        return response