reverted back to customized fork of librouteros

custom_components/mikrotik_router/librouteros_custom/connections.py

@@ -0,0 +1,37 @@
+# -*- coding: UTF-8 -*-
+
+from .exceptions import ConnectionClosed
+
+
+class SocketTransport:
+
+    def __init__(self, sock):
+        self.sock = sock
+
+    def write(self, data):
+        """
+        Write given bytes to socket. Loop as long as every byte in
+        string is written unless exception is raised.
+        """
+        self.sock.sendall(data)
+
+    def read(self, length):
+        """
+        Read as many bytes from socket as specified in length.
+        Loop as long as every byte is read unless exception is raised.
+        """
+        data = bytearray()
+        while len(data) != length:
+            tmp = None
+            try:
+                tmp = self.sock.recv((length - len(data)))
+            except:
+                raise ConnectionClosed('Socket recv failed.')
+
+            data += tmp
+            if not data:
+                raise ConnectionClosed('Connection unexpectedly closed.')
+        return data
+
+    def close(self):
+        self.sock.close()