annotate light.py @ 5:7eeda7f4f9cd

spell it to_dict, for compat with DataClassJsonMixin
author drewp@bigasterisk.com
date Sun, 28 Jan 2024 16:01:43 -0800
parents e8e4fd6d5619
children fc8ed0efcd72
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
2
c6fd04e07db0 refactor light.py
drewp@bigasterisk.com
parents:
diff changeset
1 import asyncio
c6fd04e07db0 refactor light.py
drewp@bigasterisk.com
parents:
diff changeset
2 import logging
c6fd04e07db0 refactor light.py
drewp@bigasterisk.com
parents:
diff changeset
3 from dataclasses import dataclass
c6fd04e07db0 refactor light.py
drewp@bigasterisk.com
parents:
diff changeset
4
5
7eeda7f4f9cd spell it to_dict, for compat with DataClassJsonMixin
drewp@bigasterisk.com
parents: 4
diff changeset
5 from color import Color
7eeda7f4f9cd spell it to_dict, for compat with DataClassJsonMixin
drewp@bigasterisk.com
parents: 4
diff changeset
6
2
c6fd04e07db0 refactor light.py
drewp@bigasterisk.com
parents:
diff changeset
7 log = logging.getLogger('light')
c6fd04e07db0 refactor light.py
drewp@bigasterisk.com
parents:
diff changeset
8
c6fd04e07db0 refactor light.py
drewp@bigasterisk.com
parents:
diff changeset
9
c6fd04e07db0 refactor light.py
drewp@bigasterisk.com
parents:
diff changeset
10 @dataclass
c6fd04e07db0 refactor light.py
drewp@bigasterisk.com
parents:
diff changeset
11 class Light:
c6fd04e07db0 refactor light.py
drewp@bigasterisk.com
parents:
diff changeset
12 name: str
c6fd04e07db0 refactor light.py
drewp@bigasterisk.com
parents:
diff changeset
13 address: str
c6fd04e07db0 refactor light.py
drewp@bigasterisk.com
parents:
diff changeset
14 online: bool
c6fd04e07db0 refactor light.py
drewp@bigasterisk.com
parents:
diff changeset
15 colorRequest: Color
c6fd04e07db0 refactor light.py
drewp@bigasterisk.com
parents:
diff changeset
16 colorMessage: dict
c6fd04e07db0 refactor light.py
drewp@bigasterisk.com
parents:
diff changeset
17 colorCurrent: Color
c6fd04e07db0 refactor light.py
drewp@bigasterisk.com
parents:
diff changeset
18 latencyMs: float
c6fd04e07db0 refactor light.py
drewp@bigasterisk.com
parents:
diff changeset
19
5
7eeda7f4f9cd spell it to_dict, for compat with DataClassJsonMixin
drewp@bigasterisk.com
parents: 4
diff changeset
20 def to_dict(self):
2
c6fd04e07db0 refactor light.py
drewp@bigasterisk.com
parents:
diff changeset
21 return {
c6fd04e07db0 refactor light.py
drewp@bigasterisk.com
parents:
diff changeset
22 'light': {
c6fd04e07db0 refactor light.py
drewp@bigasterisk.com
parents:
diff changeset
23 'name': self.name,
c6fd04e07db0 refactor light.py
drewp@bigasterisk.com
parents:
diff changeset
24 'address': self.address,
c6fd04e07db0 refactor light.py
drewp@bigasterisk.com
parents:
diff changeset
25 'online': self.online,
c6fd04e07db0 refactor light.py
drewp@bigasterisk.com
parents:
diff changeset
26 'colorRequest': self.colorRequest.to_js(),
c6fd04e07db0 refactor light.py
drewp@bigasterisk.com
parents:
diff changeset
27 'colorMessage': self.colorMessage,
c6fd04e07db0 refactor light.py
drewp@bigasterisk.com
parents:
diff changeset
28 'colorCurrent': self.colorCurrent.to_js(),
c6fd04e07db0 refactor light.py
drewp@bigasterisk.com
parents:
diff changeset
29 'latencyMs': self.latencyMs,
c6fd04e07db0 refactor light.py
drewp@bigasterisk.com
parents:
diff changeset
30 }
c6fd04e07db0 refactor light.py
drewp@bigasterisk.com
parents:
diff changeset
31 }
c6fd04e07db0 refactor light.py
drewp@bigasterisk.com
parents:
diff changeset
32
c6fd04e07db0 refactor light.py
drewp@bigasterisk.com
parents:
diff changeset
33
c6fd04e07db0 refactor light.py
drewp@bigasterisk.com
parents:
diff changeset
34 class Lights:
c6fd04e07db0 refactor light.py
drewp@bigasterisk.com
parents:
diff changeset
35 _d: dict[str, Light] = {}
c6fd04e07db0 refactor light.py
drewp@bigasterisk.com
parents:
diff changeset
36
c6fd04e07db0 refactor light.py
drewp@bigasterisk.com
parents:
diff changeset
37 def add(self, d: Light):
c6fd04e07db0 refactor light.py
drewp@bigasterisk.com
parents:
diff changeset
38 self._d[d.name] = d
c6fd04e07db0 refactor light.py
drewp@bigasterisk.com
parents:
diff changeset
39
c6fd04e07db0 refactor light.py
drewp@bigasterisk.com
parents:
diff changeset
40 def byName(self, name: str) -> Light:
c6fd04e07db0 refactor light.py
drewp@bigasterisk.com
parents:
diff changeset
41 return self._d[name]
c6fd04e07db0 refactor light.py
drewp@bigasterisk.com
parents:
diff changeset
42
c6fd04e07db0 refactor light.py
drewp@bigasterisk.com
parents:
diff changeset
43 async def changes(self): # yields None on any data change
c6fd04e07db0 refactor light.py
drewp@bigasterisk.com
parents:
diff changeset
44 while True:
c6fd04e07db0 refactor light.py
drewp@bigasterisk.com
parents:
diff changeset
45 yield None
5
7eeda7f4f9cd spell it to_dict, for compat with DataClassJsonMixin
drewp@bigasterisk.com
parents: 4
diff changeset
46 await asyncio.sleep(1) # todo
2
c6fd04e07db0 refactor light.py
drewp@bigasterisk.com
parents:
diff changeset
47
5
7eeda7f4f9cd spell it to_dict, for compat with DataClassJsonMixin
drewp@bigasterisk.com
parents: 4
diff changeset
48 def to_dict(self):
7eeda7f4f9cd spell it to_dict, for compat with DataClassJsonMixin
drewp@bigasterisk.com
parents: 4
diff changeset
49 return {'lights': [d.to_dict() for d in sorted(self._d.values(), key=lambda r: r.name)]}