azalea/sagemcom
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix key case

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files
xmo/dns.py
+7 -7
View File
@@ -11,14 +11,14 @@ async def set_dns_servers(client: SagemcomClient, dns_servers: tuple[IPv4Address
try: try:
forwards = await client.get_value_by_xpath('Device/DNS/Relay/Forwardings') forwards = await client.get_value_by_xpath('Device/DNS/Relay/Forwardings')
autos = {forward['uid'] for forward in forwards \ autos = {forward['uid'] for forward in forwards \
if forward.keys() >= {'uid', 'Alias', 'Interface', 'Enable'} and \ if forward.keys() >= {'uid', 'alias', 'interface', 'enable'} and \
forward['Alias'].startswith('IPCP') and \ forward['alias'].startswith('IPCP') and \
forward['Interface'].endswith('[IP_DATA]') and \ forward['interface'].endswith('[IP_DATA]') and \
forward['Enable']} forward['enable']}
statics = {forward['uid'] for forward in forwards \ statics = {forward['uid'] for forward in forwards \
if forward.keys() >= {'uid', 'Alias', 'Interface'} and \ if forward.keys() >= {'uid', 'alias', 'interface'} and \
forward['Alias'].startswith('STATIC') and \ forward['alias'].startswith('STATIC') and \
forward['Interface'].endswith(('[IP_DATA]', '[IP_BR_LAN]'))} forward['interface'].endswith(('[IP_DATA]', '[IP_BR_LAN]'))}
for uid in autos: for uid in autos:
await client.set_value_by_xpath(f"Device/DNS/Relay/Forwardings/Forwarding[@uid={uid}]/Enable", False) 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): for uid, dns_server in zip(statics, dns_servers):
xmo/wifi.py
+2 -2
View File
@@ -9,8 +9,8 @@ from . import xmo
async def disable_wifi_radios(client: SagemcomClient, radios: tuple[str] | list[str]) -> None: async def disable_wifi_radios(client: SagemcomClient, radios: tuple[str] | list[str]) -> None:
try: try:
value = await client.get_value_by_xpath('Device/WiFi/Radios') value = await client.get_value_by_xpath('Device/WiFi/Radios')
active_radios = {radio['Alias'] for radio in value if radio.keys() >= {'Alias', 'Enable'} and \ active_radios = {radio['alias'] for radio in value if radio.keys() >= {'alias', 'enable'} and \
radio['Enable']} radio['enable']}
if not len(radios): if not len(radios):
radios = click.prompt('Choose radio', type=click.Choice(active_radios), show_choices=True), radios = click.prompt('Choose radio', type=click.Choice(active_radios), show_choices=True),
invalid_radios = set(radios) - active_radios invalid_radios = set(radios) - active_radios