torjus/huecli
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Format using ruff

Browse Source
This commit is contained in:
Torjus Håkestad 2024-06-15 11:44:33 +02:00
parent d9371fe266
commit 55a82add9f
1 changed files with 26 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
huecli/__main__.py
View File

@ -12,6 +12,7 @@ class LightState(str, Enum):
on = "off" on = "off"
off = "on" off = "on"
class XYColor(NamedTuple): class XYColor(NamedTuple):
x: float x: float
y: float y: float
@ -45,6 +46,7 @@ class LightID(str, Enum):
all = "all" all = "all"
infuse = "infuse" infuse = "infuse"
def get_mqtt_broker(): def get_mqtt_broker():
if "HUECLI_BROKER" in os.environ: if "HUECLI_BROKER" in os.environ:
return os.environ["HUECLI_BROKER"] return os.environ["HUECLI_BROKER"]
@ -67,6 +69,7 @@ def complete_id(incomplete: str):
completion.append(id) completion.append(id)
return completion return completion
def complete_state(incomplete: str): def complete_state(incomplete: str):
states = ["on", "off"] states = ["on", "off"]
completion = [] completion = []
@ -133,7 +136,7 @@ def set_color(
str_id = NAME_TO_ID[id] str_id = NAME_TO_ID[id]
topic = f"zigbee2mqtt/{str_id}/set" topic = f"zigbee2mqtt/{str_id}/set"
xy = parse_color(color) xy = parse_color(color)
payload_raw= { payload_raw = {
"color": {"x": xy[0], "y": xy[1]}, "color": {"x": xy[0], "y": xy[1]},
"color_mode": "xy", "color_mode": "xy",
} }
@ -143,12 +146,14 @@ def set_color(
@app.command() @app.command()
def set_state( def set_state(
id: Annotated[ id: Annotated[
LightID, typer.Argument(help="ID of light.", autocompletion=complete_id) LightID, typer.Argument(help="ID of light.", autocompletion=complete_id)
], ],
state: Annotated[ state: Annotated[
LightState, typer.Argument(help="State of light.", autocompletion=complete_state) LightState,
]): typer.Argument(help="State of light.", autocompletion=complete_state),
],
):
""" """
Set the state of ID to STATE Set the state of ID to STATE
""" """
@ -160,10 +165,11 @@ def set_state(
@app.command() @app.command()
def set_brightness( def set_brightness(
id: Annotated[ id: Annotated[
LightID, typer.Argument(help="ID of light.", autocompletion=complete_id) LightID, typer.Argument(help="ID of light.", autocompletion=complete_id)
], ],
brightness: int): brightness: int,
):
""" """
Set brigthness of ID to BRIGHTNESS Set brigthness of ID to BRIGHTNESS
""" """
@ -174,6 +180,7 @@ def set_brightness(
payload = json.dumps({"brightness": brightness}) payload = json.dumps({"brightness": brightness})
pub.single(topic, payload, hostname=get_mqtt_broker()) pub.single(topic, payload, hostname=get_mqtt_broker())
@app.command() @app.command()
def version(): def version():
""" """
@ -182,6 +189,7 @@ def version():
version = importlib.metadata.version("huecli") version = importlib.metadata.version("huecli")
typer.echo(f"huecli {version}") typer.echo(f"huecli {version}")
def parse_color(color: str) -> XYColor: def parse_color(color: str) -> XYColor:
if color in COLOR_MAP: if color in COLOR_MAP:
return XYColor(COLOR_MAP[color][0], COLOR_MAP[color][1]) return XYColor(COLOR_MAP[color][0], COLOR_MAP[color][1])
@ -194,16 +202,18 @@ def parse_color(color: str) -> XYColor:
case _: case _:
raise ValueError("Invalid color format") raise ValueError("Invalid color format")
def _parse_rgb_color(r: float, g: float, b: float) -> XYColor: def _parse_rgb_color(r: float, g: float, b: float) -> XYColor:
if any(val > 1 for val in [r,g,b]): if any(val > 1 for val in [r, g, b]):
if any(val > 255 for val in [r,g,b]): if any(val > 255 for val in [r, g, b]):
raise ValueError("RGB values must be between 0 and 255 or 0.0 and 1.0") raise ValueError("RGB values must be between 0 and 255 or 0.0 and 1.0")
vals = [val/255 for val in [r,g,b]] vals = [val / 255 for val in [r, g, b]]
return rgb_to_xy(vals[0], vals[1], vals[2]) return rgb_to_xy(vals[0], vals[1], vals[2])
if all(val > 0 and val < 1 for val in [r,g,b]): if all(val > 0 and val < 1 for val in [r, g, b]):
return rgb_to_xy(r,g,b) return rgb_to_xy(r, g, b)
raise ValueError("RGB values must be between 0 and 255 or 0.0 and 1.0") raise ValueError("RGB values must be between 0 and 255 or 0.0 and 1.0")
def main(): def main():
app() app()