mirror of
https://github.com/python-kasa/python-kasa.git
synced 2025-05-30 21:51:24 +00:00
63 lines
1.3 KiB
Python
63 lines
1.3 KiB
Python
"""Module for cli vacuum commands.."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import asyncclick as click
|
|
from PIL import ImageShow
|
|
|
|
from kasa import (
|
|
Device,
|
|
Module,
|
|
)
|
|
|
|
from .common import (
|
|
error,
|
|
pass_dev_or_child,
|
|
)
|
|
|
|
|
|
@click.group(invoke_without_command=True)
|
|
@click.pass_context
|
|
async def vacuum(ctx: click.Context) -> None:
|
|
"""Vacuum commands."""
|
|
|
|
|
|
@vacuum.group(invoke_without_command=True, name="map")
|
|
@pass_dev_or_child
|
|
async def map_group(dev: Device):
|
|
"""Return map."""
|
|
if not (map := dev.modules.get(Module.Map)):
|
|
error("This device does not support maps.")
|
|
return
|
|
|
|
click.echo(map.map_info)
|
|
click.echo(map.map_data)
|
|
|
|
click.echo("Use `kasa vacuum map show` to display the map")
|
|
|
|
|
|
@map_group.command()
|
|
@pass_dev_or_child
|
|
async def show(dev: Device):
|
|
"""Show current map."""
|
|
if not (map := dev.modules.get(Module.Map)):
|
|
error("This device does not support maps.")
|
|
return
|
|
|
|
img = map.get_map_image()
|
|
|
|
img = img.resize((4 * img.width, 4 * img.height))
|
|
ImageShow.show(img)
|
|
|
|
|
|
@map_group.command()
|
|
@pass_dev_or_child
|
|
async def path(dev: Device):
|
|
"""Show current path."""
|
|
if not (map := dev.modules.get(Module.Map)):
|
|
error("This device does not support maps.")
|
|
return
|
|
|
|
data = map.get_path()
|
|
print(data)
|