Add command to fetch switch configuration
This commit is contained in:
parent
9261f8af49
commit
1e8e186394
1 changed files with 85 additions and 0 deletions
|
@ -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,
|
||||||
|
|
Loading…
Reference in a new issue