feat: 物联网相关代码

backend/device/websocket.py

@@ -0,0 +1,41 @@
+import json
+from hashlib import md5
+from datetime import datetime
+
+import jwt
+from channels.generic.websocket import AsyncJsonWebsocketConsumer
+
+from application.websocketConfig import set_message
+
+
+class DeviceStatusWebSocket(AsyncJsonWebsocketConsumer):
+    """设备状态ws"""
+    chat_group = 'device_status'
+
+    async def connect(self):
+        try:
+            print('设备ws创建连接', self.scope)
+            self.token = self.scope['url_route']['kwargs']['token']
+            decoded_result = jwt.decode(self.token, settings.SECRET_KEY, algorithms=['HS256'])
+            if decoded_result:
+                self.uid = decoded_result.get('user_id')
+                self.hash = md5(str(self.token).encode('utf-8'))
+                self.chat_name = f'{self.chat_group}:user_{self.uid}'
+                await self.channel_layer.group_add(self.chat_name, self.channel_name)
+                await self.accept()
+            else:
+                raise jwt.InvalidSignatureError()
+        except jwt.InvalidSignatureError:
+            await self.send_json(set_message('system', 'SYSTEM', {'message': 'Token无效'}), True)
+            await self.disconnect(None)
+        except jwt.ExpiredSignatureError:
+            await self.send_json(set_message('system', 'SYSTEM', {'message': 'Token过期'}), True)
+            await self.disconnect(None)
+
+    async def disconnect(self, code):
+        print('设备ws连接关闭')
+        await self.channel_layer.group_discard(self.chat_name, self.channel_name)
+        try:
+            await self.close(code)
+        except:
+            pass