From da9b205f77f0e88ca808bbbf2bbd95d661f29ac4 Mon Sep 17 00:00:00 2001
From: Edi Feschiyan <edi.feschiyan@konsulko.com>
Date: Mon, 9 Mar 2020 12:17:16 +0200
Subject: Initial commit for python testing tools

---
 abstractaglbaseservice.py | 81 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 81 insertions(+)
 create mode 100644 abstractaglbaseservice.py

(limited to 'abstractaglbaseservice.py')

diff --git a/abstractaglbaseservice.py b/abstractaglbaseservice.py
new file mode 100644
index 0000000..d229cc0
--- /dev/null
+++ b/abstractaglbaseservice.py
@@ -0,0 +1,81 @@
+from enum import IntEnum
+import json
+from json import JSONDecodeError
+from random import randint
+import sys
+import asyncio
+from random import randint
+from websockets import connect
+from os import environ
+from argparse import ArgumentParser
+
+import abc
+import inspect
+# https://stackoverflow.com/questions/47555934/how-require-that-an-abstract-method-is-a-coroutine
+
+IPADDR = '127.0.0.1'
+PORT = '30000'
+TOKEN = 'HELLO'
+UUID = 'magic'
+URL = f'ws://{IPADDR}:{PORT}/api?token={TOKEN}&uuid={UUID}'
+
+class AFBT(IntEnum):
+    REQUEST = 2,
+    RESPONSE = 3,
+    ERROR = 4,
+    EVENT = 5
+
+msgq = {}
+
+def addrequest(msgid, msg):
+    msgq[msgid] = {'request': msg, 'response': None}
+
+def addresponse(msgid, msg):
+    if msgid in msgq.keys():
+        msgq[msgid]['response'] = msg
+
+class AbstractAGLBaseService:
+    def __await__(self):
+        return self._async_init().__await__()
+
+    async def __aenter__(self):
+        return self._async_init()
+
+    async def _async_init(self):
+        self._conn = connect(close_timeout=0, uri=URL, subprotocols=['x-afb-ws-json1'])
+        self.websocket = await self._conn.__aenter__()
+        return self
+
+    async def __aexit__(self, *args, **kwargs):
+        await self._conn.__aexit__(*args, **kwargs)
+
+    async def close(self):
+        await self._conn.__aexit__(*sys.exc_info())
+
+    async def send(self, message):
+        await self.websocket.send(message)
+
+    async def receive(self):
+        return await self.websocket.recv()
+
+    async def listener(self):
+        try:
+            while True:
+                msg = await self.receive()
+                print(f"received {msg}")
+                try:
+                    data = json.loads(msg)
+                    if isinstance(data,list):
+                        if data[0] == AFBT.RESPONSE and str.isnumeric(data[1]):
+                            msgid = int(data[1])
+                            if msgid in msgq:
+                                addresponse(msgid, data)
+
+
+                except JSONDecodeError:
+                    print("not decoding a non-json message")
+
+        except KeyboardInterrupt:
+            pass
+        except asyncio.CancelledError:
+            print("websocket listener coroutine stopped")
-- 
cgit 1.2.3-korg