diff options
Diffstat (limited to 'app.py')
| -rw-r--r-- | app.py | 86 |
1 files changed, 74 insertions, 12 deletions
@@ -3,6 +3,7 @@ from hashlib import scrypt from os import urandom from hmac import compare_digest import sqlite3 +from colorsys import rgb_to_hsv from flask import ( Flask, @@ -14,7 +15,8 @@ from flask import ( g, flash, ) -from werkzeug.exceptions import HTTPException, NotImplemented +from werkzeug.exceptions import HTTPException, NotImplemented, BadRequest +import requests from captcha import Captcha from presets import GUEST_CAPTCHA, GUIDE_CAPTCHAS @@ -123,6 +125,19 @@ def index(): return render_template("index.html", news=results) +@app.route("/captcha") +def captcha(): + size = request.args.get("size", "") + label = request.args.get("label", "") + lock = request.args.get("lock", "") + answer = request.args.get("answer", "") + captcha = Captcha.safe_validate(size, label, lock, answer) + if captcha: + add_captcha(captcha.id) + return {"valid": True}, 200 + return {"valid": False}, 400 + + @app.route("/login", methods=["GET", "POST"]) def login(): db = get_db() @@ -206,17 +221,64 @@ def news(): return render_template("news.html", news=results) -@app.route("/captcha") -def captcha(): - size = request.args.get("size", "") - label = request.args.get("label", "") - lock = request.args.get("lock", "") - answer = request.args.get("answer", "") - captcha = Captcha.safe_validate(size, label, lock, answer) - if captcha: - add_captcha(captcha.id) - return {"valid": True}, 200 - return {"valid": False}, 400 +@app.route("/mgmt", methods=["GET", "POST"]) +def management(): + if request.method == "POST": + try: + lid_key = request.form.get("lid", "custom") + if lid_key == "custom": + lid = int(request.form.get("custom", "5")) + else: + lid = int(lid_key) + except ValueError: + raise BadRequest + set_key = request.form.get("set") + match set_key: + case "off": + tool_disable_light(lid) + case "on": + tool_enable_light(lid) + case "color": + tool_color_light(lid, request.form.get("color", "#FF00AA")) + case _: + raise BadRequest + return render_template("mgmt.html") + + +def tool_disable_light(light: int): + requests.put( + f"{LIGHT_URL}/lights/{light}/state", + json={"on": False}, + timeout=5, + ) + + +def tool_enable_light(light: int): + requests.put( + f"{LIGHT_URL}/lights/{light}/state", + json={"on": True, "ct": 366}, + timeout=5, + ) + + +def tool_color_light(light: int, color: str): + if color[0] == "#": + color = color[1:] + color = color.upper() + if len(color) != 6 or not all(c in HEX_CHARS for c in color): + color = "FF00AA" + rgb = [int(color[i : i + 2], 16) / 255 for i in range(0, 6, 2)] + hsv = rgb_to_hsv(*rgb) + requests.put( + f"{LIGHT_URL}/lights/{light}/state", + json={ + "on": True, + "hue": int(hsv[0] * 65535), + "sat": int(hsv[1] * 254), + "bri": int(hsv[2] * 254), + }, + timeout=5, + ) @app.route("/permits") |