949 lines
31 KiB
Python
949 lines
31 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
generate.py — Chore Tracker Code Generator
|
|
==========================================
|
|
Reads chores_config.yaml and generates:
|
|
• chore-tracker-esphome.yaml (flash to ESP32 via ESPHome)
|
|
• chore-tracker-ha.yaml (Home Assistant config)
|
|
• chore-tracker-dashboard.yaml (Lovelace dashboard)
|
|
|
|
Target hardware: Waveshare ESP32-S3-Touch-LCD-7 (800x480)
|
|
|
|
Home screen behaviour:
|
|
- Each kid's button has a RED outline while any chores remain undone
|
|
- Button turns SOLID GREEN when ALL chores are complete
|
|
- Resets to red automatically at midnight
|
|
|
|
Each kid must define their own chores list in chores_config.yaml.
|
|
|
|
Usage:
|
|
python3 generate.py
|
|
python3 generate.py --config my_other_config.yaml
|
|
python3 generate.py --out ./output_dir
|
|
"""
|
|
|
|
import argparse
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
try:
|
|
import yaml
|
|
except ImportError:
|
|
print("PyYAML not found. Install it with: pip install pyyaml")
|
|
sys.exit(1)
|
|
|
|
|
|
# ─────────────────────────────────────────────────────────────────────────────
|
|
# Helpers
|
|
# ─────────────────────────────────────────────────────────────────────────────
|
|
|
|
# Screen padding — space kept clear around all edges (pixels)
|
|
PAD = 12
|
|
|
|
|
|
def slug(name: str) -> str:
|
|
return name.lower().replace(" ", "_").replace("-", "_")
|
|
|
|
def kid_slug(kid: dict) -> str:
|
|
return slug(kid["name"])
|
|
|
|
def eid(kid: dict, chore: dict) -> str:
|
|
return f"{kid_slug(kid)}_{slug(chore['name'])}"
|
|
|
|
def ha_switch(kid: dict, chore: dict) -> str:
|
|
return f"switch.chore_tracker_{eid(kid, chore)}"
|
|
|
|
# mdi:checkbox-marked-circle-outline — safe default if icon is omitted
|
|
DEFAULT_ICON = "\U000F0133"
|
|
|
|
def get_chores(kid: dict) -> list:
|
|
chores = kid["chores"]
|
|
for c in chores:
|
|
if "icon" not in c:
|
|
c["icon"] = DEFAULT_ICON
|
|
return chores
|
|
|
|
|
|
# ─────────────────────────────────────────────────────────────────────────────
|
|
# ESPHome generator
|
|
# ─────────────────────────────────────────────────────────────────────────────
|
|
|
|
def gen_esphome(cfg: dict) -> str:
|
|
s = cfg["settings"]
|
|
kids = cfg["kids"]
|
|
|
|
# ── Switches (one per chore per kid) ──────────────────────────────────────
|
|
switch_blocks = []
|
|
for kid in kids:
|
|
chores = get_chores(kid)
|
|
ks = kid_slug(kid)
|
|
switch_blocks.append(f"\n # ── {kid['name']}'s chores ──────────────────────────\n")
|
|
for chore in chores:
|
|
e = eid(kid, chore)
|
|
switch_blocks.append(f"""\
|
|
- platform: template
|
|
name: "{kid['name']} - {chore['name']}"
|
|
id: {e}
|
|
icon: mdi:checkbox-marked-circle
|
|
optimistic: true
|
|
restore_mode: RESTORE_DEFAULT_OFF
|
|
on_turn_on:
|
|
then:
|
|
- script.execute: update_{ks}_ui
|
|
- homeassistant.service:
|
|
service: switch.turn_on
|
|
data:
|
|
entity_id: switch.chore_tracker_{e}
|
|
on_turn_off:
|
|
then:
|
|
- script.execute: update_{ks}_ui
|
|
- homeassistant.service:
|
|
service: switch.turn_off
|
|
data:
|
|
entity_id: switch.chore_tracker_{e}
|
|
""")
|
|
|
|
# ── Sensors (chores-done count only — no stars) ────────────────────────
|
|
sensor_blocks = []
|
|
for kid in kids:
|
|
chores = get_chores(kid)
|
|
ks = kid_slug(kid)
|
|
done_lines = "\n".join(
|
|
f" if (id({eid(kid, c)}).state) done++;" for c in chores
|
|
)
|
|
sensor_blocks.append(f"""\
|
|
- platform: template
|
|
name: "{kid['name']} Chores Done"
|
|
id: {ks}_chores_done
|
|
accuracy_decimals: 0
|
|
update_interval: 2s
|
|
lambda: |-
|
|
int done = 0;
|
|
{done_lines}
|
|
return done;
|
|
|
|
""")
|
|
|
|
# ── Midnight reset calls ──────────────────────────────────────────────────
|
|
reset_calls = "\n".join(
|
|
f" id(reset_{kid_slug(k)}_chores).execute();"
|
|
for k in kids
|
|
)
|
|
|
|
lvgl_pages = _gen_lvgl_pages(kids)
|
|
scripts = _gen_scripts(kids)
|
|
|
|
# Collect all unique MDI glyphs used (avatars + chore icons)
|
|
all_glyphs = sorted(set(
|
|
[k["avatar"] for k in kids] +
|
|
[c["icon"] for k in kids for c in get_chores(k)]
|
|
))
|
|
glyphs_str = "".join(all_glyphs)
|
|
|
|
return f"""\
|
|
################################################################################
|
|
# chore-tracker-esphome.yaml (AUTO-GENERATED — edit chores_config.yaml)
|
|
# Kids: {", ".join(k["name"] for k in kids)}
|
|
#
|
|
# HOME SCREEN BEHAVIOUR:
|
|
# Red outline = chores incomplete
|
|
# Solid green = all chores done
|
|
# Resets to red automatically at midnight
|
|
#
|
|
# Hardware: Waveshare ESP32-S3-Touch-LCD-7 (800x480)
|
|
################################################################################
|
|
|
|
esphome:
|
|
name: {s["device_name"]}
|
|
friendly_name: "{s["friendly_name"]}"
|
|
on_boot:
|
|
priority: -10
|
|
then:
|
|
- light.turn_on: backlight
|
|
- lvgl.page.show: page_home
|
|
|
|
esp32:
|
|
board: esp32-s3-devkitc-1
|
|
framework:
|
|
type: esp-idf
|
|
|
|
psram:
|
|
mode: octal
|
|
speed: 80MHz
|
|
|
|
wifi:
|
|
ssid: {s["wifi_ssid"]}
|
|
password: {s["wifi_password"]}
|
|
ap:
|
|
ssid: "{s['friendly_name']} Hotspot"
|
|
password: "choretracker"
|
|
|
|
captive_portal:
|
|
logger:
|
|
level: INFO
|
|
|
|
api:
|
|
encryption:
|
|
key: {s["api_key"]}
|
|
|
|
ota:
|
|
- platform: esphome
|
|
password: {s["ota_password"]}
|
|
|
|
# ── I2C ───────────────────────────────────────────────────────────────────────
|
|
i2c:
|
|
sda: 8
|
|
scl: 9
|
|
|
|
# ── CH422G IO Expander (controls LCD reset, touch reset, backlight) ───────────
|
|
ch422g:
|
|
- id: ch422g_hub
|
|
|
|
# ── Display ───────────────────────────────────────────────────────────────────
|
|
display:
|
|
- platform: mipi_rgb
|
|
model: ESP32-S3-TOUCH-LCD-7-800X480
|
|
id: main_display
|
|
update_interval: never
|
|
auto_clear_enabled: false
|
|
reset_pin:
|
|
ch422g: ch422g_hub
|
|
number: 3
|
|
mode:
|
|
output: true
|
|
|
|
# ── Touchscreen ───────────────────────────────────────────────────────────────
|
|
touchscreen:
|
|
platform: gt911
|
|
id: touch
|
|
update_interval: 120ms
|
|
reset_pin:
|
|
ch422g: ch422g_hub
|
|
number: 1
|
|
mode:
|
|
output: true
|
|
|
|
# ── Backlight ─────────────────────────────────────────────────────────────────
|
|
output:
|
|
- platform: template
|
|
id: lcd_backlight_out
|
|
type: binary
|
|
write_action:
|
|
- if:
|
|
condition:
|
|
lambda: return state;
|
|
then:
|
|
- switch.turn_on: lcd_backlight_raw
|
|
else:
|
|
- switch.turn_off: lcd_backlight_raw
|
|
|
|
light:
|
|
- platform: binary
|
|
name: "Display Backlight"
|
|
output: lcd_backlight_out
|
|
id: backlight
|
|
|
|
# ── Midnight reset ────────────────────────────────────────────────────────────
|
|
time:
|
|
- platform: homeassistant
|
|
id: ha_time
|
|
on_time:
|
|
- hours: 0
|
|
minutes: 0
|
|
seconds: 0
|
|
then:
|
|
- lambda: |-
|
|
{reset_calls}
|
|
- lvgl.page.show: page_home
|
|
|
|
# ── Fonts (place files in /config/esphome/fonts/) ────────────────────────────
|
|
font:
|
|
- file: "fonts/Nunito-Black.ttf"
|
|
id: font_title
|
|
size: 40
|
|
- file: "fonts/Nunito-ExtraBold.ttf"
|
|
id: font_name
|
|
size: 30
|
|
- file: "fonts/Nunito-Bold.ttf"
|
|
id: font_med
|
|
size: 22
|
|
- file: "fonts/Nunito-SemiBold.ttf"
|
|
id: font_small
|
|
size: 17
|
|
- file: "fonts/Nunito-SemiBold.ttf"
|
|
id: font_tiny
|
|
size: 13
|
|
# MDI icon font — used for chore icons and kid avatars
|
|
- file: "fonts/materialdesignicons-webfont.ttf"
|
|
id: font_mdi_large
|
|
size: 48
|
|
bpp: 4
|
|
glyphs: {glyphs_str}
|
|
- file: "fonts/materialdesignicons-webfont.ttf"
|
|
id: font_mdi_small
|
|
size: 32
|
|
bpp: 4
|
|
glyphs: {glyphs_str}
|
|
|
|
# ── Switches — backlight raw + one per chore per kid ─────────────────────────
|
|
switch:
|
|
- platform: gpio
|
|
id: lcd_backlight_raw
|
|
name: "LCD Backlight Raw"
|
|
restore_mode: ALWAYS_ON
|
|
pin:
|
|
ch422g: ch422g_hub
|
|
number: 2
|
|
mode:
|
|
output: true
|
|
{"".join(switch_blocks)}
|
|
|
|
# ── Sensors — reported to HA ─────────────────────────────────────────────────
|
|
sensor:
|
|
{"".join(sensor_blocks)}
|
|
|
|
# ── Scripts — reset + UI update ───────────────────────────────────────────────
|
|
script:
|
|
{scripts}
|
|
|
|
# ── LVGL UI ───────────────────────────────────────────────────────────────────
|
|
lvgl:
|
|
displays:
|
|
- main_display
|
|
touchscreens:
|
|
- touch
|
|
theme:
|
|
button:
|
|
radius: 20
|
|
border_width: 0
|
|
pages:
|
|
{lvgl_pages}
|
|
"""
|
|
|
|
|
|
# ─────────────────────────────────────────────────────────────────────────────
|
|
# LVGL page builder
|
|
# ─────────────────────────────────────────────────────────────────────────────
|
|
|
|
def _gen_lvgl_pages(kids: list) -> str:
|
|
pages = []
|
|
|
|
# ── HOME PAGE ─────────────────────────────────────────────────────────────
|
|
n = len(kids)
|
|
usable_w = 800 - PAD * 2
|
|
btn_w = min(200, max(140, (usable_w - 40 - 20 * (n - 1)) // n))
|
|
total_w = btn_w * n + 20 * (n - 1)
|
|
start_x = PAD + (usable_w - total_w) // 2
|
|
btn_y = PAD + 118
|
|
btn_h = 210
|
|
|
|
reset_all_calls = "\n".join(
|
|
f" - script.execute: reset_{kid_slug(k)}_chores"
|
|
for k in kids
|
|
)
|
|
|
|
home_btns = ""
|
|
for i, kid in enumerate(kids):
|
|
x = start_x + i * (btn_w + 20)
|
|
ks = kid_slug(kid)
|
|
home_btns += f"""\
|
|
- button:
|
|
id: home_btn_{ks}
|
|
x: {x}
|
|
y: {btn_y}
|
|
width: {btn_w}
|
|
height: {btn_h}
|
|
bg_color: 0xFF4757
|
|
bg_opa: COVER
|
|
border_width: 0
|
|
radius: 24
|
|
on_click:
|
|
then:
|
|
- lvgl.page.show: page_{ks}
|
|
widgets:
|
|
- label:
|
|
align: CENTER
|
|
y: -45
|
|
text: "{kid['avatar']}"
|
|
text_font: font_mdi_large
|
|
- label:
|
|
align: CENTER
|
|
y: 22
|
|
text: "{kid['name']}"
|
|
text_font: font_name
|
|
text_color: 0xFFFFFF
|
|
- label:
|
|
id: home_status_{ks}
|
|
align: CENTER
|
|
y: 66
|
|
text: "not done"
|
|
text_font: font_tiny
|
|
text_color: 0xFFFFFF
|
|
|
|
"""
|
|
|
|
pages.append(f"""\
|
|
- id: page_home
|
|
bg_color: 0xFFF9F0
|
|
widgets:
|
|
- label:
|
|
x: 0
|
|
y: 36
|
|
width: {800 - PAD * 2}
|
|
align: TOP_MID
|
|
text: "Chore Tracker"
|
|
text_font: font_title
|
|
text_color: 0x2D3436
|
|
- label:
|
|
x: 0
|
|
y: 90
|
|
width: {800 - PAD * 2}
|
|
align: TOP_MID
|
|
text: "Tap a name to check off chores"
|
|
text_font: font_small
|
|
text_color: 0xB2BEC3
|
|
- button:
|
|
x: 640
|
|
y: 428
|
|
width: 148
|
|
height: 40
|
|
bg_color: 0xFF4757
|
|
radius: 12
|
|
on_click:
|
|
then:
|
|
{reset_all_calls}
|
|
widgets:
|
|
- label:
|
|
align: CENTER
|
|
text: "Reset All"
|
|
text_color: 0xFFFFFF
|
|
text_font: font_tiny
|
|
{home_btns}""")
|
|
|
|
# ── PER-KID CHORE PAGES ───────────────────────────────────────────────────
|
|
for kid in kids:
|
|
chores = get_chores(kid)
|
|
ks = kid_slug(kid)
|
|
n_chores = len(chores)
|
|
|
|
cols = 3
|
|
rows = (n_chores + cols - 1) // cols
|
|
sidebar_w = 172
|
|
gap = 10
|
|
content_w = 800 - sidebar_w - gap - 12 # right pad
|
|
card_w = (content_w - gap * (cols - 1)) // cols
|
|
card_h = min(138, (480 - PAD - gap * (rows + 1)) // rows)
|
|
|
|
card_widgets = ""
|
|
for idx, chore in enumerate(chores):
|
|
row = idx // cols
|
|
col = idx % cols
|
|
x = sidebar_w + gap + col * (card_w + gap)
|
|
y = 12 + row * (card_h + gap)
|
|
e = eid(kid, chore)
|
|
icon_y = -(card_h // 4)
|
|
label_y = card_h // 8
|
|
card_widgets += f"""\
|
|
- button:
|
|
id: card_{e}
|
|
x: {x}
|
|
y: {y}
|
|
width: {card_w}
|
|
height: {card_h}
|
|
bg_color: 0xFFFFFF
|
|
border_color: 0xEEEEEE
|
|
border_width: 2
|
|
radius: 20
|
|
shadow_color: 0xCCCCCC
|
|
shadow_width: 4
|
|
shadow_ofs_y: 3
|
|
on_click:
|
|
then:
|
|
- switch.toggle: {e}
|
|
widgets:
|
|
- label:
|
|
id: check_{e}
|
|
align: TOP_RIGHT
|
|
x: -10
|
|
y: 8
|
|
text: ""
|
|
text_font: font_med
|
|
- label:
|
|
align: CENTER
|
|
y: {icon_y}
|
|
text: "{chore['icon']}"
|
|
text_font: font_mdi_small
|
|
text_color: 0x{kid['color']}
|
|
- label:
|
|
align: CENTER
|
|
y: {label_y}
|
|
text: "{chore['name']}"
|
|
text_font: font_small
|
|
text_color: 0x2D3436
|
|
|
|
"""
|
|
|
|
pages.append(f"""\
|
|
- id: page_{ks}
|
|
bg_color: 0xFFF9F0
|
|
widgets:
|
|
|
|
# ── Sidebar ──────────────────────────────────────────────────────────
|
|
- obj:
|
|
x: 0
|
|
y: 0
|
|
width: {sidebar_w}
|
|
height: 480
|
|
bg_color: 0x{kid['color']}
|
|
border_width: 0
|
|
radius: 0
|
|
pad_all: 0
|
|
widgets:
|
|
- label:
|
|
x: 0
|
|
y: 22
|
|
width: {sidebar_w}
|
|
align: TOP_MID
|
|
text: "{kid['avatar']}"
|
|
text_font: font_mdi_large
|
|
- label:
|
|
x: 0
|
|
y: 76
|
|
width: {sidebar_w}
|
|
align: TOP_MID
|
|
text: "{kid['name']}"
|
|
text_font: font_name
|
|
text_color: 0xFFFFFF
|
|
- bar:
|
|
id: progress_bar_{ks}
|
|
x: 14
|
|
y: 132
|
|
width: {sidebar_w - 28}
|
|
height: 16
|
|
bg_color: 0x{kid['color_dark']}
|
|
indicator:
|
|
bg_color: 0xFFFFFF
|
|
value: 0
|
|
min_value: 0
|
|
max_value: {n_chores}
|
|
- label:
|
|
id: progress_label_{ks}
|
|
x: 0
|
|
y: 156
|
|
width: {sidebar_w}
|
|
align: TOP_MID
|
|
text: "0 / {n_chores}"
|
|
text_font: font_small
|
|
text_color: 0xDDEEFF
|
|
- label:
|
|
id: all_done_label_{ks}
|
|
x: 0
|
|
y: 192
|
|
width: {sidebar_w}
|
|
align: TOP_MID
|
|
text: ""
|
|
text_font: font_small
|
|
text_color: 0xFFFFFF
|
|
- button:
|
|
x: 14
|
|
y: {480 - PAD - 44 - 6 - 44}
|
|
width: {sidebar_w - 28}
|
|
height: 44
|
|
bg_color: 0xFF4757
|
|
radius: 14
|
|
on_click:
|
|
then:
|
|
- script.execute: reset_{ks}_chores
|
|
widgets:
|
|
- label:
|
|
align: CENTER
|
|
text: "Reset"
|
|
text_color: 0xFFFFFF
|
|
text_font: font_small
|
|
- button:
|
|
x: 14
|
|
y: {480 - PAD - 44}
|
|
width: {sidebar_w - 28}
|
|
height: 44
|
|
bg_color: 0x{kid['color_dark']}
|
|
radius: 14
|
|
on_click:
|
|
then:
|
|
- lvgl.page.show: page_home
|
|
widgets:
|
|
- label:
|
|
align: CENTER
|
|
text: "Home"
|
|
text_color: 0xFFFFFF
|
|
text_font: font_small
|
|
|
|
# ── Chore cards ───────────────────────────────────────────────────────
|
|
{card_widgets}""")
|
|
|
|
return "\n".join(pages)
|
|
|
|
|
|
# ─────────────────────────────────────────────────────────────────────────────
|
|
# Script builder
|
|
# ─────────────────────────────────────────────────────────────────────────────
|
|
|
|
def _gen_scripts(kids: list) -> str:
|
|
blocks = []
|
|
for kid in kids:
|
|
chores = get_chores(kid)
|
|
ks = kid_slug(kid)
|
|
n = len(chores)
|
|
|
|
card_updates = ""
|
|
for chore in chores:
|
|
e = eid(kid, chore)
|
|
card_updates += f"""\
|
|
if (id({e}).state) {{
|
|
lv_obj_set_style_bg_color(id(card_{e}), lv_color_hex(0xF0FFF4), LV_PART_MAIN);
|
|
lv_obj_set_style_border_color(id(card_{e}), lv_color_hex(0x6BCB77), LV_PART_MAIN);
|
|
lv_obj_set_style_border_width(id(card_{e}), 3, LV_PART_MAIN);
|
|
lv_label_set_text(id(check_{e}), "\\u2705");
|
|
}} else {{
|
|
lv_obj_set_style_bg_color(id(card_{e}), lv_color_hex(0xFFFFFF), LV_PART_MAIN);
|
|
lv_obj_set_style_border_color(id(card_{e}), lv_color_hex(0xEEEEEE), LV_PART_MAIN);
|
|
lv_obj_set_style_border_width(id(card_{e}), 2, LV_PART_MAIN);
|
|
lv_label_set_text(id(check_{e}), "");
|
|
}}
|
|
"""
|
|
|
|
count_done = " + ".join(f"(id({eid(kid,c)}).state ? 1 : 0)" for c in chores)
|
|
reset_lines = "\n".join(
|
|
f" id({eid(kid,c)}).turn_off();" for c in chores
|
|
)
|
|
|
|
blocks.append(f"""\
|
|
- id: reset_{ks}_chores
|
|
mode: single
|
|
then:
|
|
- lambda: |-
|
|
{reset_lines}
|
|
- script.execute: update_{ks}_ui
|
|
|
|
""")
|
|
|
|
blocks.append(f"""\
|
|
- id: update_{ks}_ui
|
|
mode: single
|
|
then:
|
|
- lambda: |-
|
|
int done = {count_done};
|
|
int total = {n};
|
|
char buf[24];
|
|
|
|
// Progress bar + label
|
|
lv_bar_set_value(id(progress_bar_{ks}), done, LV_ANIM_ON);
|
|
snprintf(buf, sizeof(buf), "%d / {n}", done);
|
|
lv_label_set_text(id(progress_label_{ks}), buf);
|
|
|
|
// All-done message in sidebar
|
|
lv_label_set_text(id(all_done_label_{ks}), done == total ? "All done!" : "");
|
|
|
|
// Home button: RED outline = incomplete, SOLID GREEN = all done
|
|
if (done == total) {{
|
|
lv_obj_set_style_bg_opa(id(home_btn_{ks}), LV_OPA_COVER, LV_PART_MAIN);
|
|
lv_obj_set_style_bg_color(id(home_btn_{ks}), lv_color_hex(0x6BCB77), LV_PART_MAIN);
|
|
lv_obj_set_style_border_width(id(home_btn_{ks}), 0, LV_PART_MAIN);
|
|
lv_label_set_text(id(home_status_{ks}), "\\u2713 all done!");
|
|
lv_obj_set_style_text_color(id(home_status_{ks}), lv_color_hex(0xFFFFFF), LV_PART_MAIN);
|
|
}} else {{
|
|
lv_obj_set_style_bg_opa(id(home_btn_{ks}), LV_OPA_COVER, LV_PART_MAIN);
|
|
lv_obj_set_style_bg_color(id(home_btn_{ks}), lv_color_hex(0xFF4757), LV_PART_MAIN);
|
|
lv_obj_set_style_border_width(id(home_btn_{ks}), 0, LV_PART_MAIN);
|
|
snprintf(buf, sizeof(buf), "%d left", total - done);
|
|
lv_label_set_text(id(home_status_{ks}), buf);
|
|
lv_obj_set_style_text_color(id(home_status_{ks}), lv_color_hex(0xFF4757), LV_PART_MAIN);
|
|
}}
|
|
|
|
// Card colours
|
|
{card_updates}
|
|
""")
|
|
|
|
return "\n".join(blocks)
|
|
|
|
|
|
# ─────────────────────────────────────────────────────────────────────────────
|
|
# Home Assistant YAML generator
|
|
# ─────────────────────────────────────────────────────────────────────────────
|
|
|
|
def gen_ha(cfg: dict) -> str:
|
|
s = cfg["settings"]
|
|
kids = cfg["kids"]
|
|
|
|
sensor_blocks = []
|
|
for kid in kids:
|
|
chores = get_chores(kid)
|
|
ks = kid_slug(kid)
|
|
n = len(chores)
|
|
states_list = ", ".join(f"states('{ha_switch(kid,c)}')" for c in chores)
|
|
|
|
sensor_blocks.append(f"""\
|
|
- name: "{kid['name']} Chores Done Today"
|
|
unique_id: {ks}_chores_done_today
|
|
icon: mdi:check-circle
|
|
state: >
|
|
{{% set chores = [{states_list}] %}}
|
|
{{{{ chores | select('equalto', 'on') | list | count }}}}
|
|
|
|
- name: "{kid['name']} All Chores Done"
|
|
unique_id: {ks}_all_chores_done
|
|
icon: mdi:check-all
|
|
state: >
|
|
{{{{ states('sensor.{ks}_chores_done_today') | int == {n} }}}}
|
|
|
|
""")
|
|
|
|
input_buttons = "".join(f"""\
|
|
{kid_slug(k)}_reset_chores:
|
|
name: "Reset {k['name']}'s Chores"
|
|
icon: mdi:restart
|
|
""" for k in kids)
|
|
input_buttons += """\
|
|
reset_all_chores:
|
|
name: "Reset All Chores"
|
|
icon: mdi:restart-alert
|
|
"""
|
|
|
|
auto_blocks = []
|
|
|
|
for kid in kids:
|
|
ks = kid_slug(kid)
|
|
off_calls = "\n".join(
|
|
f" - service: switch.turn_off\n target:\n entity_id: {ha_switch(kid, c)}"
|
|
for c in get_chores(kid)
|
|
)
|
|
auto_blocks.append(f"""\
|
|
- id: {ks}_reset_from_ha
|
|
alias: "Chore Tracker - Reset {kid['name']}'s chores from HA"
|
|
trigger:
|
|
- platform: state
|
|
entity_id: input_button.{ks}_reset_chores
|
|
action:
|
|
{off_calls}
|
|
|
|
""")
|
|
|
|
all_off = "\n".join(
|
|
f" - service: switch.turn_off\n target:\n entity_id: {ha_switch(kid, c)}"
|
|
for kid in kids for c in get_chores(kid)
|
|
)
|
|
auto_blocks.append(f"""\
|
|
- id: reset_all_from_ha
|
|
alias: "Chore Tracker - Reset ALL chores from HA"
|
|
trigger:
|
|
- platform: state
|
|
entity_id: input_button.reset_all_chores
|
|
action:
|
|
{all_off}
|
|
|
|
""")
|
|
|
|
for kid in kids:
|
|
ks = kid_slug(kid)
|
|
n = len(get_chores(kid))
|
|
auto_blocks.append(f"""\
|
|
- id: {ks}_all_done_notify
|
|
alias: "Chore Tracker - {kid['name']} all done!"
|
|
trigger:
|
|
- platform: state
|
|
entity_id: sensor.{ks}_all_chores_done
|
|
to: "True"
|
|
action:
|
|
- service: {s['notify_service']}
|
|
data:
|
|
title: "{kid['name']} finished all chores!"
|
|
message: "{kid['name']} completed all {n} chores today!"
|
|
|
|
""")
|
|
|
|
reminder_actions = "".join(f"""\
|
|
- if:
|
|
condition: template
|
|
value_template: "{{{{ states('sensor.{kid_slug(k)}_all_chores_done') != 'True' }}}}"
|
|
then:
|
|
- service: {s['notify_service']}
|
|
data:
|
|
title: "{k['name']} has unfinished chores"
|
|
message: >
|
|
{k['name']} has done
|
|
{{{{ states('sensor.{kid_slug(k)}_chores_done_today') }}}}/{len(get_chores(k))} chores today.
|
|
""" for k in kids)
|
|
|
|
auto_blocks.append(f"""\
|
|
- id: chore_reminder_evening
|
|
alias: "Chore Tracker - Evening reminder"
|
|
trigger:
|
|
- platform: time
|
|
at: "{s['reminder_time']}:00"
|
|
action:
|
|
{reminder_actions}
|
|
""")
|
|
|
|
return f"""\
|
|
################################################################################
|
|
# chore-tracker-ha.yaml (AUTO-GENERATED — edit chores_config.yaml)
|
|
# Kids: {", ".join(k["name"] for k in kids)}
|
|
#
|
|
# BIDIRECTIONAL SYNC:
|
|
# Screen to HA: Each switch calls homeassistant.service on toggle
|
|
# HA to Screen: ESPHome native API handles this automatically
|
|
################################################################################
|
|
|
|
input_button:
|
|
{input_buttons}
|
|
template:
|
|
- sensor:
|
|
{"".join(sensor_blocks)}
|
|
automation:
|
|
{"".join(auto_blocks)}
|
|
"""
|
|
|
|
|
|
# ─────────────────────────────────────────────────────────────────────────────
|
|
# Lovelace dashboard generator
|
|
# ─────────────────────────────────────────────────────────────────────────────
|
|
|
|
def gen_dashboard(cfg: dict) -> str:
|
|
kids = cfg["kids"]
|
|
|
|
summary_cards = "\n".join(f"""\
|
|
- type: markdown
|
|
content: >
|
|
### {kid['avatar']} {kid['name']}
|
|
{{{{
|
|
'All done!' if states('sensor.{kid_slug(kid)}_all_chores_done') == 'True'
|
|
else states('sensor.{kid_slug(kid)}_chores_done_today') ~ '/{len(get_chores(kid))} chores done'
|
|
}}}}
|
|
""" for kid in kids)
|
|
|
|
chore_cards = []
|
|
for kid in kids:
|
|
chores = get_chores(kid)
|
|
ks = kid_slug(kid)
|
|
chore_rows = "\n".join(f"""\
|
|
- entity: {ha_switch(kid, c)}
|
|
name: "{c['icon']} {c['name']}"
|
|
""" for c in chores)
|
|
|
|
chore_cards.append(f"""\
|
|
- type: entities
|
|
title: "{kid['avatar']} {kid['name']}'s Chores"
|
|
entities:
|
|
{chore_rows} - type: divider
|
|
- entity: sensor.{ks}_chores_done_today
|
|
name: "Chores done today"
|
|
- entity: sensor.{ks}_all_chores_done
|
|
name: "All Done?"
|
|
- type: button
|
|
name: "Reset {kid['name']}'s Chores"
|
|
tap_action:
|
|
action: call-service
|
|
service: input_button.press
|
|
target:
|
|
entity_id: input_button.{ks}_reset_chores
|
|
""")
|
|
|
|
return f"""\
|
|
################################################################################
|
|
# chore-tracker-dashboard.yaml (AUTO-GENERATED — edit chores_config.yaml)
|
|
# Kids: {", ".join(k["name"] for k in kids)}
|
|
################################################################################
|
|
|
|
title: "Chore Tracker"
|
|
views:
|
|
- title: Today
|
|
icon: mdi:checkbox-marked-circle
|
|
path: chores
|
|
cards:
|
|
- type: horizontal-stack
|
|
cards:
|
|
{summary_cards}
|
|
- type: button
|
|
name: "Reset ALL Chores"
|
|
icon: mdi:restart-alert
|
|
tap_action:
|
|
action: call-service
|
|
service: input_button.press
|
|
target:
|
|
entity_id: input_button.reset_all_chores
|
|
{"".join(chore_cards)}
|
|
"""
|
|
|
|
|
|
# ─────────────────────────────────────────────────────────────────────────────
|
|
# Main
|
|
# ─────────────────────────────────────────────────────────────────────────────
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description="Chore Tracker config generator")
|
|
parser.add_argument("--config", default="chores_config.yaml")
|
|
parser.add_argument("--out", default=".")
|
|
args = parser.parse_args()
|
|
|
|
config_path = Path(args.config)
|
|
if not config_path.exists():
|
|
print(f"❌ Config not found: {config_path}")
|
|
sys.exit(1)
|
|
|
|
out_dir = Path(args.out)
|
|
out_dir.mkdir(parents=True, exist_ok=True)
|
|
|
|
class SecretLoader(yaml.SafeLoader):
|
|
pass
|
|
SecretLoader.add_constructor(
|
|
"!secret",
|
|
lambda loader, node: f"!secret {loader.construct_scalar(node)}"
|
|
)
|
|
|
|
with open(config_path) as f:
|
|
cfg = yaml.load(f, Loader=SecretLoader)
|
|
|
|
kids = cfg.get("kids", [])
|
|
if not kids:
|
|
print("❌ No kids defined in config!")
|
|
sys.exit(1)
|
|
|
|
errors = [k["name"] for k in kids if "chores" not in k or not k["chores"]]
|
|
if errors:
|
|
print(f"❌ These kids have no chores defined: {', '.join(errors)}")
|
|
print(" Add a 'chores:' list under each kid in chores_config.yaml")
|
|
sys.exit(1)
|
|
|
|
# Warn about missing icons (will use default, not crash)
|
|
for k in kids:
|
|
for c in k.get("chores", []):
|
|
if "icon" not in c:
|
|
print(f"⚠️ {k['name']} / '{c['name']}' has no icon — using default")
|
|
|
|
total_chores = sum(len(get_chores(k)) for k in kids)
|
|
print(f"✅ Config loaded: {len(kids)} kid(s), {total_chores} total chores")
|
|
for k in kids:
|
|
chores = get_chores(k)
|
|
print(f" {k['avatar']} {k['name']}: {len(chores)} chores"
|
|
f" — {', '.join(c['name'] for c in chores)}")
|
|
print()
|
|
|
|
files = {
|
|
"chore-tracker-esphome.yaml": gen_esphome(cfg),
|
|
"chore-tracker-ha.yaml": gen_ha(cfg),
|
|
"chore-tracker-dashboard.yaml": gen_dashboard(cfg),
|
|
}
|
|
|
|
for name, content in files.items():
|
|
path = out_dir / name
|
|
path.write_text(content)
|
|
print(f"📄 {path}")
|
|
|
|
print()
|
|
print("✅ Done!")
|
|
print(" 1. Flash chore-tracker-esphome.yaml -> ESPHome dashboard")
|
|
print(" 2. Merge chore-tracker-ha.yaml -> HA config + restart HA")
|
|
print(" 3. Paste chore-tracker-dashboard.yaml -> New HA dashboard")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main() |