Add command to fetch switch configuration

This commit is contained in:
Correl Roush 2024-11-12 16:45:51 -05:00
parent 9261f8af49
commit 1e8e186394

View file

@ -18,6 +18,17 @@ class SwitchStatus(str, enum.Enum):
off = "off" off = "off"
@dataclasses.dataclass
class SwitchConfiguration:
outlet: int
name: str
power_resume_delay: int
ring_on_reset: bool
safe_shutdown: bool
safe_reboot: bool
shutdown_delay: int
@dataclasses.dataclass @dataclasses.dataclass
class Switch: class Switch:
address: str address: str
@ -76,6 +87,33 @@ class Switch:
for outlet in range(1, outlets + 1) for outlet in range(1, outlets + 1)
} }
def switch_config(self, switch_id: int) -> dict[int, SwitchConfiguration]:
url = yarl.URL(f"http://{self.address}").joinpath(
f"iswitch{switch_id:02d}",
"config.htm",
)
response = requests.get(
str(url),
auth=requests.auth.HTTPBasicAuth(self.user, self.password),
)
soup = bs4.BeautifulSoup(response.content, features="html.parser")
inputs = {i["name"]: i for i in soup.find_all("input")}
outlets = int(inputs["OUTLET"]["value"])
configs = {}
for outlet in range(1, outlets + 1):
letter = chr(ord("a") + (outlets - 1))
config = SwitchConfiguration(
outlet=outlet,
name=inputs[f"T{outlet}"]["value"].strip(),
power_resume_delay=int(inputs[f"power_{letter}"]["value"]),
ring_on_reset=inputs[f"act{outlet}"].has_attr("checked"),
safe_shutdown=inputs[f"shut{outlet}"].has_attr("checked"),
safe_reboot=inputs[f"rbt{outlet}"].has_attr("checked"),
shutdown_delay=int(inputs[f"dlt{letter}"]["value"]),
)
configs[outlet] = config
return configs
app = typer.Typer() app = typer.Typer()
console = rich.console.Console() console = rich.console.Console()
@ -113,6 +151,53 @@ def status(context: typer.Context, switch_id: SwitchIdArgument) -> None:
console.print(table) console.print(table)
@app.command()
def config(context: typer.Context, switch_id: SwitchIdArgument) -> None:
switch: Switch = context.obj
configs = switch.switch_config(switch_id)
def y_n(value: bool) -> str:
return "Y" if value else "N"
if not console.is_interactive:
for key in sorted(configs.keys()):
config = configs[key]
console.print(
"\t".join(
[
config.name,
str(config.power_resume_delay),
y_n(config.ring_on_reset),
y_n(config.safe_shutdown),
y_n(config.safe_reboot),
str(config.shutdown_delay),
]
)
)
else:
table = rich.table.Table(
"Outlet",
"Name",
"Power Resume Delay",
"Ring On/Reset",
"Safe Shutdown/Reboot",
)
for key in sorted(configs.keys()):
config = configs[key]
table.add_row(
str(key),
config.name,
f"{config.power_resume_delay} sec",
y_n(config.ring_on_reset),
"[{} / {}] {}".format(
y_n(config.safe_shutdown),
y_n(config.safe_reboot),
f"{config.shutdown_delay} sec",
),
)
console.print(table)
@app.callback() @app.callback()
def main( def main(
context: typer.Context, context: typer.Context,