site_config.py

   1 #!/usr/bin/env python3
   2
   3 import logging
   4 import platform
   5 from dataclasses import dataclass
   6 from typing import Callable, Optional
   7
   8 # Note: this module is fairly early loaded.  Be aware of dependencies.
   9 import config
  10 from type.locations import Location
  11
  12 logger = logging.getLogger(__name__)
  13
  14 args = config.add_commandline_args(
  15     f'Global Site Config ({__file__})',
  16     'Args related to global site-specific configuration',
  17 )
  18 args.add_argument(
  19     '--site_config_override_location',
  20     default='NONE',
  21     const='NONE',
  22     nargs='?',
  23     choices=['HOUSE', 'CABIN', 'NONE'],
  24     help='Where are we, HOUSE, CABIN?  Overrides standard detection code.',
  25 )
  26
  27
  28 @dataclass
  29 class SiteConfig(object):
  30     location_name: str
  31     location: Location
  32     network: str
  33     network_netmask: str
  34     network_router_ip: str
  35     presence_location: Location
  36     is_anyone_present: Callable
  37     arper_minimum_device_count: int
  38     arper_cache_file: str
  39
  40
  41 def get_location_name():
  42     """
  43     Where are we?
  44
  45     >>> location = get_location_name()
  46     >>> location == 'HOUSE' or location == 'CABIN'
  47     True
  48
  49     """
  50     return get_config().location_name
  51
  52
  53 def get_location():
  54     """
  55     Returns location as an enum instead of a string.
  56
  57     >>> from type.locations import Location
  58     >>> location = get_location()
  59     >>> location == Location.HOUSE or location == Location.CABIN
  60     True
  61
  62     """
  63     return get_config().location
  64
  65
  66 def is_anyone_present_wrapper(location: Location):
  67     import base_presence
  68
  69     p = base_presence.PresenceDetection()
  70     return p.is_anyone_in_location_now(location)
  71
  72
  73 def other_location() -> str:
  74     hostname = platform.node()
  75     if '.house' in hostname:
  76         location = 'CABIN'
  77     elif '.cabin' in hostname:
  78         location = 'HOUSE'
  79     else:
  80         raise Exception(f"{hostname} doesn't help me know where I'm running?!")
  81     return location
  82
  83
  84 def this_location() -> str:
  85     hostname = platform.node()
  86     if '.house' in hostname:
  87         location = 'HOUSE'
  88     elif '.cabin' in hostname:
  89         location = 'CABIN'
  90     else:
  91         raise Exception(f"{hostname} doesn't help me know where I'm running?!")
  92     return location
  93
  94
  95 def effective_location(location_override: Optional[str] = None) -> str:
  96     if location_override is None:
  97         try:
  98             location_override = config.config['site_config_override_location']
  99         except KeyError:
 100             location_override = None
 101
 102     if location_override is None or location_override == 'NONE':
 103         location = this_location()
 104     else:
 105         logger.debug(f'site_config\'s location_override was set to: {location_override}')
 106         location = location_override
 107     return location
 108
 109
 110 def get_config(location_override: Optional[str] = None):
 111     """
 112     Get a configuration dataclass with information that is
 113     site-specific including the current running location.
 114
 115     >>> cfg = get_config()
 116     >>> cfg.location_name == 'HOUSE' or cfg.location_name == 'CABIN'
 117     True
 118
 119     """
 120     location = effective_location(location_override)
 121     if location == 'HOUSE':
 122         return SiteConfig(
 123             location_name='HOUSE',
 124             location=Location.HOUSE,
 125             network='10.0.0.0/24',
 126             network_netmask='255.255.255.0',
 127             network_router_ip='10.0.0.1',
 128             presence_location=Location.HOUSE,
 129             is_anyone_present=lambda x=Location.HOUSE: is_anyone_present_wrapper(x),
 130             arper_minimum_device_count=50,
 131             arper_cache_file='/home/scott/cache/.arp_table_cache_house',
 132         )
 133     elif location == 'CABIN':
 134         return SiteConfig(
 135             location_name='CABIN',
 136             location=Location.CABIN,
 137             network='192.168.0.0/24',
 138             network_netmask='255.255.255.0',
 139             network_router_ip='192.168.0.1',
 140             presence_location=Location.CABIN,
 141             is_anyone_present=lambda x=Location.CABIN: is_anyone_present_wrapper(x),
 142             arper_minimum_device_count=15,
 143             arper_cache_file='/home/scott/cache/.arp_table_cache_cabin',
 144         )
 145     else:
 146         raise Exception(f'Unknown site location: {location}')
 147
 148
 149 if __name__ == '__main__':
 150     import doctest
 151
 152     doctest.testmod()