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
|
from dotenv import load_dotenv, find_dotenv
import os
import zerorpc
import gevent
import time
from io import BytesIO
import base64
from PIL import Image
load_dotenv(find_dotenv())
class CortexRPC(object):
def __init__(self, fn):
super().__init__()
self._listener = fn
self.fetching = False
self.connect()
def connect(self):
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.connected)
self.rpc.run()
def connected(self):
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")
print("RPC connected...".format(val))
except zerorpc.TimeoutExpired:
print("RPC timeout :(")
return val
# print("defer sending a file...")
# gevent.sleep(1)
# return self.get_last_frame()
def get_last_frame(self):
if self.fetching:
return "working"
path = os.getenv('TEST_FRAMES_PATH')
onlyfiles = sorted([f for f in os.listdir(path) if os.path.isfile(os.path.join(path, f))])
self.fetching = True
for f in onlyfiles:
gevent.sleep(1/10)
output = BytesIO()
im = Image.open(os.path.join(path, f))
# im = im.convert('RGB')
im.save(output, format="JPEG")
frame = output.getvalue()
output.close()
self.send_frame(f.replace('png', 'jpg'), frame)
self.fetching = False
return "ok"
def send_param(self, key, value):
print("aloha. {}: {}".format(key, value))
if self._listener is not None:
self._listener(key, value)
else:
print("No listener provided to CortexRPC")
return "ok"
def get_params(self):
print("Get params call received!")
return "lmao ok"
def send_frame(self, fn, frame):
if self.relay is not None:
# print("Send frame " + fn)
self.relay.send_frame(fn, frame)
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)
|
