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
|
#!/usr/bin/env python3
"""Most basic definition of a smart device: it must have a name and a
MAC address and may have some optional keywords. All devices have
these whether they are lights, outlets, thermostats, etc..."""
import re
from typing import List, Optional
import arper
class Device(object):
"""Most basic definition of a smart device: it must have a name and a
MAC address and may have some optional keywords. All devices have
these whether they are lights, outlets, thermostats, etc..."""
def __init__(
self,
name: str,
mac: str,
keywords: Optional[str] = "",
):
self.name = name
self.mac = mac
self.keywords = keywords
self.arper = arper.Arper()
if keywords is not None:
self.kws: List[str] = keywords.split(' ')
else:
self.kws = []
def get_name(self) -> str:
return self.name
def get_mac(self) -> str:
return self.mac
def get_ip(self) -> Optional[str]:
return self.arper.get_ip_by_mac(self.mac)
def has_static_ip(self) -> bool:
for kw in self.kws:
m = re.search(r'static:([\d\.]+)', kw)
if m is not None:
ip = m.group(1)
assert self.get_ip() == ip
return True
return False
# Add command -> URL logic here.
def get_keywords(self) -> Optional[List[str]]:
return self.kws
def has_keyword(self, keyword: str) -> bool:
for kw in self.kws:
if kw == keyword:
return True
return False
def get_on_limit_seconds(self) -> Optional[int]:
for kw in self.kws:
m = re.search(r"timeout:(\d+)", kw)
if m is not None:
return int(m.group(1)) * 60
return None
|
