3 from dataclasses import dataclass
6 from typing import Callable
8 # Note: this module is fairly early loaded. Be aware of dependencies.
10 from locations import Location
12 logger = logging.getLogger(__name__)
14 args = config.add_commandline_args(
16 'Args related to __file__'
19 '--site_config_override_location',
23 choices=('HOUSE', 'CABIN', 'NONE'),
24 help='Where are we, HOUSE, CABIN?',
29 class SiteConfig(object):
34 network_router_ip: str
35 presence_location: Location
36 is_anyone_present: Callable[None, bool]
37 arper_minimum_device_count: int
40 def get_location_name():
44 >>> location = get_location_name()
45 >>> location == 'HOUSE' or location == 'CABIN'
49 return get_config().location_name
54 Returns location as an enum instead of a string.
56 >>> from locations import Location
57 >>> location = get_location()
58 >>> location == Location.HOUSE or location == Location.CABIN
62 return get_config().location
65 def is_anyone_present_wrapper(location: Location):
67 p = presence.PresenceDetection()
68 return p.is_anyone_in_location_now(location)
73 Get a configuration dataclass with information that is
74 site-specific including the current running location.
76 >>> cfg = get_config()
77 >>> cfg.location_name == 'HOUSE' or cfg.location_name == 'CABIN'
81 hostname = platform.node()
83 location_override = config.config['site_config_override_location']
85 location_override = 'NONE'
86 if location_override == 'NONE':
87 if '.house' in hostname:
89 elif '.cabin' in hostname:
91 if location == 'HOUSE':
93 location_name = 'HOUSE',
94 location = Location.HOUSE,
95 network = '10.0.0.0/24',
96 network_netmask = '255.255.255.0',
97 network_router_ip = '10.0.0.1',
98 presence_location = Location.HOUSE,
99 is_anyone_present = lambda x=Location.HOUSE: is_anyone_present_wrapper(x),
100 arper_minimum_device_count = 50,
102 elif location == 'CABIN':
104 location_name = 'CABIN',
105 location = Location.CABIN,
106 network = '192.168.0.0/24',
107 network_netmask = '255.255.255.0',
108 network_router_ip = '192.168.0.1',
109 presence_location = Location.CABIN,
110 is_anyone_present = lambda x=Location.CABIN: is_anyone_present_wrapper(x),
111 arper_minimum_device_count = 15,
114 raise Exception(f'Unknown site location: {location}')
117 if __name__ == '__main__':