def _override_headless(headless):
"""Override configuration file setting for headless session state.
:param headless: Boolean flag to run the session headless
"""
if bool_or_none(headless):
environ["Driver_headless"] = str(bool_or_none(headless))
def test_boolOrNone(self):
self.assertIsNone(bool_or_none(None))
self.assertTrue(bool_or_none(True))
self.assertFalse(bool_or_none(False))
self.assertTrue(bool_or_none(1))
self.assertFalse(bool_or_none(0))
self.assertTrue(bool_or_none("1"))
self.assertTrue(bool_or_none("True"))
self.assertFalse(bool_or_none("False"))
self.assertFalse(bool_or_none(3))
def __setattr__(self, name, value):
if name in (
"lan_1_index",
"lan_2_index",
"lan_3_index",
"lan_4_index",
"lan_5_index",
"lan_6_index",
"lan_ip_routed_index",
):
value = int_or_none(value)
else:
value = bool_or_none(value)
super(LAN_Access, self).__setattr__(name, value)
def __setattr__(self, name, value):
if name in (
"internet_management",
"telnet_server",
"http_server",
"https_server",
"ssh_server",
"snmp_server",
"disable_ping_from_internet",
):
value = bool_or_none(value)
else:
value = int_or_none(value)
super(IPv6Management, self).__setattr__(name, value)
def __setattr__(self, name, value):
if name == "max_login_failures":
value = int_or_none(value)
if value is not None and value > BRUTE_FORCE_MAX_LOGIN_FAILURES:
raise ValueError(
f"{value} Exceeds Max login failures: {BRUTE_FORCE_MAX_LOGIN_FAILURES}"
)
elif name == "penalty_period":
value = int_or_none(value)
if value is not None and value > BRUTE_FORCE_MAX_PENALITY:
raise ValueError(
f"{value} Exceeds Max penalty: {BRUTE_FORCE_MAX_PENALITY} seconds"
)
else:
value = bool_or_none(value)
super(BruteForceProtection, self).__setattr__(name, value)
def __setattr__(self, name, value):
if name == "hostname":
if not (valid_hostname(value) or valid_ipv4_address(value)
or valid_ipv6_address(value)):
raise ValueError(f"Invalid hostname: {value}")
if name in [
"use_https",
"loggedin",
"reboot_required",
"headless",
"search_driver",
]:
value = bool_or_none(value)
elif name == "port":
value = port_or_none(value)
elif name in ["implicit_wait_time", "explicit_wait_time"]:
value = int_or_none(value)
super(DrayTekWebAdmin, self).__setattr__(name, value)
def __setattr__(self, name, value):
if name != "domain_name_allowed":
value = bool_or_none(value)
super(InternetAccessControl, self).__setattr__(name, value)
def __setattr__(self, name, value):
value = bool_or_none(value)
super(DeviceManagement, self).__setattr__(name, value)
def __setattr__(self, name, value):
if name in ("disable_auto_logout", "enable_validation_code"):
value = bool_or_none(value)
super(Management, self).__setattr__(name, value)
def __setattr__(self, name, value):
if name in ("enable", "ssl_enable"):
value = bool_or_none(value)
else:
value = port_or_none(value)
super(CVM_AccessControl, self).__setattr__(name, value)
def __setattr__(self, name, value):
value = bool_or_none(value)
super(Encryption, self).__setattr__(name, value)
def __setattr__(self, name, value):
if name == "user_defined_ports":
value = bool_or_none(value)
else:
value = port_or_none(value)
super(ManagementPort, self).__setattr__(name, value)
def __setattr__(self, name, value):
if name in ["enable_agent", "enable_v3_agent"]:
value = bool_or_none(value)
super(SNMP, self).__setattr__(name, value)
