add bluetooth

Author: andy

devices/wtvb01_bt50.py

@@ -13,10 +13,13 @@ def __init__(self):
         self.baud = 115200
         self.serial: Optional[SerialDriver, None] = None
 
-    def build_device(self):
+    def build_device_by_serial(self):
         self.serial = SerialDriver()
         self.serial.init(self.baud)
 
+    def build_device_by_bluetooth(self):
+        pass
+
     def read_data(self):
         """
         read buffer
@@ -87,7 +90,7 @@ def get_viration_hz(self, data) -> dict:
 
 if __name__ == '__main__':
     device = WVTB01_BT50()
-    device.build_device()
+    device.build_device_by_serial()
     while True:
         data = device.read_data()
 

drivers/bluetooth_driver.py

@@ -0,0 +1,42 @@
+import bluetooth
+
+
+class BluetoothDevice:
+
+    @classmethod
+    def find_deveces(cls):
+        print("Finding nearby bluetooth devices, please wait for several seconds...")
+        devices = bluetooth.discover_devices(duration=30, lookup_names=True)
+        for address, name in devices:
+            print(address, name)
+
+    def __init__(self):
+        self.sock = None
+
+    def build_device(self, device_mac_addr):
+        print(f"Connecting {device_mac_addr}...")
+        try:
+            self.sock = bluetooth.BluetoothSocket(bluetooth.RFCOMM)
+            self.sock.connect((device_mac_addr, 1))
+        except Exception as e:
+            print(f"Connecting {device_mac_addr} failed")
+            print(e)
+
+    def send(self, data):
+        self.sock.send(data)
+
+    def receive(self, lenth):
+        data = self.sock.recv(lenth)
+        return data
+
+
+if __name__ == '__main__':
+    b = BluetoothDevice()
+    devices = b.find_deveces()
+
+    addr = "06:9C:F5:AC:F7:BC"
+    b.build_device(addr)
+    # print(b.receive(28))
+    # for i in devices:
+    #     b.build_device(i)
+    #     # print(b.receive(28))

drivers/test.py

@@ -0,0 +1,26 @@
+import bluetooth
+
+# 创建一个蓝牙服务端套接字
+server_sock = bluetooth.BluetoothSocket(bluetooth.RFCOMM)
+
+# 绑定本地的一个端口
+port = bluetooth.PORT_ANY
+server_sock.bind(port)
+
+# 开始监听，最多允许1个连接
+server_sock.listen(1)
+
+print("Waiting for connection...")
+
+# 接受客户端的连接请求
+client_sock, address = server_sock.accept()
+
+print("Connected to ", address)
+
+# 接收数据
+data = client_sock.recv(1024)
+print("Received data:", data)
+
+# 关闭套接字
+client_sock.close()
+server_sock.close()