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
|
from dotenv import load_dotenv, find_dotenv
import os
import zerorpc
import gevent
import time
import sys
from io import BytesIO
import base64
from PIL import Image
import atexit
load_dotenv(find_dotenv())
class CortexRPC(object):
def __init__(self, get_fn, set_fn, ready_fn, cmd_fn):
super().__init__()
self._get_fn = get_fn
self._set_fn = set_fn
self._cmd_fn = cmd_fn
self.connect(ready_fn)
def connect(self, ready_fn):
print('Connecting to RPC on port {}'.format(os.getenv('RPC_PORT')))
self.rpc = zerorpc.Server(self)
self.rpc.bind('tcp://127.0.0.1:' + os.getenv('RPC_PORT'))
gevent.spawn(self.rpc.run)
print('RPC running on port {}'.format(os.getenv('RPC_PORT')))
print('Connecting to Relay on port {}'.format(os.getenv('RELAY_PORT')))
self.relay = zerorpc.Client()
self.relay.connect('tcp://127.0.0.1:' + os.getenv('RELAY_PORT'))
val = None
try:
val = self.relay.connected('rpc')
atexit.register(self.disconnecting)
print('RPC connected...'.format(val))
except zerorpc.TimeoutExpired:
print('RPC timeout :(')
ready_fn(self)
# return val
def disconnecting(self):
print('Disconnecting from relay', flush=True, file=sys.stderr)
self.relay.disconnecting()
def ping(self):
return 'pong'
def send_command(self, cmd, payload):
# print('your command, {}: {}'.format(cmd, payload))
if self._cmd_fn is not None:
return self._cmd_fn(cmd, payload)
else:
print('No command listener provided to CortexRPC')
return 'ok'
def set_param(self, key, value):
print('aloha. {}: {}'.format(key, value))
if self._set_fn is not None:
self._set_fn(key, value)
else:
print('No set param listener provided to CortexRPC')
return 'ok'
def get_params(self):
print('Get params call received!')
return self._get_fn()
def send_pil_image(self, fn, meta, im):
output = BytesIO()
im = im.convert('RGB')
# im.save(output, format='JPEG')
im.save(output, format='PNG')
frame = output.getvalue()
output.close()
# self.send_frame(fn.replace('png', 'jpg'), meta, frame)
self.send_frame(fn, meta, frame)
def send_status(self, key, value):
if self.relay is not None:
self.relay.send_status(key, value)
def send_frame(self, fn, meta, frame):
if self.relay is not None:
# print('Send frame ' + fn)
try:
self.relay.send_frame(fn, meta, frame)
except Exception as e:
print('Caught exception...? Wha >>>>>>>>>>>>>>>>>>>')
print(e)
return self.connected()
else:
print('Can\'t send frame, relay not ready...')
if __name__ == '__main__':
def log_param(key, value):
print('Log param. {}: {}'.format(key, value))
# this will block
rpc = CortexRPC(log_param)
|
