class TcpClient(QObject):
packet = pyqtSignal(str)
connected = pyqtSignal()
disconnected = pyqtSignal()
def __init__(self, host):
super().__init__()
self._sock = QTcpSocket()
self._pending = None
self._host = host
self._log = logging.getLogger("tcp")
self._sock.connected.connect(self.greet)
self._sock.connected.connect(self.connected.emit)
self._sock.disconnected.connect(self.disconnected.emit)
self._sock.readyRead.connect(self.recv)
def set_host(self, host):
self._sock.disconnectFromHost()
self._host = host
def _cancel_pending(self):
self._pending.stop()
self._pending = None
self._sock.connected.disconnect(self._connection_success)
@pyqtSlot()
def _connection_success(self):
self._cancel_pending()
self._log.debug("connection successful")
@pyqtSlot()
def _connection_failure(self):
self._cancel_pending()
self._log.debug("connection failed")
def try_connect(self, port):
if self._pending:
self._cancel_pending()
self._sock.disconnectFromHost()
if self._sock.state() != QAbstractSocket.UnconnectedState:
self._sock.waitForDisconnected()
msg = "connecting to ({}:{})"
msg = msg.format(self._host.toString(), port)
self._log.debug(msg)
self._pending = QTimer()
self._pending.timeout.connect(self._connection_failure)
self._pending.setSingleShot(True)
self._sock.connected.connect(self._connection_success)
self._pending.start(10000)
self._sock.connectToHost(self._host, port)
def greet(self):
self.send(query_tag("port", "udp"), "")
def recv(self):
while self._sock.canReadLine():
data = bytes(self._sock.readLine()) # handle QByteArray
message = data.decode()
self.packet.emit(message)
def send(self, tag, payload):
message = tag + ":" + payload + "\n"
data = message.encode()
self._sock.write(data)
class TcpClient(QObject):
packet = pyqtSignal(str)
connected = pyqtSignal()
disconnected = pyqtSignal()
def __init__(self, host):
super().__init__()
self._sock = QTcpSocket()
self._pending = None
self._host = host
self._log = logging.getLogger('tcp')
self._sock.connected.connect(self.greet)
self._sock.connected.connect(self.connected.emit)
self._sock.disconnected.connect(self.disconnected.emit)
self._sock.readyRead.connect(self.recv)
def set_host(self, host):
self._sock.disconnectFromHost()
self._host = host
def _cancel_pending(self):
self._pending.stop()
self._pending = None
self._sock.connected.disconnect(self._connection_success)
@pyqtSlot()
def _connection_success(self):
self._cancel_pending()
self._log.debug('connection successful')
@pyqtSlot()
def _connection_failure(self):
self._cancel_pending()
self._log.debug('connection failed')
def try_connect(self, port):
if self._pending:
self._cancel_pending()
self._sock.disconnectFromHost()
if self._sock.state() != QAbstractSocket.UnconnectedState:
self._sock.waitForDisconnected()
msg = 'connecting to ({}:{})'
msg = msg.format(self._host.toString(), port)
self._log.debug(msg)
self._pending = QTimer()
self._pending.timeout.connect(self._connection_failure)
self._pending.setSingleShot(True)
self._sock.connected.connect(self._connection_success)
self._pending.start(10000)
self._sock.connectToHost(self._host, port)
def greet(self):
self.send(query_tag('port', 'udp'), '')
def recv(self):
while self._sock.canReadLine():
data = bytes(self._sock.readLine()) # handle QByteArray
message = data.decode()
self.packet.emit(message)
def send(self, tag, payload):
message = tag + ':' + payload + '\n'
data = message.encode()
self._sock.write(data)
