From 1925e3e5d61e588dec4156280aa4c65dc97c94ad Mon Sep 17 00:00:00 2001
From: caoqianming <caoqianming@foxmail.com>
Date: Thu, 27 Feb 2025 11:34:19 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20em=E5=A2=9E=E5=8A=A0=E9=87=87=E9=9B=86?=
 =?UTF-8?q?=E6=95=B0=E6=8D=AE=E6=96=B9=E6=B3=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/em/cd.py          | 19 +++++++++++++++++++
 apps/em/serializers.py |  4 ++++
 apps/em/urls.py        |  3 ++-
 apps/em/views.py       | 22 +++++++++++++++++++++-
 4 files changed, 46 insertions(+), 2 deletions(-)
 create mode 100644 apps/em/cd.py

diff --git a/apps/em/cd.py b/apps/em/cd.py
new file mode 100644
index 00000000..b71f6151
--- /dev/null
+++ b/apps/em/cd.py
@@ -0,0 +1,19 @@
+import socket
+from rest_framework.exceptions import ParseError
+
+def get_tyy_data(*args):
+    sc = socket.socket()
+    try:
+        sc.connect((args[0], int(args[1])))
+    except Exception:
+        raise ParseError("无法连接到采集器")
+    sc.send(b"R")
+    resp = sc.recv(1024)
+    if len(resp) < 8:
+        raise ParseError("设备未启动")
+    json_data = resp[5:-4]
+    json_str = json_data.decode('utf-8')
+    return "str", json_str
+
+if __name__ == '__main__':
+    print(get_tyy_data())
\ No newline at end of file
diff --git a/apps/em/serializers.py b/apps/em/serializers.py
index 0278349a..657f6098 100644
--- a/apps/em/serializers.py
+++ b/apps/em/serializers.py
@@ -72,3 +72,7 @@ class EInspectSerializer(CustomModelSerializer):
         model = EInspect
         fields = "__all__"
         read_only_fields = EXCLUDE_FIELDS
+
+
+class CdSerializer(serializers.Serializer):
+    method = serializers.CharField(label="方法名")
\ No newline at end of file
diff --git a/apps/em/urls.py b/apps/em/urls.py
index e9600734..2d162196 100644
--- a/apps/em/urls.py
+++ b/apps/em/urls.py
@@ -1,6 +1,6 @@
 from django.urls import path, include
 from rest_framework.routers import DefaultRouter
-from apps.em.views import EquipmentViewSet, EcheckRecordViewSet, EInspectViewSet, EcateViewSet
+from apps.em.views import EquipmentViewSet, EcheckRecordViewSet, EInspectViewSet, EcateViewSet, CdView
 
 API_BASE_URL = 'api/em/'
 HTML_BASE_URL = 'em/'
@@ -12,4 +12,5 @@ router.register('einspect', EInspectViewSet, basename='einspect')
 router.register('ecate', EcateViewSet, basename='ecate')
 urlpatterns = [
     path(API_BASE_URL, include(router.urls)),
+    path(API_BASE_URL + 'cd/', CdView.as_view()),
 ]
diff --git a/apps/em/views.py b/apps/em/views.py
index a280f4a8..9174e32b 100644
--- a/apps/em/views.py
+++ b/apps/em/views.py
@@ -4,7 +4,7 @@ from drf_yasg import openapi
 from django.utils import timezone
 from apps.em.models import Equipment, EcheckRecord, EInspect, Ecate
 from apps.utils.viewsets import CustomModelViewSet, CustomGenericViewSet
-from apps.em.serializers import EquipmentSerializer, EcheckRecordSerializer, EInspectSerializer, EcateSerializer
+from apps.em.serializers import EquipmentSerializer, EcheckRecordSerializer, EInspectSerializer, EcateSerializer, CdSerializer
 from apps.em.filters import EquipFilterSet
 from rest_framework.exceptions import ParseError
 from rest_framework.mixins import ListModelMixin, CreateModelMixin, DestroyModelMixin
@@ -17,6 +17,9 @@ from rest_framework.response import Response
 from django.conf import settings
 from django.db.models import Count, Case, When, IntegerField
 from apps.enp.services import get_last_envdata
+from rest_framework.views import APIView
+from apps.utils.mixins import MyLoggingMixin
+import importlib
 
 # Create your views here.
 
@@ -167,3 +170,20 @@ class EInspectViewSet(CustomModelViewSet):
     serializer_class = EInspectSerializer
     select_related_fields = ["equipment", "inspect_user"]
     filterset_fields = ["equipment"]
+
+
+class CdView(MyLoggingMixin, APIView):
+
+    @swagger_auto_schema(request_body=CdSerializer)
+    def post(self, request, format=None):
+        """执行采集数据方法
+
+        执行采集数据方法
+        """
+        method = request.data.get("method")
+        m = method.split("(")[0]
+        args = method.split("(")[1].split(")")[0].split(",")
+        module, func = m.rsplit(".", 1)
+        m = importlib.import_module(module)
+        f = getattr(m, func)
+        return Response(f(*args))
\ No newline at end of file