Bootstrap some basic session handling

mitel_ommclient2/session.py

@@ -0,0 +1,80 @@
+#!/usr/bin/env python3
+
+from time import sleep
+
+from . import connection
+from . import messages
+
+class Session:
+    """
+        Synchronous API session handler
+
+        :param host: Hostname or IP address of OMM
+        :param username: Username
+        :param password: Password
+        :param port: Port
+        :param connection_class: One of :class:`mitel_ommclient2.connection.Connection` or :class:`mitel_ommclient2.connection.SSLConnection`
+
+        Usage::
+            >>> s = Session("omm.local", "admin", "admin")
+            >>> s.request(mitel_ommclient2.messages.Ping())
+    """
+
+    def __init__(self, host, username, password, port=None, connection_class=None):
+        self.host = host
+        self.username = username
+        self.password = password
+        self.port = port
+        self.connection_class = connection_class
+        if self.connection_class is None:
+            self.connection_class = connection.SSLConnection
+
+        self._connection = None
+
+        self._ensure_connection()
+
+    def _wait_for_respose(self):
+        """
+            Wait until data got received and return message string
+        """
+
+        while True:
+            r = self.connection.recv()
+            if r is not None:
+                return r
+            sleep(0.1)
+
+    def _ensure_connection(self):
+        """
+            Make sure we are connected and logged in
+        """
+
+        if self._connection is None:
+            kwargs = {}
+            if self.port is not None:
+                kwargs["port"] = self.port
+            self._connection = self.connection_class(self.host, **kwargs)
+
+            self._connection.connect()
+
+            self._connection.send(messages.Open(self.username, self.password))
+
+            res = self._wait_for_respose()
+
+    def request(self, request):
+        """
+            Sends a request and waits for response
+
+            :param request: Request object
+
+            Usage::
+                >>> r = s.request(mitel_ommclient2.messages.Ping())
+                >>> r.name
+                'PingResp'
+        """
+
+        message = messages.construct(request)
+        self.connection.send(message)
+
+        res = self._wait_for_respose()
+        return messages.parse(res)