Example #1
0
def edit(ctx: Configuration, entity, newstate, attributes, merge, json):
    """Edit entity state from Home Assistant."""
    ctx.auto_output('data')
    new_state = newstate
    if json:
        _LOGGING.debug(
            "JSON found overriding/creating new state for entity %s", entity)
        wanted_state = json_.loads(json)
    elif new_state or attributes:
        wanted_state = {}
        existing_state = api.get_state(ctx, entity)

        if existing_state:
            ctx.echo("Existing state found for %s", entity)
            if merge:
                wanted_state = existing_state
        else:
            ctx.echo("No existing state found for '%s'", entity)

        if attributes:
            attributes_dict = helper.to_attributes(attributes)

            new_attr = wanted_state.get('attributes', {})
            new_attr.update(attributes_dict)
            # This is not honoring merge!
            wanted_state['attributes'] = new_attr

        if newstate:
            wanted_state['state'] = newstate
        else:
            if not existing_state:
                raise ValueError("No new or existing state provided.")
            wanted_state['state'] = existing_state['state']

    else:
        existing = api.get_state(ctx, entity)
        if existing:
            existing_raw = helper.raw_format_output(ctx.output, existing,
                                                    ctx.yaml())
        else:
            existing_raw = helper.raw_format_output(ctx.output, {}, ctx.yaml())

        new = click.edit(existing_raw, extension='.{}'.format(ctx.output))

        if new is not None:
            ctx.echo("Updating '%s'", entity)
            if ctx.output == 'yaml':
                wanted_state = ctx.yamlload(new)
            if ctx.output == 'json':
                wanted_state = json_.loads(new)

            api.set_state(ctx, entity, wanted_state)
        else:
            ctx.echo("No edits/changes returned from editor.")
            return

    _LOGGING.debug("wanted: %s", str(wanted_state))
    result = api.set_state(ctx, entity, wanted_state)
    ctx.echo("Entity %s updated successfully", entity)
    _LOGGING.debug("Updated to: %s", result)
Example #2
0
def state(ctx, entity, newstate, attributes, merge, json):
    """edit state from Home Assistant"""
    if json:
        response = req_raw(ctx, "post", "states/{}".format(entity), json)
    elif newstate or attributes:

        wanted_state = {}
        existing_state = None

        response = req_raw(ctx, "get", "states/{}".format(entity))

        if (response.ok):
            click.echo("Existing state found for {}".format(entity))
            existing_state = response.json()
            if (merge):
                wanted_state = existing_state
        else:
            click.echo("No existing state found for '{}'".format(entity))

        if attributes:
            import shlex
            lexer = shlex.shlex(attributes, posix=True)
            lexer.whitespace_split = True
            lexer.whitespace = ','
            attributes_dict = dict(pair.split('=', 1) for pair in lexer)

            newattr = wanted_state.get("attributes", {})
            newattr.update(attributes_dict)
            wanted_state["attributes"] = newattr

        if newstate:
            wanted_state["state"] = newstate
        else:
            if not existing_state:
                raise ValueError("No new or existing state provided.")
            wanted_state["state"] = existing_state["state"]

        print("wanted:" + str(wanted_state))
        newjson = raw_format_output("json", wanted_state)

        response = req_raw(ctx, "post", "states/{}".format(entity), newjson)
    else:  ## no data passed, just use editor
        existing = req_raw(ctx, "get", "states/{}".format(entity)).json()

        existing = raw_format_output(ctx.output, existing)
        new = click.edit(existing, extension=".{}".format(ctx.output))

        if new is not None:
            click.echo("Updating '{}'".format(entity))
            if (ctx.output == "yaml"):
                new = json_.dumps(yaml.load(new))
            response = req_raw(ctx, "post", "states/{}".format(entity), new)
        else:
            click.echo("No edits/changes.")
