bigdataAPI/apps/IOTCard/views.py

from rest_framework import status
from rest_framework import viewsets
from rest_framework.response import Response
from django.http import HttpResponse
from rest_framework.decorators import action
from rest_framework.serializers import ValidationError

import requests
import uuid
import os
import datetime
from io import BytesIO
import pandas as pd
from django.utils.encoding import escape_uri_path

from .serializers import PlatSimInfoSerializer
from .models import PlatSimInfo
from utils.paginations import CustomPagination
from utils.utils import DeviceInfoUtils, GetSIMInfo


# Create your views here.
def time_dif(checkdatetime):
    # 获取传入时间和当前时间差
    nowdatetime = datetime.datetime.now()
    checkdatetime = datetime.datetime.strptime(checkdatetime, "%Y-%m-%d %H:%M:%S")

    timedif = nowdatetime - checkdatetime
    return timedif.days


class PlatformIOTCardViewSet(viewsets.ModelViewSet):
    serializer_class = PlatSimInfoSerializer
    pagination_class = CustomPagination
    queryset = PlatSimInfo.objects.all().order_by("-id")

    def list(self, request, *args, **kwargs):
        device_type = request.query_params.get("device_type")
        deviceId = request.query_params.get("deviceId")
        simId = request.query_params.get("simId")
        expire_start_time = request.query_params.get("expire_start_time")
        expire_end_time = request.query_params.get("expire_end_time")
        queryset = self.get_queryset()
        if device_type:
            queryset = queryset.filter(device_type=device_type)
        if deviceId:
            if len(deviceId.strip()) < 4:
                raise ValidationError("设备号长度不能小于4位")
            queryset = queryset.filter(deviceId__icontains=deviceId.strip())
        if simId:
            if len(simId.strip()) < 4:
                raise ValidationError("物联网卡号长度不能小于4位")
            queryset = queryset.filter(simId__icontains=simId.strip())
        if expire_start_time:
            queryset = queryset.filter(expiry_date__range=(expire_start_time, expire_end_time))
        page = self.paginate_queryset(queryset)
        serializer = self.get_serializer(page, many=True)
        return self.get_paginated_response(serializer.data)

    def create(self, request, *args, **kwargs):
        device_id = request.data.get("device_id").strip()
        simId = request.data.get("simId").strip()
        if len(simId) > 20 or len(simId) < 19:
            raise ValidationError("物联网卡长度介于19-20位")

        if self.get_queryset().filter(simId=simId).exists():
            raise ValidationError("该物联网卡号已存在,请核查!")

        if self.get_queryset().filter(deviceId=device_id, input_type=2).exists():
            raise ValidationError("该设备已存在手动录入类型,请核查!")

        device_info_list = DeviceInfoUtils().get_equip_list(
            d_id=device_id,
            isfullId=1
        )
        if device_info_list:
            device_info = device_info_list[0]
            device_type = device_info["device_type"]
            platform = device_info["plat"]
        else:
            device_type = ""
            platform = ""
        if device_type not in ["测报灯", "孢子仪"]:
            raise ValidationError("不支持该类型设备手动录入物联网卡！")
        if platform == "四情平台":
            platformid = 1
        elif platform == "大数据平台":
            platformid = 2
        else:
            raise ValidationError("平台不存在该设备ID,请核查！")
        requests.post(url="http://www.yfzhwlw.com/equip_simiccid", data={"e_id": device_id, "iccid": simId}, timeout=10)
        requests.post(url="http://127.0.0.1:8002/api/api_gateway?method=forecast.send_control.device_sim",
                      data={"device_id": device_id, "iccid": simId, "type": "change"}, timeout=10)
        sim_operators, account_status, active_date, data_plan, data_usage, data_balance, expiry_date = GetSIMInfo(
            simId).get_sim_info()
        request_data = {
            "simId": simId,
            "sim_operators": sim_operators,
            "input_type": 2,
            "deviceId": device_id,
            "device_type": device_type,
            "platform": platformid,
            "account_status": account_status,
            "active_date": active_date,
            "data_plan": data_plan,
            "data_usage": data_usage,
            "data_balance": data_balance,
            "expiry_date": expiry_date
        }
        serializer = self.get_serializer(data=request_data)
        serializer.is_valid(raise_exception=True)
        self.perform_create(serializer)
        headers = self.get_success_headers(serializer.data)
        return Response(serializer.data, status=status.HTTP_201_CREATED, headers=headers)

    def update(self, request, pk, *args, **kwargs):
        req = request.data.get("req")
        instance = self.get_object()
        if req == "upgrade":
            iccid = instance.simId
            sim_operators, account_status, active_date, data_plan, data_usage, data_balance, expiry_date = GetSIMInfo(
                iccid).get_sim_info()
            update_data = {
                "sim_operators": sim_operators,
                "account_status": account_status,
                "active_date": active_date,
                "data_plan": data_plan,
                "data_usage": data_usage,
                "data_balance": data_balance,
                "expiry_date": expiry_date
            }
        elif req == "change":
            deviceId = instance.deviceId
            if instance.input_type == 1:
                raise ValidationError("设备读取物联网卡禁止更改")
            simId = request.POST.get("simId").strip()
            if len(simId) > 20 or len(simId) < 19:
                raise ValidationError("物联网卡长度介于19-20位")
            if self.get_queryset().filter(simId=simId).exists():
                raise ValidationError("预更改卡号已存在,请核对!")
            try:
                requests.post(url="http://www.yfzhwlw.com/equip_simiccid", data={"e_id": deviceId, "iccid": simId},
                              timeout=10)
                requests.post(url="http://127.0.0.1:8002/api/api_gateway?method=forecast.send_control.device_sim",
                              data={"device_id": deviceId, "iccid": simId, "type": "change"}, timeout=10)
            except:
                pass
            sim_operators, account_status, active_date, data_plan, data_usage, data_balance, expiry_date = GetSIMInfo(
                simId).get_sim_info()
            update_data = {
                "simId": simId,
                "sim_operators": sim_operators,
                "account_status": account_status,
                "active_date": active_date,
                "data_plan": data_plan,
                "data_usage": data_usage,
                "data_balance": data_balance,
                "expiry_date": expiry_date
            }
        else:
            raise ValidationError("未识别req参数")
        serializer = self.get_serializer(instance, data=update_data, partial=True)
        serializer.is_valid(raise_exception=True)
        self.perform_update(serializer)
        return Response(serializer.data)

    @action(methods=['get'], detail=False, url_path='get_device_type', url_name='get_device_type')
    def get_device_type(self, request, *args, **kwargs):
        queryset = PlatSimInfo.objects.raw("SELECT id,device_type FROM plat_sim_info GROUP BY device_type;")
        data = []
        for i in queryset:
            data.append(i.device_type)
        return Response(data)

    @action(methods=['get'], detail=False, url_path='down_excel', url_name='down_excel')
    def down_excel(self, request, *args, **kwargs):
        filename = request.query_params.get("filename")
        if filename == "发货表示例模板.xlsx":
            save_data = dict(设备ID=[])
        elif filename == "合宙续存流量卡筛选模板.xlsx":
            save_data = dict(ICCID=[], 续费价格=[], 当月消耗流量=[])
        elif filename == "SIMBoss续存流量卡筛选模板.xlsx":
            save_data = dict(ICCID=[], 筛查月份1=[], 筛查月份2=[], 筛查月份3=[], 筛查月份4=[], 筛查月份5=[],
                             筛查月份6=[], 筛查月份7=[], 筛查月份8=[], 筛查月份9=[])
        else:
            now_dir = os.path.dirname(__file__)
            file_path = os.path.join(now_dir, "excel_folder", filename)
            if os.path.exists(file_path) == False:
                raise ValidationError("文件不存在,请确认文件名")
            else:
                with open(file_path, 'rb') as f:
                    data = f.read()
            response = HttpResponse(data, content_type="application/vnd.ms-excel")
            response['Content-Disposition'] = 'attachment;filename={}'.format(escape_uri_path(filename))
            os.remove(file_path)
            return response
        df = pd.DataFrame(data=save_data)
        output = BytesIO()
        df.to_excel(output, index=False)
        output.seek(0)
        response = HttpResponse(content_type='application/vnd.ms-excel')
        response['Content-Disposition'] = 'attachment;filename={}'.format(escape_uri_path(filename))
        response.write(output.getvalue())
        return response

    @action(methods=['post'], detail=False, url_path='delivery_card_filter', url_name='delivery_card_filter')
    def card_delivery_filter(self, request, *args, **kwargs):
        file = request.FILES.get('file')
        file_dataframe = pd.read_excel(file)
        save_data = dict(deviceid=[], fulldeviceid=[], device_type=[], is_recharge=[], backword=[])

        for deviceId in file_dataframe["设备ID"]:
            if isinstance(deviceId, int):
                deviceId = str(int(deviceId))
            else:
                deviceId = deviceId.strip()
            sim_queryset = self.get_queryset().filter(deviceId__endswith=deviceId, input_type=1)
            if sim_queryset.count() == 0:
                fulldeviceid = "未知"
                device_type = "未知"
                is_recharge = "未知"
                backword = "根据表内设备号未找到记录"
            elif sim_queryset.count() == 1:
                sim_obj = sim_queryset.first()
                fulldeviceid = sim_obj.deviceId
                device_type = sim_obj.device_type
                if sim_obj.expiry_date == "未知":
                    is_recharge = "未知"
                else:
                    time_difference = time_dif(sim_obj.expiry_date)
                    if time_difference >= -180:
                        is_recharge = "是"
                    else:
                        is_recharge = "否"
                backword = "{}".format(sim_obj.simId)
                if sim_obj.device_type == "孢子仪" or sim_obj.device_type == "测报灯":
                    sim2_queryset = self.get_queryset().filter(deviceId=fulldeviceid, input_type=2).order_by("-id")
                    if sim2_queryset:
                        sim2_obj = sim2_queryset[0]
                        if sim2_obj.expiry_date == "未知":
                            is_recharge = "卡1" + is_recharge + ",卡2未知"
                        else:
                            time_difference = time_dif(sim2_obj.expiry_date)
                            if time_difference >= -180:
                                is_recharge = "卡1" + is_recharge + ",卡2是"
                            else:
                                is_recharge = "卡1" + is_recharge + ",卡2否"
                        backword = "卡1:{};卡2:{}".format(sim_obj.simId, sim2_obj.simId)
                    else:
                        print("卡二不存在")

            else:
                fulldeviceid = "未知"
                device_type = "未知"
                is_recharge = "未知"
                backword = "根据表内设备号存在多条记录"
            save_data["deviceid"].append(deviceId)
            save_data["fulldeviceid"].append(fulldeviceid)
            save_data["is_recharge"].append(is_recharge)
            save_data["device_type"].append(device_type)
            save_data["backword"].append(backword)

        df = pd.DataFrame(data=save_data)
        save_filename = str(uuid.uuid4()) + ".xlsx"
        now_dir = os.path.dirname(__file__)
        save_path = os.path.join(now_dir, "excel_folder", save_filename)
        df.to_excel(save_path, index=False, header=["设备ID", "完整设备号(自行判断)", "设备类型", "是否需要充值", "说明备注"])
        return Response(save_filename)

    @action(methods=['post'], detail=False, url_path='xucun_card_filter', url_name='xucun_card_filter')
    def xucun_card_filter(self, request, *args, **kwargs):
        simtype = request.data.get("simtype")
        renew_file = request.FILES.get("file")
        file_dataframe = pd.read_excel(renew_file)
        if simtype == "1":  # 合宙卡
            save_data = dict(ICCID=[], 续费价格=[], 是否需要充值=[], 备注=[])
            sim_list = list(file_dataframe["ICCID"])
            price_list = list(file_dataframe["续费价格"])
            usage_list = list(file_dataframe["当月消耗流量"])
            for index, usage in enumerate(usage_list):
                save_data["ICCID"].append(str(sim_list[index]))
                save_data["续费价格"].append(str(price_list[index]))
                try:
                    usage_value = float(usage)
                except:
                    usage_value = ""
                if usage_value == 0 or usage_value == "":
                    card_queryset = self.get_queryset().filter(simId=str(sim_list[index]))
                    if card_queryset.exists():
                        card_obj = card_queryset.first()
                        deviceId = card_obj.deviceId
                        device_info_list = DeviceInfoUtils().get_equip_list(d_id=deviceId, isfullId=1)
                        if device_info_list:
                            device_info = device_info_list[0]
                            uptime = device_info["uptime"]
                        else:
                            uptime = "1970-01-01 00:00:00"
                        time_diff_days = time_dif(uptime)
                        if time_diff_days > 270:
                            save_data["是否需要充值"].append("否")
                            save_data["备注"].append("对应设备号:{}".format(deviceId))
                        else:
                            save_data["是否需要充值"].append("是")
                            save_data["备注"].append("对应设备号:{}".format(deviceId))
                    else:
                        save_data["是否需要充值"].append("未知")
                        save_data["备注"].append("未找到该卡记录,请确认是否未录入平台")
                else:
                    save_data["是否需要充值"].append("是")
                    save_data["备注"].append("当月存在流量使用")
        else:  # SIMboss
            save_data = dict(ICCID=[], 是否需要充值=[])
            for index, sim in enumerate(file_dataframe["ICCID"]):
                save_data["ICCID"].append(str(sim))
                status = 0
                for i in range(9):
                    if float(file_dataframe["筛查月份{}".format(str(i + 1))][index]) > 1:
                        status = 1
                        break
                if status == 1:
                    save_data["是否需要充值"].append("是")
                else:
                    save_data["是否需要充值"].append("否")

        df = pd.DataFrame(data=save_data)
        save_filename = str(uuid.uuid4()) + ".xlsx"
        now_dir = os.path.dirname(__file__)
        save_path = os.path.join(now_dir, "excel_folder", save_filename)
        df.to_excel(save_path, index=False)
        return Response(save_filename)

    @action(methods=['post'], detail=False, url_path='inquiries', url_name='inquiries')
    def inquiries(self, request):
        iccid = request.data.get("iccid")
        if iccid:
            sim_operators, account_status, active_date, data_plan, data_usage, data_balance, expiry_date = GetSIMInfo(
                iccid).get_sim_info()
            data = {
                "simId": iccid,
                "sim_operators": "合宙" if sim_operators == 1 else "SIMBoss" if sim_operators == 2 else "未知",
                "account_status": account_status,
                "active_date": active_date,
                "data_plan": data_plan,
                "data_usage": data_usage,
                "data_balance": data_balance,
                "expiry_date": expiry_date
            }
            return Response(data)
        else:
            raise ValidationError("iccid参数缺失")