switch.py

#!/usr/bin/env python3
# Switch button widget
#
# Tested with an `input_boolean` and `light`.

import argparse
import json

from utils import entity, icon_folder, service

parser = argparse.ArgumentParser()
parser.add_argument("--entity_id", action="store", required=True)
parser.add_argument("--icon", action="store")
parser.add_argument("--toggle", action="store_true")
parser.add_argument(
    "--text",
    action="store",
    help="Optionally overwrite the text to display (instead of the state).",
)
args = parser.parse_args()

if args.toggle:
    domain = args.entity_id.split(".")[0]
    response = service(domain, "toggle", args.entity_id)

state = entity(args.entity_id)["state"]
info = dict(
    text=state if args.text is None else args.text,
    background_color="255,255,0,255" if state == "on" else "75,75,75,255",
)
if args.icon:
    # download icons from https://github.com/encharm/Font-Awesome-SVG-PNG
    # and place them in `../icons/`
    info["icon_path"] = str(icon_folder / args.icon)

print(json.dumps(info))