def listcmd(ctx: Configuration, devicefilter: str):
"""List all devices from Home Assistant."""
ctx.auto_output("table")
devices = api.get_devices(ctx)
result = [] # type: List[Dict]
if devicefilter == ".*":
result = devices
else:
devicefilterre = re.compile(devicefilter) # type: Pattern
for device in devices:
if devicefilterre.search(device['name']):
result.append(device)
cols = [
('ID', 'id'),
('NAME', 'name'),
('MODEL', 'model'),
('MANUFACTURER', 'manufacturer'),
('AREA', 'area_id'),
]
ctx.echo(
helper.format_output(
ctx, result, columns=ctx.columns if ctx.columns else cols
)
)
def assign(
ctx: Configuration,
area_id_or_name,
names: List[str],
match: Optional[str] = None,
):
"""Update area on one or more devices.
NAMES - one or more name or id (Optional)
"""
ctx.auto_output("data")
devices = api.get_devices(ctx)
result = [] # type: List[Dict]
area = api.find_area(ctx, area_id_or_name)
if not area:
_LOGGING.error("Could not find area with id or name: %s",
area_id_or_name)
sys.exit(1)
if match:
if match == ".*":
result = devices
else:
devicefilterre = re.compile(match) # type: Pattern
for device in devices:
if devicefilterre.search(device['name']):
result.append(device)
for id_or_name in names:
device = next(
(x for x in devices if x['id'] == id_or_name),
None # type: ignore
)
if not device:
device = next(
(x for x in devices if x['name'] == id_or_name),
None, # type: ignore
)
if not device:
_LOGGING.error("Could not find device with id or name: %s",
id_or_name)
sys.exit(1)
result.append(device)
for device in result:
output = api.assign_area(ctx, device['id'], area['area_id'])
if output['success']:
ctx.echo("Successfully assigned '{}' to '{}'".format(
area['name'], device['name']))
else:
_LOGGING.error("Failed to assign '%s' to '%s'", area['name'],
device['name'])
ctx.echo(str(output))
def assign(
ctx: Configuration,
area_id_or_name,
names: List[str],
match: Optional[str] = None,
):
"""Update area on one or more devices.
NAMES - one or more name or id (Optional)
"""
ctx.auto_output("data")
devices = api.get_devices(ctx)
result = [] # type: List[Dict]
area = api.find_area(ctx, area_id_or_name)
if not area:
_LOGGING.error(
"Could not find area with id or name: %s", area_id_or_name
)
sys.exit(1)
if match:
if match == ".*":
result = devices
else:
devicefilterre = re.compile(match) # type: Pattern
for device in devices:
if devicefilterre.search(device['name']):
result.append(device)
for id_or_name in names:
device = next( # type: ignore
(x for x in devices if x['id'] == id_or_name), None
)
if not device:
device = next( # type: ignore
(x for x in devices if x['name'] == id_or_name), None
)
if not device:
_LOGGING.error(
"Could not find device with id or name: %s", id_or_name
)
sys.exit(1)
result.append(device)
for device in result:
output = api.assign_area(ctx, device['id'], area['area_id'])
if output['success']:
ctx.echo(
"Successfully assigned '{}' to '{}'".format(
area['name'], device['name']
)
)
else:
_LOGGING.error(
"Failed to assign '%s' to '%s'", area['name'], device['name']
)
ctx.echo(str(output))