import json from rest_framework.exceptions import ParseError, APIException from apps.ecm.service import EcmService from apps.hrm.services import HrmService from apps.third.tapis import dhapis, xxapis, spapis from apps.third.erros import TAPI_CODE_WRONG from apps.third.clients import dhClient, spClient, xxClient from apps.utils.mixins import MyLoggingMixin from rest_framework.response import Response from rest_framework.views import APIView from rest_framework.permissions import IsAuthenticated, IsAdminUser from apps.utils.viewsets import CustomGenericViewSet from rest_framework.mixins import CreateModelMixin from rest_framework.decorators import action from apps.third.serializers import RequestCommonSerializer from rest_framework import serializers import stomp from django.conf import settings # Create your views here. class DahuaTestView(MyLoggingMixin, APIView): """ 大华测试接口 """ permission_classes = [IsAuthenticated] def get(self, request, *args, **kwargs): _, res = dhClient.request( url='/evo-apigw/evo-brm/1.0.0/person/subsystem/{}'.format(2059335), method='get') return Response(res) class SpTestView(APIView): """ 音响测试接口 """ permission_classes = [IsAuthenticated] def get(self, request, *args, **kwargs): """ 音响测试接口 音响测试接口 """ params = {"page": 1, "pageSize": 10000} _, res = spClient.request(**spapis['device_list'], params=params) return Response(res) class XxTestView(APIView): """ 寻息测试接口 """ permission_classes = [IsAuthenticated] def get(self, request, *args, **kwargs): ok, res = xxClient.request( url='/api/application/build/buildListV2', json={}) if ok == 'success': return Response(res) elif ok == 'fail': raise ParseError(**res) else: raise APIException(**res) # 寻息事件订阅 class XxListener(stomp.ConnectionListener): def on_error(self, frame): print('received an error "%s"' % frame.body) def on_message(self, frame): data = json.loads(frame.body) print('received a message "%s"' % frame.body) # if settings.XX_ENABLED: # c = stomp.Connection([(settings.XX_MQ_HOST, settings.XX_MQ_PORT)]) # c.set_listener('', XxListener()) # c.connect(settings.XX_USERNAME, settings.XX_LICENCE) # c.subscribe(settings.XX_QUEUE, id='') class XxCommonViewSet(CreateModelMixin, CustomGenericViewSet): perms_map = {'post': '*'} serializer_class = RequestCommonSerializer def create(self, request, *args, **kwargs): """ 寻息通用调用接口 寻息通用调用接口 """ serializer = self.get_serializer(data=request.data) serializer.is_valid(raise_exception=True) vdata = serializer.validated_data if vdata.get('code', ''): xxapi = xxapis.get(vdata['code'], None) if xxapi is None: raise ParseError(**TAPI_CODE_WRONG) vdata['url'] = xxapi['url'] vdata['method'] = xxapi['method'] _, res = xxClient.request( url=vdata['url'], method=vdata.get('method', 'post'), params=vdata.get('params', {}), json=vdata.get('json', {})) return Response(res) @action(methods=['get'], detail=False, permission_classes=[IsAuthenticated]) def codes(self, request, pk=None): """获取请求短标识 获取请求短标识 """ return Response(xxapis) class DhCommonViewSet(CreateModelMixin, CustomGenericViewSet): perms_map = {'post': '*'} serializer_class = RequestCommonSerializer def create(self, request, *args, **kwargs): """ 大华通用调用接口 大华通用调用接口 """ serializer = self.get_serializer(data=request.data) serializer.is_valid(raise_exception=True) vdata = serializer.validated_data if vdata.get('code', ''): xxapi = xxapis.get(vdata['code'], None) if xxapi is None: raise ParseError(**TAPI_CODE_WRONG) vdata['url'] = xxapi['url'] vdata['method'] = xxapi['method'] _, res = dhClient.request( url=vdata['url'], method=vdata.get('method', 'post'), params=vdata.get('params', {}), json=vdata.get('json', {})) return Response(res) @action(methods=['get'], detail=False, permission_classes=[IsAuthenticated]) def codes(self, request, pk=None): """获取请求短标识 获取请求短标识 """ return Response(dhapis) @action(methods=['post'], detail=False, permission_classes=[IsAdminUser], serializer_class=serializers.Serializer) def subscribe(self, request, pk=None): """事件订阅 事件订阅 """ json_data = { "param": { "monitors": [ { "monitor": settings.BASE_URL + '/api/third/dahua/mq/', "monitorType": "url", "events": [ # { # "category": "business", # }, { "category": "alarm", }, { "category": "state", } ] } ], "subsystem": { "subsystemType": 0, "name": "10.0.11.101_8000", "magic": "10.0.11.101_8000" } } } dhClient.request(**dhapis['mq_subscribe'], json=json_data) return Response() @action(methods=['delete'], detail=False, permission_classes=[IsAdminUser], serializer_class=serializers.Serializer) def unsubscribe(self, request, pk=None): """取消事件订阅 取消事件订阅 """ dhClient.request(**dhapis['mq_unsubscribe'], params={'name': '10.0.11.101_8000'}) return Response() @action(methods=['post'], detail=False, authentication_classes=[], permission_classes=[], serializer_class=serializers.Serializer) def mq(self, request, pk=None): """大华事件处理 大华事件处理 """ data = json.loads(request.body) method = data['method'] category = data['category'] subsystem = data.get('subsystem', None) print(data) # info = data.get('info', {}) if method == 'department.update': pass elif method == 'person.update': pass elif method == 'alarm.msg' and subsystem == 'evo-accesscontrol': """ 刷卡事件 """ HrmService.swipe(data=data) elif category == 'alarm' and subsystem == 'admin': """ 视频报警 """ EcmService.dispatch_dahua_event(data=data) return Response()