from flask import Blueprint, request, jsonify
from web_server_models import copters, WebCopter
from server import Client

selfcheck_api = Blueprint('selfcheck_api', __name__, template_folder='templates')


@selfcheck_api.route('/selfcheck/selected', methods=["GET", "POST"])
def selfcheck_selected():
    data = dict()
    ip = request.args.get("ip")
    for copter in copters:
        if copter.ip == ip:
            copter.refresh()
            data = {
                'anim_id': copter.anim_id,
                'batt_voltage': copter.batt_voltage,
                'cell_voltage': copter.cell_voltage,
                'selfcheck': copter.selfcheck,
                'time': copter.time,
                'name': copter.name,
            }
    # data = {"anim_id": "No animation", "batt_voltage": 3.259999990463257, "cell_voltage": 1.0850000381469727,
    #        "ip": "192.168.43.31", "name": "CLever7", "selfcheck": "OK", "time": 1554723269.57106}
    return jsonify(data)


@selfcheck_api.route('/selfcheck/all', methods=["GET", "POST"])
def selfcheck_all():
    data = []
    for copter in copters:
        copter.refresh()
        data.append({
            'anim_id': copter.anim_id,
            'batt_voltage': copter.batt_voltage,
            'cell_voltage': copter.cell_voltage,
            'selfcheck': copter.selfcheck,
            'ip': copter.ip,
            'time': copter.time,
            'name': copter.name,
        })
    # data = [{"anim_id": "No animation", "batt_voltage": 4.259999990463257, "cell_voltage": 4.0850000381469727,
    #        "ip": "192.168.43.31", "name": "CLever7", "selfcheck": "OK", "time": 4554723269.57106}]
    # data *= 12
    return jsonify(data)


@selfcheck_api.route('/refresh_copters')
def refresh_copters():
    try:
        for client_ip in Client.clients.keys():
            is_in = False
            for copter in copters:
                if client_ip == copter.ip:
                    is_in = True
            if not is_in:
                copters.append(WebCopter(client_ip, Client.clients[client_ip]))
        return jsonify({'m': 'Ok'})
    except:
        return jsonify({'m': 'Error'})


@selfcheck_api.route('/test_led/selected', methods=['GET', 'POST'])
def test_led_selected():
    ip = request.args.get("ip")
    for copter in copters:
        if copter.ip == ip:
            copter.client.send_message("led_test")
    return jsonify({'m': 'ok'})