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://172.16.220.3/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.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("预更改卡号已存在,请核对!") try: requests.post(url="http://172.16.220.3/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参数缺失")