Since this thing is on the innerwebs I suppose it should have a

[python_utils.git] / smart_home / device.py

diff --git a/smart_home/device.py b/smart_home/device.py
index 27860c5bbcc75bdbb97defabf820f0a601a9e0f6..64302ad791b35131e597472547e81027c6ec4451 100644 (file)
--- a/smart_home/device.py
+++ b/smart_home/device.py
@@ -1,20 +1,36 @@
 #!/usr/bin/env python3
 
+# © Copyright 2021-2022, Scott Gasch
+
+"""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 Any, List, Optional, Tuple
+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[List[str]],
+        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 = keywords.split()
+            self.kws: List[str] = keywords.split(' ')
         else:
             self.kws = []
 
@@ -24,6 +40,20 @@ class Device(object):
     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