Create btclient.py

2022-03-04 16:57:07 -05:00
parent 35f3dd3b6c
commit c7d604ba21
1 changed files with 105 additions and 0 deletions
--- a/software/pc/btclient.py
+++ b/software/pc/btclient.py
@@ -0,0 +1,105 @@
+"""
+    Copyright (C) 2017 - PRESENT  Zhengyu Peng, https://zpeng.me
+
+    This program is free software: you can redistribute it and/or modify
+    it under the terms of the GNU General Public License as published by
+    the Free Software Foundation, either version 3 of the License, or
+    (at your option) any later version.
+
+    This program is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU General Public License for more details.
+
+    You should have received a copy of the GNU General Public License
+    along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+    ----------
+
+    `                      `
+    -:.                  -#:
+    -//:.              -###:
+    -////:.          -#####:
+    -/:.://:.      -###++##:
+    ..   `://:-  -###+. :##:
+           `:/+####+.   :##:
+    .::::::::/+###.     :##:
+    .////-----+##:    `:###:
+     `-//:.   :##:  `:###/.
+       `-//:. :##:`:###/.
+         `-//:+######/.
+           `-/+####/.
+             `+##+.
+              :##:
+              :##:
+              :##:
+              :##:
+              :##:
+               .+:
+
+"""
+
+from PySide6.QtCore import QObject, Signal, Slot
+import socket
+
+
+class BluetoothClient(QObject):
+    status = Signal(int, object)
+    message = Signal(object, object)
+    ERROR = -1
+    LISTEN = 1
+    CONNECTED = 2
+    STOP = 3
+
+    SIG_NORMAL = 0
+    SIG_STOP = 1
+    SIG_DISCONNECT = 2
+
+    def __init__(self, mac, port):
+        QObject.__init__(self)
+
+        self.mac = mac
+        self.port = port
+        self.bt_socket = socket.socket(
+            socket.AF_BLUETOOTH, socket.SOCK_STREAM, socket.BTPROTO_RFCOMM)
+        self.bt_socket.settimeout(1)
+
+        self.signal = self.SIG_NORMAL
+
+    @Slot()
+    def start(self):
+        try:
+            self.bt_socket.connect((self.mac, self.port))
+        except OSError as err:
+            print(err)
+            # self.status.emit(self.STOP, '')
+        else:
+            # print('connected')
+            self.status.emit(self.CONNECTED, self.mac)
+
+            while True:
+                if self.signal == self.SIG_NORMAL:
+                    try:
+
+                        data = self.bt_socket.recv(4096)
+                    except socket.timeout as t_out:
+                        pass
+                    else:
+                        if data:
+                            self.message.emit(
+                                self.mac+' ('+str(self.port)+')',
+                                data.decode())
+                        else:
+                            break
+                elif self.signal == self.SIG_DISCONNECT:
+                    self.signal = self.SIG_NORMAL
+                    self.bt_socket.close()
+                    break
+        finally:
+            self.status.emit(self.STOP, '')
+
+    def send(self, msg):
+        self.bt_socket.sendall(msg.encode())
+
+    def close(self):
+        self.signal = self.SIG_DISCONNECT