import json

from functools import partial

from pkg_resources import resource_filename

import jsonschema

from flask import Blueprint, jsonify, request, abort, current_app

from .auth import httpauth

from .utils import first, fuzzy_list_find, ActiveStatus, ActiveStatusv14

sensors_views = Blueprint("sensors", __name__)

ALLOWED_SENSORS_KEYS = json.load(

    open(resource_filename("spaceapi", "schema/sensors.json"), encoding="utf-8")

)

RADIATON_SUBKEYS = frozenset(("alpha", "beta", "gamma", "beta_gamma"))

get_identification_key = partial(first, keys=frozenset(("name", "location")))

def set_value(data, key):

    active = ActiveStatus()

    active_v14 = ActiveStatusv14()

    try:

        subkey = get_identification_key(data)

    except ValueError:

        current_app.logger.error("Subkey Error")

        return abort(400)

    try:

        index = fuzzy_list_find(active["sensors"][key], subkey, data[subkey])

        active["sensors"][key][index].update(data)

    except ValueError:

        active["sensors"][key].append(data)

    try:

        index_v14 = fuzzy_list_find(active_v14["sensors"][key], subkey, data[subkey])

        active_v14["sensors"][key][index].update(data)

    except ValueError:

        active_v14["sensors"][key].append(data)

    return jsonify(active)

def set_radiation_value(data):

    active = ActiveStatus()

    radiation_keys = [k for k in RADIATON_SUBKEYS if k in data]

    if not radiation_keys:

        return abort(400)

    for first_subkey in radiation_keys:

        try:

            second_subkey = get_identification_key(data[first_subkey])

        except ValueError:

            return abort(400)

        try:

            index = fuzzy_list_find(

                active["sensors"]["radiation"][first_subkey],

                second_subkey,

                data[first_subkey][second_subkey],

            )

            active["sensors"]["radiation"][first_subkey][index].update(data)

        except ValueError:

            active["sensors"]["radiaton"][first_subkey].append(data)

    return jsonify(active)

@sensors_views.route("/set/<key>", methods=["POST"])

@httpauth.login_required

def set_sensors(key):

    active = ActiveStatus()

    if key in ALLOWED_SENSORS_KEYS and key in active["sensors"]:

        data = request.json

        try:

            try:

                jsonschema.validate(data, ALLOWED_SENSORS_KEYS[key])

            except jsonschema.ValidationError: