1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
|
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
# 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 AGLBaseService:
api = None
url = None
ip = None
port = None
token = None
uuid = None
def __init__(self, api, ip, port, url=None, token='HELLO', uuid='magic'):
self.api = api
self.url = url
self.ip = ip
self.port = port
self.token = token
self.uuid = uuid
def __await__(self):
return self._async_init().__await__()
async def __aenter__(self):
return self._async_init()
async def _async_init(self):
# setting ping_interval to None because AFB does not support websocket ping
# if set to !None, the library will close the socket after the default timeout
URL = f'ws://{self.ip}:{self.port}/api?token={self.token}&uuid={self.uuid}'
self._conn = connect(close_timeout=0, uri=URL, subprotocols=['x-afb-ws-json1'], ping_interval=None)
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:
print("Received keyboard interrupt, exiting")
except asyncio.CancelledError:
print("Websocket listener coroutine stopped")
except Exception as e:
print("vote du phoque?!?!? : " + str(e))
async def subscribe(self, event):
msgid = randint(0, 999999)
msg = f'[{AFBT.REQUEST},"{msgid}","{self.api}/subscribe",{{"value": "{event}"}}]'
await self.send(msg)
async def unsubscribe(self, event):
verb = 'unsubscribe'
msgid = randint(0, 999999)
msg = f'[2,"{msgid}","{self.api}/{verb}",{{"value": "{event}"}}]'
addrequest(msgid, msg)
await self.send(msg)
|
