LED-Control/server.py

46 lines
999 B
Python
Raw Normal View History

2021-10-25 20:04:55 +01:00
from flask import Flask, request
2021-10-26 16:55:30 +01:00
import ast
2021-10-26 15:32:50 +01:00
import subprocess
2021-10-25 20:04:55 +01:00
from colormap import rgb2hex
2021-10-26 15:56:50 +01:00
from tenacity import retry, stop_after_attempt
2021-10-25 20:04:55 +01:00
app = Flask(__name__)
2021-10-26 15:56:50 +01:00
@retry(stop=stop_after_attempt(5))
2021-10-26 16:55:30 +01:00
def run_bulb_action(value):
2021-10-26 15:48:40 +01:00
result = subprocess.run(
2021-10-26 16:55:30 +01:00
[
f"gatttool -i hci0 -b b2:3b:03:00:14:d6 --char-write-req -a 0x0009 -n {value}"
],
2021-10-26 15:48:40 +01:00
stdout=subprocess.PIPE,
shell=True,
).stdout.decode("utf-8")
print(result)
if "Characteristic value was written successfully" not in result:
raise Exception
else:
2021-10-26 16:55:30 +01:00
return True
@app.route("/on")
def on():
2021-10-26 16:58:10 +01:00
return {run_bulb_action("cc2333")}
2021-10-26 10:39:04 +01:00
@app.route("/off")
2021-10-26 10:40:10 +01:00
def off():
2021-10-26 16:58:10 +01:00
return {run_bulb_action("cc2433")}
2021-10-25 20:04:55 +01:00
2021-10-26 10:33:37 +01:00
@app.route("/colour")
2021-10-26 16:55:30 +01:00
def colour():
rgb = list(ast.literal_eval(request.args.get("rgb")))
2021-10-26 16:33:17 +01:00
r = rgb[0]
g = rgb[1]
b = rgb[2]
hex = rgb2hex(r, g, b)
hex = hex.replace("#", "")
2021-10-26 16:55:30 +01:00
value = "56" + hex + "00f0aa"
2021-10-26 16:58:10 +01:00
return {run_bulb_action(value)}