223 lines
7.6 KiB
Python
223 lines
7.6 KiB
Python
from apps.third.models import TDevice, Tlog
|
|
from apps.third.serializers import BindAreaSerializer, BltQuerySerializer, BltSerializer, LabelLocationSerializer, TDeviceSerializer, TlogSerializer
|
|
from apps.utils.viewsets import CustomGenericViewSet
|
|
from rest_framework.mixins import ListModelMixin, CreateModelMixin
|
|
from apps.third.clients import xxClient, dhClient, spClient
|
|
from apps.third.tapis import xxapis, dhapis, spapis
|
|
from rest_framework.response import Response
|
|
from rest_framework.serializers import Serializer
|
|
from rest_framework.decorators import action
|
|
from apps.am.models import Area
|
|
from rest_framework.views import APIView
|
|
from rest_framework.exceptions import ParseError
|
|
|
|
|
|
class BltViewSet(CustomGenericViewSet):
|
|
"""定位标签列表
|
|
|
|
定位标签列表
|
|
"""
|
|
queryset = TDevice.objects.filter(type=TDevice.DEVICE_BLT)
|
|
serializer_class = TDeviceSerializer
|
|
|
|
@action(methods=['get'], detail=False, perms_map={'get': '*'})
|
|
def count_bind(self, request):
|
|
"""统计绑定定位卡人/设备数
|
|
|
|
统计绑定定位卡人/设备数
|
|
"""
|
|
qs = self.queryset.exclude(employee=None)
|
|
ret = {}
|
|
ret['total'] = qs.count()
|
|
ret['count_employee'] = qs.filter(employee__type='employee').count()
|
|
ret['count_remployee'] = qs.filter(employee__type='remployee').count()
|
|
ret['count_visitor'] = qs.filter(employee__type='visitor').count()
|
|
return Response(ret)
|
|
|
|
@action(methods=['get'], detail=False, perms_map={'get': '*'})
|
|
def count_now(self, request):
|
|
"""统计在厂终端数
|
|
|
|
统计在厂终端数
|
|
"""
|
|
json = {
|
|
"pageNum": 1,
|
|
"numPerPage": 2000,
|
|
"online": "online"
|
|
}
|
|
_, res = xxClient.request(**xxapis['blt_list'], json=json)
|
|
ret = {}
|
|
ret['total_blt'] = res['totalCount']
|
|
blt_list = res['recordList']
|
|
macs = []
|
|
for i in blt_list:
|
|
macs.append(i['mac'])
|
|
qs = self.queryset.filter().exclude(employee=None, code__in=macs)
|
|
ret['total'] = qs.count()
|
|
ret['count_employee'] = qs.filter(employee__type='employee').count()
|
|
ret['count_remployee'] = qs.filter(employee__type='remployee').count()
|
|
ret['count_visitor'] = qs.filter(employee__type='visitor').count()
|
|
return Response(ret)
|
|
|
|
@action(methods=['post'], detail=False, perms_map={'post': '*'},
|
|
serializer_class=BltQuerySerializer)
|
|
def query(self, request):
|
|
"""在线标签列表信息
|
|
|
|
在线标签列表信息
|
|
"""
|
|
data = request.data
|
|
if data.get('area', None):
|
|
try:
|
|
railId = Area.objects.get(id=data['area']).third_info['xx_rail']['id']
|
|
except Exception:
|
|
raise ParseError('区域信息不正确')
|
|
json = {
|
|
"railId": railId,
|
|
"type": ""
|
|
}
|
|
_, res = xxClient.request(**xxapis['rail_ibeacon_list'], json=json)
|
|
else:
|
|
json = {
|
|
"pageNum": 1,
|
|
"numPerPage": 2000,
|
|
"online": "online"
|
|
}
|
|
_, res = xxClient.request(**xxapis['blt_list'], json=json)
|
|
blt_list = res['recordList']
|
|
macs = []
|
|
for i in blt_list:
|
|
macs.append(i['mac'])
|
|
qs = self.queryset.filter(code__in=macs).exclude(employee=None)
|
|
qs_data = list(BltSerializer(instance=qs, many=True).data)
|
|
qs_dict = {}
|
|
for i in qs_data:
|
|
qs_dict[i['code']] = i
|
|
for i in blt_list:
|
|
i['my_info'] = {}
|
|
if i['mac'] in qs_dict:
|
|
i['my_info'] = qs_dict[i['mac']]
|
|
return Response(blt_list)
|
|
|
|
|
|
class TDeviceViewSet(ListModelMixin, CustomGenericViewSet):
|
|
"""
|
|
三方设备接口
|
|
"""
|
|
queryset = TDevice.objects.all()
|
|
serializer_class = TDeviceSerializer
|
|
ordering = ['code', '-create_time']
|
|
filterset_fields = ['type']
|
|
|
|
@action(methods=['post'], detail=False, perms_map={'post': '*'},
|
|
serializer_class=Serializer)
|
|
def vchannel(self, request):
|
|
"""
|
|
视频通道列表
|
|
|
|
视频通道列表
|
|
"""
|
|
request.data.update({'channelTypeList': ["1"]})
|
|
_, res = dhClient.request(**dhapis['channel_list'], json=request.data)
|
|
codes = []
|
|
if res.get('pageData', None):
|
|
for i in res['pageData']:
|
|
codes.append(i['channelCode'])
|
|
tds_info = TDeviceSerializer(instance=TDevice.objects.filter(code__in=codes), many=True).data
|
|
for x in res['pageData']:
|
|
for y in tds_info:
|
|
if x['channelCode'] == y['code']:
|
|
x['my_info'] = y
|
|
return Response(res)
|
|
|
|
@action(methods=['post'], detail=False, perms_map={'post': '*'},
|
|
serializer_class=Serializer)
|
|
def speaker(self, request):
|
|
"""
|
|
喇叭列表
|
|
|
|
喇叭列表
|
|
"""
|
|
_, res = spClient.request(**spapis['device_list'], params=request.data)
|
|
return Response(res)
|
|
|
|
@action(methods=['post'], detail=False, perms_map={'post': '*'},
|
|
serializer_class=Serializer)
|
|
def dchannel(self, request):
|
|
"""
|
|
闸机通道列表
|
|
|
|
闸机通道列表
|
|
"""
|
|
request.data.update({'channelTypeList': ["7"]})
|
|
_, res = dhClient.request(**dhapis['channel_list'], json=request.data)
|
|
return Response(res)
|
|
|
|
@action(methods=['post'], detail=False, perms_map={'post': 'tdevice:label_location'},
|
|
serializer_class=LabelLocationSerializer)
|
|
def label_location(self, request):
|
|
"""
|
|
标注坐标位置
|
|
|
|
标注坐标位置
|
|
"""
|
|
serializer = self.get_serializer(data=request.data)
|
|
serializer.is_valid(raise_exception=True)
|
|
vdata = serializer.validated_data
|
|
td = TDevice.objects.filter(code=vdata['code']).first()
|
|
if td:
|
|
td.location = vdata['location']
|
|
td.type = vdata['type']
|
|
td.area = vdata['area']
|
|
td.name = vdata['name']
|
|
td.update_by = request.user
|
|
td.save()
|
|
td.areas.clear()
|
|
for i in vdata['areas']:
|
|
td.areas.add(i)
|
|
else:
|
|
td = TDevice()
|
|
td.type = vdata['type']
|
|
td.code = vdata['code']
|
|
td.location = vdata['location']
|
|
td.area = vdata['area']
|
|
td.name = vdata['name']
|
|
td.create_by = request.user
|
|
td.save()
|
|
td.areas.clear()
|
|
for i in vdata['areas']:
|
|
td.areas.add(i)
|
|
return Response()
|
|
|
|
@action(methods=['post'], detail=False, perms_map={'post': 'tdevice:bind_area'},
|
|
serializer_class=BindAreaSerializer)
|
|
def bind_area(self, request):
|
|
"""
|
|
绑定所在区域
|
|
|
|
绑定所在区域
|
|
"""
|
|
serializer = self.get_serializer(data=request.data)
|
|
serializer.is_valid(raise_exception=True)
|
|
vdata = serializer.vdata
|
|
area = Area.objects.get(id=vdata['area'])
|
|
for i in vdata['codes']:
|
|
td = TDevice.objects.filter(code=i['code']).first()
|
|
if td:
|
|
td.area = area
|
|
td.update_by = request.user
|
|
td.save()
|
|
else:
|
|
td = TDevice()
|
|
td.type = TDevice.DEVICE_VCHANNEL
|
|
td.code = i['code']
|
|
td.area = area
|
|
td.create_by = request.user
|
|
td.save()
|
|
return Response()
|
|
|
|
|
|
class TlogViewSet(ListModelMixin, CustomGenericViewSet):
|
|
queryset = Tlog.objects.all()
|
|
serializer_class = TlogSerializer
|