Fix key case

This commit is contained in:
up-n-atom
2023-12-07 16:06:51 -05:00
committed by GitHub
parent da61a123b9
commit fc820e095c
2 changed files with 9 additions and 9 deletions
+7 -7
View File
@@ -11,14 +11,14 @@ async def set_dns_servers(client: SagemcomClient, dns_servers: tuple[IPv4Address
try:
forwards = await client.get_value_by_xpath('Device/DNS/Relay/Forwardings')
autos = {forward['uid'] for forward in forwards \
if forward.keys() >= {'uid', 'Alias', 'Interface', 'Enable'} and \
forward['Alias'].startswith('IPCP') and \
forward['Interface'].endswith('[IP_DATA]') and \
forward['Enable']}
if forward.keys() >= {'uid', 'alias', 'interface', 'enable'} and \
forward['alias'].startswith('IPCP') and \
forward['interface'].endswith('[IP_DATA]') and \
forward['enable']}
statics = {forward['uid'] for forward in forwards \
if forward.keys() >= {'uid', 'Alias', 'Interface'} and \
forward['Alias'].startswith('STATIC') and \
forward['Interface'].endswith(('[IP_DATA]', '[IP_BR_LAN]'))}
if forward.keys() >= {'uid', 'alias', 'interface'} and \
forward['alias'].startswith('STATIC') and \
forward['interface'].endswith(('[IP_DATA]', '[IP_BR_LAN]'))}
for uid in autos:
await client.set_value_by_xpath(f"Device/DNS/Relay/Forwardings/Forwarding[@uid={uid}]/Enable", False)
for uid, dns_server in zip(statics, dns_servers):
+2 -2
View File
@@ -9,8 +9,8 @@ from . import xmo
async def disable_wifi_radios(client: SagemcomClient, radios: tuple[str] | list[str]) -> None:
try:
value = await client.get_value_by_xpath('Device/WiFi/Radios')
active_radios = {radio['Alias'] for radio in value if radio.keys() >= {'Alias', 'Enable'} and \
radio['Enable']}
active_radios = {radio['alias'] for radio in value if radio.keys() >= {'alias', 'enable'} and \
radio['enable']}
if not len(radios):
radios = click.prompt('Choose radio', type=click.Choice(active_radios), show_choices=True),
invalid_radios = set(radios) - active_radios