Example #3
0
def fire(ctx: Configuration, event, json):
    """Fire event in Home Assistant."""
    if json:
        click.echo("Fire {}".format(event))
        response = api.fire_event(ctx, event, json)
    else:
        existing = raw_format_output(ctx.output, {})  # type: ignore
        new = click.edit(existing, extension='.{}'.format(ctx.output))

        if new:
            click.echo("Fire {}".format(event))
            if ctx.output == 'yaml':
                data = yaml.load(new)
            else:
                data = json_.loads(new)

            response = api.fire_event(ctx, event, data)
        else:
            click.echo("No edits/changes.")
            return

    if response:
        ctx.echo(raw_format_output(ctx.output, response))  # type: ignore
def fire(ctx: Configuration, event, json):
    """Fire event in Home Assistant."""
    if json:
        click.echo("Fire {}".format(event))
        response = api.fire_event(ctx, event, json)
    else:
        existing = raw_format_output(ctx.output, [{}], ctx.yaml())
        new = click.edit(existing, extension='.{}'.format(ctx.output))

        if new:
            click.echo("Fire {}".format(event))
            if ctx.output == 'yaml':
                data = ctx.yamlload(new)
            else:
                data = json_.loads(new)

            response = api.fire_event(ctx, event, data)
        else:
            click.echo("No edits/changes.")
            return

    if response:
        ctx.echo(raw_format_output(ctx.output, [response], ctx.yaml()))
Example #5
0
def event(ctx, event, json):
    """Edit/fire event in Home Assistant"""
    if json:
        click.echo("Fire {}".format(event))
        response = req_raw(ctx, "post", "events/{}".format(event), json)
        response.raise_for_status()
    else:  ## no data passed, just use editor
        existing = raw_format_output(ctx.output, {})
        new = click.edit(existing, extension=".{}".format(ctx.output))

        if new is not None:
            click.echo("Fire {}".format(event))
            if (ctx.output == "yaml"):
                new = json_.dumps(yaml.load(new))
            response = req_raw(ctx, "post", "events/{}".format(event), new)
            response.raise_for_status()
        else:
            print("No edits/changes.")
def edit(ctx: Configuration, entity, newstate, attributes, merge, json):
    """Edit entity state from Home Assistant."""
    ctx.auto_output('data')
    if json:
        _LOGGING.debug(
            "json found overriding/creating new state for entity %s", entity
        )
        wanted_state = json_.loads(json)
    elif newstate or attributes:
        wanted_state = {}
        existing_state = api.get_state(ctx, entity)

        if existing_state:
            ctx.echo("Existing state found for %s", entity)
            if merge:
                wanted_state = existing_state
        else:
            ctx.echo("No existing state found for '%s'", entity)

        if attributes:
            attributes_dict = helper.to_attributes(attributes)

            newattr = wanted_state.get('attributes', {})
            newattr.update(attributes_dict)
            # this is not hornoring merge!
            wanted_state['attributes'] = newattr

        if newstate:
            wanted_state['state'] = newstate
        else:
            if not existing_state:
                raise ValueError("No new or existing state provided.")
            wanted_state['state'] = existing_state['state']

    else:
        existing = api.get_state(ctx, entity)
        if existing:
            existingraw = helper.raw_format_output(
                ctx.output, [existing], ctx.yaml()
            )[0]
        else:
            existingraw = helper.raw_format_output(
                ctx.output, [{}], ctx.yaml()
            )[0]

        new = click.edit(existingraw, extension='.{}'.format(ctx.output))

        if new is not None:
            ctx.echo("Updating '%s'", entity)
            if ctx.output == 'yaml':
                wanted_state = ctx.yamlload(new)
            if ctx.output == 'json':
                wanted_state = json_.loads(new)

            api.set_state(ctx, entity, wanted_state)
        else:
            ctx.echo("No edits/changes returned from editor.")
            return

    _LOGGING.debug("wanted: %s", str(wanted_state))
    result = api.set_state(ctx, entity, wanted_state)
    ctx.echo("Entity %s updated succesfully", entity)
    _LOGGING.debug("Updated to: %s", result)