Changeset - 275690f3bbdc
[Not reviewed]
default
0 3 0
Dennis Fink - 4 years ago 2021-03-20 12:36:37
dennis.fink@c3l.lu
Black autoformat
3 files changed with 20 insertions and 20 deletions:
0 comments (0 inline, 0 general)
setup.py
Show inline comments
 
from pprint import pprint
 

	
 
from setuptools import setup, find_packages
 
from setuptools import find_packages, setup
 

	
 
packages = find_packages()
 
packages.append('spaceapi.templates')
 
packages.append('spaceapi.static')
 
packages.append('spaceapi.schema')
 
packages.append("spaceapi.templates")
 
packages.append("spaceapi.static")
 
packages.append("spaceapi.schema")
 

	
 
setup(
 
    name='c3l_spaceapi',
 
    version='0.0.14',
 
    url=None,
 
    license='GPLv3+',
 
    author='Dennis Fink',
 
    author_email='dennis.fink@c3l.lu',
 
    description='spaceapi endpoint for c3l.lu',
 
    license="GPLv3+",
 
    author="Dennis Fink",
 
    author_email="dennis.fink@c3l.lu",
 
    description="spaceapi endpoint for c3l.lu",
 
    packages=packages,
 
    package_data={
 
        'spaceapi.templates': ['*'],
 
        'spaceapi.static': ['*'],
 
        'spaceapi.schema': ['*'],
 
        "spaceapi.templates": ["*"],
 
        "spaceapi.static": ["*"],
 
        "spaceapi.schema": ["*"],
 
    },
 
    install_requires=[
 
        'Flask',
 
        'Flask-HTTPAuth',
 
        'Flask-Bootstrap',
 
        'jsonschema',
 
        'tweepy',
 
        'Mastodon.py',
 
    ]
 
        "Flask",
 
        "Flask-HTTPAuth",
 
        "Flask-Bootstrap",
 
        "jsonschema",
 
        "tweepy",
 
        "Mastodon.py",
 
    ],
 
)
spaceapi/sensors.py
Show inline comments
 
import json
 
from functools import partial
 

	
 
from pkg_resources import resource_filename
 

	
 
import jsonschema
 
from flask import Blueprint, abort, current_app, jsonify, request
 
from pkg_resources import resource_filename
 

	
 
from .auth import httpauth
 
from .utils import ActiveStatus, ActiveStatusv14, first, fuzzy_list_find
 

	
 
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])
 
        if key == "barometer":
 
            data["unit"] == "hPA"
 
        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])
 
        if key == "barometer":
 
            data["unit"] = "hPa"
 
        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:
 
                current_app.logger.error("Validation Error")
 
                return abort(400)
 

	
 
            if key != "radiation":
 
                return set_value(data, key)
 
            else:
 
                return set_radiation_value(data)
 

	
 
        except ValueError:
 
            current_app.logger.error("Value Error")
spaceapi/views.py
Show inline comments
 
@@ -21,99 +21,100 @@ def index():
 
        return jsonify(ActiveStatus())
 
    return render_template("index.html", status=ActiveStatus())
 

	
 

	
 
@root_views.route("/status.json")
 
def status_json():
 
    return jsonify(ActiveStatus())
 

	
 

	
 
@root_views.route("/v14/status.json")
 
def v14_json():
 
    return jsonify(ActiveStatusv14())
 

	
 

	
 
@root_views.route("/reload")
 
@httpauth.login_required
 
def reload():
 
    active = ActiveStatus()
 
    active.reload()
 
    return jsonify(active)
 

	
 

	
 
@root_views.route("/v14/reload")
 
@httpauth.login_required
 
def v14_reload():
 
    active = ActiveStatusv14()
 
    active.reload()
 
    return jsonify(active)
 

	
 

	
 
@root_views.route("/open", methods=("GET", "POST"))
 
@httpauth.login_required
 
def open():
 
    if request.method == "POST":
 
        active = ActiveStatus()
 
        activev14 = ActiveStatusv14()
 

	
 
        try:
 
            if (
 
                httpauth.username()
 
                in current_app.config["STATE_TRIGGER_PERSON_ALLOWED"]
 
            ):
 
                trigger_person = httpauth.username()
 
            else:
 
                trigger_person = None
 
        except KeyError:
 
            trigger_person = None
 

	
 
        if "close" in request.form:
 
            new_state = False
 
        elif "open" in request.form:
 
            new_state = True
 

	
 
        if "message" in request.form:
 
            message = request.form.get("message")
 
        else:
 
            message = None
 

	
 
        active.set_new_state(
 
            value=new_state, trigger_person=trigger_person, message=message
 
        )
 
        activev14.set_new_state(
 
            value=new_state, trigger_person=trigger_person, message=message
 
        )
 
        active.save_last_state()
 
        activev14.save_last_state()
 
        return redirect(url_for("root.index"))
 

	
 
    return render_template("open.html")
 

	
 

	
 
@root_views.route("/present", methods=("GET", "POST"))
 
@httpauth.login_required
 
def present():
 
    if request.method == "POST":
 

	
 
        active = ActiveStatus()
 
        activev14 = ActiveStatusv14()
 

	
 
        if active["state"]["open"]:
 
            user = (
 
                httpauth.username()
 
                if "user" not in request.form
 
                else request.form["user"]
 
            )
 
            if "present" in request.form:
 
                active.add_user_present(user)
 
                activev14.add_user_present(user)
 
            elif "leave" in request.form:
 
                active.remove_user_present(user)
 
                activev14.remove_user_present(user)
 
            else:
 
                return redirect(url_for("root.index"))
 

	
 
            active.save_last_state()
 
            activev14.save_last_state()
 

	
 
        return redirect(url_for("root.index"))
 

	
 
    return render_template("present.html")
0 comments (0 inline, 0 general)