esp32-modules/message.py at master · spasea/esp32-modules · GitHub

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
from usocketio.transport import SocketIO
import machine


class Events:
    def __init__(self, room_id: str = None, sensor_id: str = None):
        self.sensor_id = sensor_id
        self.room_id = room_id

        assert self.sensor_id is not None or self.room_id is not None, 'Invalid initial params'

    def base(self) -> str:
        return ':'.join([self.room_id, self.sensor_id])

    def server_state(self) -> str:
        return ':'.join([self.base(), 'server_state'])

    def server_reboot(self) -> str:
        return ':'.join([self.base(), 'server_reboot'])

    def client_state(self) -> str:
        return ':'.join([self.base(), 'client_state'])

    def server_load_state(self) -> str:
        return ':'.join([self.base(), 'server_load_state'])

    def server_change(self) -> str:
        return ':'.join([self.base(), 'server_change'])

    def state(self) -> str:
        return ':'.join([self.base(), 'state'])

    def custom(self, event) -> str:
        return ':'.join([self.base(), event])


class BaseSensor:
    def __init__(self, event_instance: Events, socketio: SocketIO, get_state, load_state):
        self.socketio = socketio
        self.event_instance = event_instance
        self.get_state = get_state
        self.load_state = load_state

    def reboot(self):
        self.socketio.emit(self.event_instance.client_state(), self.get_state())

        machine.reset()

    def post_state(self):
        self.socketio.emit(self.event_instance.state(), self.get_state())

    def add_events(self):
        self.socketio.emit('join', self.event_instance.base())

        @self.socketio.on(self.event_instance.server_reboot())
        def handle(msg):
            self.reboot()

        @self.socketio.on(self.event_instance.server_state())
        def handle(msg):
            self.post_state()

        @self.socketio.on(self.event_instance.server_load_state())
        def handle(payload):
            self.load_state(payload)

        @self.socketio.at_interval(5)
        def handle():
            self.socketio.emit(self.event_instance.state(), self.get_state())