lab_control/connection/tcp_connection.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

import socket

class TCPConnection:
    def __init__(self, ip: str, port: int):
        self._socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self._socket.connect((address, port))

    def configure(self, config: dict) -> None:
        self._socket.settimeout(config["timeout"])

    def send(self, request, responseExpected=True):
        self._socket.sendall(request.encode())

        if responseExpected:
            try:
                response = self._socket.recv(4096).decode()
            except TimeoutError:
                response = None
            return response