server.py

import os
import time
import redis
import logging

from flask import Flask, render_template, request, abort

import animator

from helpers import colorTools, microcontroller, animations, dimmer
from helpers.state import State, Multiplier

SECONDS_TO_IDLE = 60 * 29

# State management
cacheBust = int(time.time())
r = redis.Redis(charset="utf-8", decode_responses=True)
r.ping()

logging.basicConfig(format='%(levelname)s: %(message)s', level=logging.INFO)

app = Flask(__name__)


@app.route("/")
def homepage():
    return render_template('homepage.html', cacheBust=cacheBust)


@app.route("/random/<type>", methods=['POST'])
def random_color(type):
    onlyOnIdle = request.args.get('automated')
    now = int(time.time())

    state = State.fromRedis(r)

    if onlyOnIdle:
        secondUntilIdle = state.lastModified + SECONDS_TO_IDLE - now

        if state.lockedBy:
            logging.info(f"Ignoring automated colour request, colours locked by {state.lockedBy}")
            return '', 304

        if secondUntilIdle > 0:
            logging.info(f"Ignoring automated colour request, not idle for another {secondUntilIdle / 60} mins")
            return '', 304

    hue = colorTools.generateRandomDifferentHue(state.colors[0].hue) if len(state.colors) > 0 else colorTools.generateRandomHue()
    color = colorTools.Color.fromHue(hue)

    if type == 'single':
        colors = color.multiplyBy(microcontroller.NUM_LEDS)

        newState = State(now, [color], Multiplier.SINGLE_COLOR)
    elif type == 'columns':
        secondHue = colorTools.generateRandomDifferentHue(hue)
        secondColor = colorTools.Color.fromHue(secondHue)
        colors = colorTools.generateGradientColumns(color, secondColor)

        newState = State(now, [color, secondColor], Multiplier.COLUMNS_GRADIENT)
    elif type == 'gradient':
        secondHue = colorTools.generateRandomDifferentHue(hue)
        secondColor = colorTools.Color.fromHue(secondHue)
        steps = microcontroller.NUM_LEDS
        colors = colorTools.generateColorGradient(color, secondColor, steps)

        newState = State(now, [color, secondColor], Multiplier.GRADIENT)
    else:
        abort(400, description=f'Unknown type of random display, given "{type}"')

    animator.send(None)

    response = microcontroller.sendColors(dimmer.dimColorsIfNight(colors))

    newState.save(r)

    return response


@app.route("/lights", methods=['POST'])
def lights():
    inputData = request.get_json(force=True)

    multiplier = inputData['multiplier']
    rawColors = inputData['colors']

    if len(rawColors) == 0:
        abort(400, description='No colors given')

    colors = list(map(colorTools.Color.fromDict, rawColors))
    now = int(time.time())

    if multiplier == 'columns':
        ledColors = colorTools.generateLedColumns(colors)
        newState = State(now, colors, Multiplier.COLUMNS)
    elif isinstance(multiplier, int):
        ledColors = colorTools.generateLedBlocks(colors, multiplier)
        newState = State(now, colors, Multiplier.REPEATING)
    else:
        abort(400, description=f'Unknown multiplier "{multiplier}"')

    animator.send(None)

    response = microcontroller.sendColors(dimmer.dimColorsIfNight(ledColors))

    newState.save(r)

    return response


@app.route("/animations/<name>", methods=['POST'])
def animation(name):
    if name not in animations.animations:
        animationNames = ", ".join(animations.animations.keys())
        abort(400, description=f'No animation found with name "{name}". Available: {animationNames}')
    else:
        animationInstance = animations.animations[name]()

    animator.send(animationInstance)

    lastModified = int(time.time())
    newState = State(lastModified, [], Multiplier.NONE, name)
    newState.save(r)

    return ''


@app.route("/state")
def led_state():
    # Blah blah blah API boundaries
    return r.get(State.REDIS_KEY)


@app.route("/lock", methods=['POST'])
def lock():
    state = State.fromRedis(r)
    state.lockedBy = request.remote_addr

    state.save(r)

    return '', 201


@app.route("/lock", methods=['DELETE'])
def unlock():
    state = State.fromRedis(r)
    state.lockedBy = None

    state.save(r)

    return '', 204


if __name__ == "__main__":
    port = int(os.environ.get('PORT', 8080))
    app.run(host='0.0.0.0', port=port)