Ugh, a bunch of things. @overrides. --lmodule. Chromecasts. etc...

[python_utils.git] / site_config.py

diff --git a/site_config.py b/site_config.py
index 95ff5d403dcd900861c3a2c2ff8b37f3c5672c0a..e3b186daa2839c79063ca399997db1f9508fd021 100644 (file)
--- a/site_config.py
+++ b/site_config.py
@@ -3,59 +3,99 @@
 from dataclasses import dataclass
 import logging
 import platform
-from typing import Optional
+from typing import Callable
 
+# Note: this module is fairly early loaded.  Be aware of dependencies.
 import config
+import presence
 
 logger = logging.getLogger(__name__)
 args = config.add_commandline_args(
     f'({__file__})',
     'Args related to __file__'
 )
-
 args.add_argument(
-    '--site_config_location',
-    default='AUTO',
-    const='AUTO',
+    '--site_config_override_location',
+    default='NONE',
+    const='NONE',
     nargs='?',
-    choices=('HOUSE', 'CABIN', 'AUTO'),
-    help='Where are we, HOUSE, CABIN or AUTO?',
+    choices=('HOUSE', 'CABIN', 'NONE'),
+    help='Where are we, HOUSE, CABIN?',
 )
 
 
 @dataclass
 class SiteConfig(object):
+    location: str
     network: str
     network_netmask: str
     network_router_ip: str
+    presence_location: presence.Location
+    is_anyone_present: Callable[None, bool]
+    arper_minimum_device_count: int
 
 
 def get_location():
-    location = config.config['site_config_location']
-    if location == 'AUTO':
-        hostname = platform.node()
+    """
+    Where are we?
+
+    >>> location = get_location()
+    >>> location == 'HOUSE' or location == 'CABIN'
+    True
+
+    """
+    return get_config().location
+
+
+def is_anyone_present_wrapper(location: presence.Location):
+    p = presence.PresenceDetection()
+    return p.is_anyone_in_location_now(location)
+
+
+def get_config():
+    """
+    Get a configuration dataclass with information that is
+    site-specific including the current running location.
+
+    >>> cfg = get_config()
+    >>> cfg.location == 'HOUSE' or cfg.location == 'CABIN'
+    True
+
+    """
+    hostname = platform.node()
+    try:
+        location_override = config.config['site_config_override_location']
+    except KeyError:
+        location_override = 'NONE'
+    if location_override == 'NONE':
         if '.house' in hostname:
             location = 'HOUSE'
         elif '.cabin' in hostname:
             location = 'CABIN'
-        else:
-            raise Exception(f'Unknown hostname {hostname}, help.')
-    return location
-
-
-def get_config():
-    location = get_location()
     if location == 'HOUSE':
         return SiteConfig(
+            location = 'HOUSE',
             network = '10.0.0.0/24',
             network_netmask = '255.255.255.0',
             network_router_ip = '10.0.0.1',
+            presence_location = presence.Location.HOUSE,
+            is_anyone_present = lambda x=presence.Location.HOUSE: is_anyone_present_wrapper(x),
+            arper_minimum_device_count = 50,
         )
     elif location == 'CABIN':
         return SiteConfig(
+            location = 'CABIN',
             network = '192.168.0.0/24',
             network_netmask = '255.255.255.0',
             network_router_ip = '192.168.0.1',
+            presence_location = presence.Location.CABIN,
+            is_anyone_present = lambda x=presence.Location.CABIN: is_anyone_present_wrapper(x),
+            arper_minimum_device_count = 15,
         )
     else:
-        raise Exception('Unknown site location')
+        raise Exception(f'Unknown site location: {location}')
+
+
+if __name__ == '__main__':
+    import doctest
+    doctest.testmod()